Spaces:
Running
Running
File size: 3,802 Bytes
0f82a63 6161c48 e5c9720 6161c48 9378914 6161c48 0550657 6161c48 af57ef0 6161c48 e5c9720 af57ef0 6161c48 6da37cf 9378914 6161c48 6da37cf 9378914 af57ef0 6161c48 e5c9720 af57ef0 6161c48 e5c9720 af57ef0 6da37cf 0550657 af57ef0 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 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
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)
<<<<<<< HEAD
def setPrompt(pprompt,ragFlag):
logger.debug("### setPrompt")
userPrompt = "USER PROMPT"
=======
def runLLM(prompt):
logger.debug("### runLLM")
result = ""
return(result)
def setPrompt(pprompt,ragFlag):
logger.debug("### setPrompt")
userPrompt = ""
>>>>>>> 6da37cf832a8bf0248b12ed1c31fd19a9fd63832
return userPrompt
def runLLM(prompt):
logger.debug("### runLLM")
result = "LLM RESULT"
return(result)
<<<<<<< HEAD
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}")
=======
def on_submitButton_clicked():
logger.debug("### on_submitButton_clicked")
prompt = setPrompt("","")
runLLM(prompt)
>>>>>>> 6da37cf832a8bf0248b12ed1c31fd19a9fd63832
######################################################################
# 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)
|