Spaces:
Running
Running
File size: 3,352 Bytes
0f82a63 6161c48 e5c9720 6161c48 9378914 6161c48 0550657 6161c48 6da37cf 9378914 03669ec 6161c48 e5c9720 03669ec 6161c48 e5c9720 157d667 e5c9720 6161c48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
import streamlit as st
import logging
# Function to load the CSS file
def load_css(file_name):
with open(file_name) as f:
st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
# Load the custom CSS
load_css("./.streamlit/main.css")
st.markdown("<h1 style='text-align: center; color: #666666;'>Vector Database RAG Proof of Concept</h1>", unsafe_allow_html=True)
st.markdown("<h6 style='text-align: center; color: #666666;'>V1</h6>", unsafe_allow_html=True)
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.DEBUG)
def setPrompt(pprompt,ragFlag):
logger.debug("### setPrompt")
userPrompt = ""
return userPrompt
def runLLM(prompt):
logger.debug("### runLLM")
result = "LLM RESULT"
return(result)
def on_submitButton_clicked():
logger.debug("\n### on_submitButton_clicked")
st.session_state.sysTAtext = st.session_state.sysTA
logger.info(f"sysTAtext: {st.session_state.sysTAtext}")
st.session_state.userpTAtext = setPrompt("","")
st.session_state.userpTA = st.session_state.userpTAtext
logger.info(f"userpTAtext: {st.session_state.userpTAtext}")
st.session_state.rspTAtext = runLLM(st.session_state.userpTAtext)
st.session_state.rspTA = st.session_state.rspTAtext
logger.info(f"rspTAtext: {st.session_state.rspTAtext}")
######################################################################
# MAINLINE
#
logger.info("#### MAINLINE ENTERED.")
col1, col2 = st.columns(2)
with col1:
if "sysTA" not in st.session_state:
st.session_state.sysTA = st.text_area(label="sysTA",value="fdsaf fsdafdsa")
elif "sysTAtext" in st.session_state:
st.session_state.sysTA = st.text_area(label="sysTA",value=st.session_state.sysTAtext)
else:
st.session_state.sysTA = st.text_area(label="sysTA",value=st.session_state.sysTA)
if "userpTA" not in st.session_state:
userTextArea = st.text_area(label="userpTA",value="fdsaf fsdafdsa")
elif "userpTAtext" in st.session_state:
st.session_state.userpTA = st.text_area(label="userpTA",value=st.session_state.userpTAtext)
else:
st.session_state.userpTA = st.text_area(label="userpTA",value=st.session_state.userpTA)
with col2:
if "ragpTA" not in st.session_state:
ragPromptTextArea = st.text_area(label="ragpTA",value="fdsaf fsdafdsa")
elif "ragpTAtext" in st.session_state:
st.session_state.ragpTA = st.text_area(label="ragpTA",value=st.session_state.ragpTAtext)
else:
st.session_state.ragTA = st.text_area(label="ragTA",value=st.session_state.ragTA)
if "rspTA" not in st.session_state:
responseTextArea = st.text_area(label="rspTA",value="fdsaf fsdafdsa")
elif "rspTAtext" in st.session_state:
st.session_state.rspTA = st.text_area(label="rspTA",value=st.session_state.rspTAtext)
else:
st.session_state.rspTA = st.text_area(label="rspTA",value=st.session_state.rspTA)
with st.sidebar:
st.selectRag = st.checkbox("Enable Query With RAG",value=False,key="selectRag",help=None,on_change=None,args=None,kwargs=None,disabled=False,label_visibility="visible")
st.submitButton = st.button("Run LLM Query",key=None,help=None,on_click=on_submitButton_clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
|