Update app.py

app.py

@@ -71,24 +71,38 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-# Query input
-query = st.text_input("Enter your query", "")
-
-if st.button("Send"):
-    if query:
-        # Add user query to chat history
-        st.session_state.messages.append({"role": "user", "content": query})
-        with st.chat_message("user"):
-            st.markdown(query)
-        
-        # Chat with OpenAI
+
+# User input
+if prompt := st.chat_input("Enter your query"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
         openai_api_key = st.secrets["OPENAI_API_KEY"]
-        response = chat_with_openai(query, vectordb, openai_api_key)
+        response = chat_with_openai(prompt, vectordb, openai_api_key)
+        st.markdown(response)
+        
+    st.session_state.messages.append({"role": "assistant", "content": response})
         
-        # Add AI response to chat history
-        st.session_state.messages.append({"role": "assistant", "content": response})
-        with st.chat_message("assistant"):
-            st.markdown(response)
+# # Query input
+# query = st.text_input("Enter your query", "")
+
+# if st.button("Send"):
+#     if query:
+#         # Add user query to chat history
+#         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({"role": "assistant", "content": response})
+#         with st.chat_message("assistant"):
+#             st.markdown(response)