Update app.py

app.py

@@ -110,26 +110,57 @@ def page1():
         # Add question to chat history
         st.session_state.chat_history.append(("User", question))
         process_question(question)
+        display_chat_history()
 
     # Function to process the question
     def process_question(question):
+        # Add user question to chat history
+        st.session_state.chat_history.append(("User", question))
+    
         # 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
+        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
+            )
+    
+            # 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))
+    
+                # Display the chat history
+                display_chat_history()
+    
+    def display_chat_history():
+        for role, message in st.session_state.chat_history:
+            if role == "User":
+                with st.chat_message("User"):
+                    st.markdown(message)
+            elif role == "Assistant":
+                with st.chat_message("Assistant"):
+                    st.markdown(message)
 
     # Example query buttons
     col3, col4 = st.columns(2)  # Create two columns for buttons