Update app.py

app.py

@@ -113,44 +113,43 @@ def page1():
 
     # Function to process the question
     def process_question(question):
-        # Clear the previous run status
+        # Clear the previous run status and chat history
         st.session_state.run = None
-
-
+        st.session_state.chat_history = [("User", question)]  # Start with the current user question
+    
         # Add message to the thread
         client.beta.threads.messages.create(
             thread_id=st.session_state.thread.id,
             role="user",
             content=question
         )
-
+    
         # Create a new run to process the messages in the thread
         st.session_state.run = client.beta.threads.runs.create(
             thread_id=st.session_state.thread.id,
             assistant_id=st.session_state.assistant.id,
         )
-
-        # Wait for the run to complete and then refresh the app
+    
+        # Wait for the run to complete
         while not hasattr(st.session_state.run, 'status') or st.session_state.run.status not in ["completed", "failed"]:
             time.sleep(1)
             st.session_state.run = client.beta.threads.runs.retrieve(
                 thread_id=st.session_state.thread.id,
                 run_id=st.session_state.run.id,
             )
-
+    
         if st.session_state.run.status == "completed":
             # Retrieve the list of messages
             st.session_state.messages = client.beta.threads.messages.list(
                 thread_id=st.session_state.thread.id
             )
-
-            # Display the latest message from the assistant
+    
+            # Get the latest message from the assistant
             last_assistant_message = next((message for message in reversed(st.session_state.messages.data) if message.role == "assistant"), None)
             if last_assistant_message:
-
-                # Update chat history with only the latest assistant's response
-                st.session_state.chat_history = [("User", question), ("Assistant", last_assistant_message.content[0].text.value)]
-
+                # Update chat history with the assistant's response
+                st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
+    
 
     # Example query buttons
     col3, col4 = st.columns(2)  # Create two columns for buttons