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)