abstracted

functions/gptResponse.py

@@ -8,7 +8,8 @@ load_dotenv()
 openai_key = os.getenv(
     "OPENAI_API_KEY"
 )  # may wanna ask user for this or handle error when its not there
-
+# if not openai_key:
+#     raise ValueError("OpenAI API key not found in environment variables.")
 
 def get_response(user_query, chat_history, context):
     template = """

pages/ragChat.py

@@ -6,22 +6,25 @@ from functions.web_chain import vectorize, loadUrlData, get_pdf_text
 import asyncio
 
 
-async def main():
-    sidebar()
+async def add_data():
     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")
+    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'):
+    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:
                     if "retriever" not in st.session_state:
-                        st.session_state.retriever = vectorize(loadUrlData(url), "document")
+                        st.session_state.retriever = vectorize(
+                            loadUrlData(url), "document"
+                        )
                 except Exception as e:
                     st.error(f"Failed to load URL: {e}")
             if uploaded_files:
@@ -37,10 +40,15 @@ async def main():
                 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
+
+
+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?")]
+            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:
@@ -57,23 +65,41 @@ async def main():
             with st.chat_message("Human"):
                 st.write(user_query)
 
-            if 'retriever' in st.session_state:
+            if "retriever" in st.session_state:
                 try:
-                    ragAnswer = await st.session_state.retriever.amax_marginal_relevance_search(user_query, k=4, fetch_k=10)
+                    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)
+                        response = get_response(
+                            user_query, st.session_state.chat_history, context
+                        )
                     if response:
-                        st.session_state.chat_history.append(AIMessage(content=response))
+                        st.session_state.chat_history.append(
+                            AIMessage(content=response)
+                        )
                         with st.chat_message("AI"):
                             st.write(response)
                     else:
-                        st.write("No response received.") 
+                        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):
+        await add_data()
+    else:
+        await rag_chat()
+
 if __name__ == "__main__":
+    sidebar()
     asyncio.run(main())