minor tweaks

pages/ragChat.py

@@ -11,9 +11,13 @@ async def main():
     st.title("Upload Data")
 
     uploaded_files = st.file_uploader("Upload PDFs", accept_multiple_files=True)
+    st.warning("If you plan to add more files, after processing initial files, make sure the uploaded files you already processed are removed")
     url = st.text_input("Enter a website link")
+    if "button_pressed" not in st.session_state:
+        st.session_state.button_pressed = False
     if st.button('Process URL and Files'):
-        with st.spinner("Vectorizing Data, wait times vary depending size..."):
+        st.session_state.button_pressed = True
+        with st.spinner("Vectorizing Data, wait times vary depending on size..."):
             if url:
                 try:
                     if "retriever" not in st.session_state:
@@ -33,42 +37,43 @@ async def main():
                 except Exception as e:
                     st.error(f"Failed to load PDF: {e}")
             st.success("Data is ready to be queried!")
+    
+    if st.session_state.button_pressed:
+        if "chat_history" not in st.session_state:
+            st.session_state.chat_history = [AIMessage(content="Hello, I am a bot. How can I help you?")]
 
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = [AIMessage(content="Hello, I am a bot. How can I help you?")]
+        st.title("RAG CHAT")
+        for message in st.session_state.chat_history:
+            if isinstance(message, AIMessage):
+                with st.chat_message("AI"):
+                    st.write(message.content)
+            elif isinstance(message, HumanMessage):
+                with st.chat_message("Human"):
+                    st.write(message.content)
 
-    st.title("RAG CHAT")
-    for message in st.session_state.chat_history:
-        if isinstance(message, AIMessage):
-            with st.chat_message("AI"):
-                st.write(message.content)
-        elif isinstance(message, HumanMessage):
+        user_query = st.chat_input("Type your message here...", key="chat_input")
+        if user_query:
+            st.session_state.chat_history.append(HumanMessage(content=user_query))
             with st.chat_message("Human"):
-                st.write(message.content)
+                st.write(user_query)
 
-    user_query = st.chat_input("Type your message here...", key="chat_input")
-    if user_query:
-        st.session_state.chat_history.append(HumanMessage(content=user_query))
-        with st.chat_message("Human"):
-            st.write(user_query)
-
-        if 'retriever' in st.session_state:
-            try:
-                ragAnswer = await st.session_state.retriever.amax_marginal_relevance_search(user_query, k=2, fetch_k=10)
-                context = []
-                for i, doc in enumerate(ragAnswer):
-                    print(f"{i}: {doc.page_content}")
-                    context.append(doc.page_content)
-                with st.spinner("Generating Response"):
-                    response = get_response(user_query, st.session_state.chat_history, context)
-                if response:
-                    st.session_state.chat_history.append(AIMessage(content=response))
-                    with st.chat_message("AI"):
-                        st.write(response)
-                else:
-                    st.write("No response received.") 
-            except Exception as e:
-                st.error(f"Error during retrieval or response generation: {e}")
+            if 'retriever' in st.session_state:
+                try:
+                    ragAnswer = await st.session_state.retriever.amax_marginal_relevance_search(user_query, k=4, fetch_k=10)
+                    context = []
+                    for i, doc in enumerate(ragAnswer):
+                        print(f"{i}: {doc.page_content}")
+                        context.append(doc.page_content)
+                    with st.spinner("Generating Response"):
+                        response = get_response(user_query, st.session_state.chat_history, context)
+                    if response:
+                        st.session_state.chat_history.append(AIMessage(content=response))
+                        with st.chat_message("AI"):
+                            st.write(response)
+                    else:
+                        st.write("No response received.") 
+                except Exception as e:
+                    st.error(f"Error during retrieval or response generation: {e}")
 
 if __name__ == "__main__":
     asyncio.run(main())