Update app.py

app.py

@@ -52,20 +52,65 @@ def chat_with_openai(query, vectordb, openai_api_key):
     res = chat(messages)
     return res.content
 
+# # Streamlit UI
+# st.title("Document Processing and AI Chat with LangChain")
+
+# # Load vector database
+# zip_file_path = "chroma_db_compressed_.zip"
+# extract_path = "./chroma_db_extracted"
+# vectordb = load_vector_db(zip_file_path, extract_path)
+
+# # Query input
+# query = st.text_input("Enter your query", "Recommend a company to work as a data scientist in the health sector")
+
+# if st.button("Get Answer"):
+#     # Chat with OpenAI
+#     openai_api_key = st.secrets["OPENAI_API_KEY"]
+#     response = chat_with_openai(query, vectordb, openai_api_key)
+#     st.write("Response from AI:")
+#     st.write(response)
+
+
 # Streamlit UI
-st.title("Document Processing and AI Chat with LangChain")
+st.title("Data Roles Company Finder Chatbot")
+st.write("This app helps users find companies hiring for data roles, providing information such as job title, salary estimate, job description, company rating, and more.")
 
 # Load vector database
 zip_file_path = "chroma_db_compressed_.zip"
 extract_path = "./chroma_db_extracted"
 vectordb = load_vector_db(zip_file_path, extract_path)
 
+# Initialize session state for chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        SystemMessage(content="You are a helpful assistant.")
+    ]
+
+# Display chat history
+for message in st.session_state.messages:
+    if isinstance(message, HumanMessage):
+        st.write(f"You: {message.content}")
+    else:
+        st.write(f"AI: {message.content}")
+
 # Query input
 query = st.text_input("Enter your query", "Recommend a company to work as a data scientist in the health sector")
 
-if st.button("Get Answer"):
-    # Chat with OpenAI
-    openai_api_key = st.secrets["OPENAI_API_KEY"]
-    response = chat_with_openai(query, vectordb, openai_api_key)
-    st.write("Response from AI:")
-    st.write(response)
+if st.button("Send"):
+    if query:
+        # Add user query to chat history
+        st.session_state.messages.append(HumanMessage(content=query))
+        
+        # Chat with OpenAI
+        openai_api_key = st.secrets["OPENAI_API_KEY"]
+        response = chat_with_openai(query, vectordb, openai_api_key)
+        
+        # Add AI response to chat history
+        st.session_state.messages.append(SystemMessage(content=response))
+        
+        # Display chat history
+        for message in st.session_state.messages:
+            if isinstance(message, HumanMessage):
+                st.write(f"You: {message.content}")
+            else:
+                st.write(f"AI: {message.content}")