removed unused session state & fixed bug

pages/ragChat.py

@@ -14,10 +14,8 @@ async def add_data():
         "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"):
-        st.session_state.button_pressed = True
         with st.spinner("Vectorizing Data, wait times vary depending on size..."):
             if url:
                 try:
@@ -27,6 +25,7 @@ async def add_data():
                         )
                 except Exception as e:
                     st.error(f"Failed to load URL: {e}")
+
             if uploaded_files:
                 try:
                     texts = get_pdf_text(uploaded_files)
@@ -39,67 +38,67 @@ async def add_data():
                         st.error("PDF has no meta data text")
                 except Exception as e:
                     st.error(f"Failed to load PDF: {e}")
+
             st.success("Data is ready to be queried!")
-            st.session_state.data_hungry = False
+    st.session_state.data_hungry = False
 
 
 async def rag_chat():
-    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)
-
-        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))
+    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(user_query)
+                st.write(message.content)
 
-            if "retriever" in st.session_state:
-                try:
-                    ragAnswer = (
-                        await st.session_state.retriever.amax_marginal_relevance_search(
-                            user_query, k=4, fetch_k=10
-                        )
+    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=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}")
+                )
+                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}")
 
 
 async def main():
-    st.session_state.data_hungry = st.toggle("Add Custom Data", False)
-
-    if(st.session_state.data_hungry):
+    if st.session_state.data_hungry:
         await add_data()
     else:
         await rag_chat()
 
+
 if __name__ == "__main__":
+    st.session_state.data_hungry = st.toggle("Add Custom Data", False)
     sidebar()
     asyncio.run(main())