Update app.py

app.py

@@ -52,24 +52,6 @@ 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("Data Roles Company Finder Chatbot")
@@ -82,35 +64,93 @@ 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.")
-    ]
+    st.session_state.messages = []
 
 # 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}")
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 # Query input
-query = st.text_input("Enter your query", "Recommend a company to work as a data scientist in the health sector")
+query = st.text_input("Enter your query", "")
 
 if st.button("Send"):
     if query:
         # Add user query to chat history
-        st.session_state.messages.append(HumanMessage(content=query))
+        st.session_state.messages.append({"role": "user", "content": query})
+        with st.chat_message("user"):
+            st.markdown(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))
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        with st.chat_message("assistant"):
+            st.markdown(response)
+
+
+            
+# # 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", "List three companies where I can work as a business analyst with their location and salary")
+
+# 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("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", "List three companies where I can work as a business analyst with their location and salary")
+
+# 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}")
+#         # 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}")