Update app.py

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 from utils import *
 import constants
 
-# Creating Session State Variable
+# Creating Session State Variables
+
 if 'Website_URL' not in st.session_state:
     st.session_state['Website_URL'] =''
 if 'HuggingFace_API_Key' not in st.session_state:
@@ -13,6 +14,19 @@ if 'Pinecone_EBV' not in st.session_state:
     st.session_state['Pinecone_ENV'] =''
 if 'Pinecone_API_Key' not in st.session_state:
     st.session_state['Pinecone_API_Key'] =''
+if 'API_Key' not in st.session_state:
+    st.session_state['API_Key'] = ''
+if 'history' not in st.session_state:
+    st.session_state['history'] = ''
+
+
+
+# clear the chat history from streamlit session state
+def clear_history():
+    if 'history' in st.session_state:
+        del st.session_state['history']
+
+
 
 #
 st.title('🤖 AI Assistance For Website') 
@@ -65,7 +79,36 @@ submit = st.button("Search")
 
 if submit:
     #Proceed only if API keys are provided
-    if st.session_state['Website_URL'] !="" and st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_ENV'] !="" and st.session_state['Pinecone_API_Key']!="" :
+    if st.session_state['Website_URL'] !="" and st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_ENV'] !="" and st.session_state['Pinecone_API_Key']!="" and st.session_state['API_Key'] != '':
+
+        st.session_state['API_Key'] = st.text_input("Paste in your OPENAI API key?", type="password")
+        
+        q = st.text_input('Ask a question about the content of your file:')
+        if q:  # if the user entered a question and hit enter
+            if 'vs' in st.session_state:  # if there's the vector store (user uploaded, split and embedded a file)
+                vector_store = st.session_state.vs
+                st.write(f'k: {k}')
+                answer = ask_and_get_answer(vector_store, q, k)
+    
+                # text area widget for the LLM answer
+                st.text_area('LLM Answer: ', value=answer)
+    
+                st.divider()
+    
+                # if there's no chat history in the session state, create it
+                if 'history' not in st.session_state:
+                    st.session_state.history = ''
+    
+                # the current question and answer
+                value = f'Q: {q} \nA: {answer}'
+    
+                st.session_state.history = f'{value} \n {"-" * 100} \n {st.session_state.history}'
+                h = st.session_state.history
+    
+                # text area widget for the chat history
+                st.text_area(label='Chat History', value=h, key='history', height=400)
+
+        '''       
 
         #Creating embeddings instance
         embeddings=create_embeddings()
@@ -75,6 +118,8 @@ if submit:
         index=pull_from_pinecone(st.session_state['Pinecone_API_Key'],st.session_state['Pinecone_ENV'],st.session_state['Pinecone_Index'],embeddings)
         st.write("Pinecone index retrieval done...")
 
+        
+
         #Fetch relavant documents from Pinecone index
         relavant_docs=get_similar_docs(index,prompt,document_count)
         st.write(relavant_docs)
@@ -89,7 +134,7 @@ if submit:
             st.write("**Info**: "+document.page_content)
             st.write("**Link**: "+ document.metadata['source'])
        
-
+        '''
 
     else:
         st.sidebar.error("Ooopssss!!! Please provide API keys.....")