Michael McGuinness
72 lines
1.6 KiB
Python
72 lines
1.6 KiB
Python
import json
|
|
import boto3
|
|
from langchain.prompts import PromptTemplate
|
|
from langchain.embeddings.bedrock import BedrockEmbeddings
|
|
from langchain.llms.bedrock import Bedrock
|
|
from langchain_community.vectorstores import Chroma
|
|
from constants import (
|
|
CHROMA_SETTINGS,
|
|
EMBEDDING_MODEL_NAME,
|
|
PERSIST_DIRECTORY,
|
|
MODEL_ID,
|
|
MODEL_BASENAME,
|
|
SOURCE_DIRECTORY,
|
|
USER_LIST,
|
|
)
|
|
from langchain.chains import RetrievalQA
|
|
|
|
import streamlit as st
|
|
from streamlit_extras.add_vertical_space import add_vertical_space
|
|
|
|
import pandas as pd
|
|
import numpy as np
|
|
from datetime import datetime
|
|
import random
|
|
import os
|
|
import dateutil
|
|
import util
|
|
import anthropic
|
|
import re
|
|
import snowflake.connector
|
|
from sf_conn import get_secret, save_to_sf
|
|
from io import StringIO
|
|
import pandas as pd
|
|
|
|
|
|
(REDIRECT_URI, create_batch_url, doczy_pipeline) = util.load_page_details(5)
|
|
user_list = USER_LIST
|
|
|
|
st.set_page_config(layout="wide")
|
|
# Sidebar contents
|
|
with st.sidebar:
|
|
st.title("Doczy.AI ™")
|
|
st.markdown(
|
|
"""
|
|
## About
|
|
This app extracts data from contracts
|
|
|
|
"""
|
|
)
|
|
add_vertical_space(15)
|
|
# st.write("Doczy")
|
|
|
|
_, c1 = st.columns([5, 1])
|
|
# util.setup_page(REDIRECT_URI)
|
|
try:
|
|
util.setup_page(REDIRECT_URI)
|
|
except:
|
|
st.write("SSO Failed")
|
|
st.session_state["user_info"] = {
|
|
"mail": "maamseek@aarete.com",
|
|
"displayName": "Mayank Aamseek",
|
|
}
|
|
try:
|
|
c1.write(f"User: **{st.session_state.user_info['displayName']}**")
|
|
user_mail = st.session_state.user_info["mail"]
|
|
except KeyError as e:
|
|
st.write("Session Expired.")
|
|
st.stop()
|
|
|
|
data_dict = pd.read_csv("contract_fields.csv")
|
|
st.data_editor(data_dict, height=700)
|