Update app.py

app.py

@@ -19,13 +19,16 @@ else:
 
 ASSISTANT_ID = "asst_Iu6cCU4CyugmAD2yPgktyivH"
 
-# Initialize session state for chat history
+# Initialize session state
 if "messages" not in st.session_state:
-    st.session_state["messages"] = []
+    st.session_state.messages = []
+if "thread_id" not in st.session_state:
+    st.session_state.thread_id = None
 
-# Clear chat button above chat input
+# Clear chat button
 if st.button("Clear Chat", use_container_width=True):
     st.session_state.messages = []
+    st.session_state.thread_id = None
     st.rerun()
 
 # Display chat history
@@ -39,11 +42,14 @@ if prompt := st.chat_input():
     st.chat_message("user").write(prompt)
 
     try:
-        # Create a new thread for conversation
-        thread = client.beta.threads.create()
-        thread_id = thread.id
+        # Create a new thread only if one doesn't exist yet
+        if st.session_state.thread_id is None:
+            thread = client.beta.threads.create()
+            st.session_state.thread_id = thread.id
+
+        thread_id = st.session_state.thread_id
 
-        # Send user message to OpenAI API
+        # Send user message to OpenAI API in the existing thread
         client.beta.threads.messages.create(
             thread_id=thread_id,
             role="user",
@@ -65,13 +71,17 @@ if prompt := st.chat_input():
 
         # Retrieve assistant response
         messages = client.beta.threads.messages.list(thread_id=thread_id)
-        assistant_message = messages.data[0].content[0].text.value
+        # Get the most recent assistant message
+        for message in reversed(messages.data):
+            if message.role == "assistant":
+                assistant_message = message.content[0].text.value
+                break
 
         # Display assistant's response
         st.chat_message("assistant").write(assistant_message)
 
         # Store in session state
         st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-    
+
     except Exception as e:
         st.error(f"Error: {str(e)}")