Update app.py

app.py

@@ -111,35 +111,26 @@ def page1():
         st.session_state.chat_history.append(("User", question))
         process_question(question)
 
-    # Modified function to process the question
+# Function to process the question
     def process_question(question):
-        # Add new message to the thread
-        client.beta.threads.messages.create(
+        # Add message to the thread
+        st.session_state.messages = 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
+        # Do a 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,
         )
 
-        # Check if the run is completed and display the last assistant message
-        if hasattr(st.session_state.run, 'status') and 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
-            )
-
-            # Find the last assistant message
-            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))
+        # Wait and rerun logic
+        if st.session_state.retry_error < 3:
+            time.sleep(1)  # Wait 1 second before checking run status
+            st.rerun()
 
-    # [Rest of your existing code in page1 remains unchanged]
 
     # Display chat history in Streamlit chat format
     for role, message in st.session_state.chat_history: