Update app.py

app.py

@@ -1,4 +1,4 @@
- # Importing required packages
+  # Importing required packages
 import streamlit as st
 import openai
 import uuid
@@ -148,7 +148,7 @@ def page1():
             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 the assistant's response
-                st.session_state.chat_history = [("User", question), ("Assistant", last_assistant_message.content[0].text.value)]
+                st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
     
 
     # Example query buttons
@@ -182,19 +182,6 @@ def page1():
         )
 
 
-    # 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))
-
     # Display chat history in Streamlit chat format
     for role, message in st.session_state.chat_history:
         if role == "User":