Update app.py

app.py

@@ -113,23 +113,42 @@ def page1():
 
     # Function to process the question
     def process_question(question):
+        # Clear the previous run status
+        st.session_state.run = None
+    
         # Add message to the thread
-        st.session_state.messages = client.beta.threads.messages.create(
+        client.beta.threads.messages.create(
             thread_id=st.session_state.thread.id,
             role="user",
             content=question
         )
-
-        # Do a run to process the messages in the thread
+    
+        # 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 and rerun logic
-        if st.session_state.retry_error < 3:
-            time.sleep(1)  # Wait 1 second before checking run status
-            st.rerun()
+    
+        # Wait for the run to complete and then refresh the app
+        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
+            last_assistant_message = next((message for message in reversed(st.session_state.messages.data) if message.role == "assistant"), None)
+            if last_assistant_message:
+                # Add assistant's response to chat history
+                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