Update app.py

app.py

@@ -94,7 +94,7 @@ def page1():
 
     with col1:
         st.title('Welcome to MediMetrics!')
-        st.subheader('powered by :orange[BinDoc GmbH] ')
+        st.subheader('powered by :orange_heart:[BinDoc GmbH] ')
 
     with col2:
         # Load and display the image in the right column
@@ -113,43 +113,57 @@ def page1():
 
     # Function to process the question
     def process_question(question):
-        # Clear the previous run status
+        # Clear the previous run status and messages
         st.session_state.run = None
-    
+        st.session_state.messages = []
+        
+        # Start with the current user question in chat history
+        st.session_state.chat_history = [("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
+            
+            # 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
+                # Update chat history with the assistant's response
                 st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
+        
+        # Display the updated 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