Update app.py

app.py

@@ -1,4 +1,4 @@
- # Importing required packages
+  # Importing required packages
 import streamlit as st
 import openai
 import uuid
@@ -94,7 +94,7 @@ def page1():
 
     with col1:
         st.title('Welcome to MediMetrics!')
-        st.subheader('powered by :orange_heart:[BinDoc GmbH] ')
+        st.subheader('powered by :orange[BinDoc GmbH] ')
 
     with col2:
         # Load and display the image in the right column
@@ -105,41 +105,31 @@ def page1():
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
 
-    # Define flags in the session state for processing control
-    if 'processing' not in st.session_state:
-        st.session_state.processing = False
-
     # Function to handle question submission
     def submit_question(question):
-        if not st.session_state.processing:
-            st.session_state.processing = True
-            st.session_state.chat_history.append(("User", question))
-            process_question(question)
-            st.session_state.processing = False
-            st.experimental_rerun()
+        # Add question to chat history
+        st.session_state.chat_history.append(("User", question))
+        process_question(question)
 
     # Function to process the question
     def process_question(question):
-        # Clear the previous run status and messages
+        # Clear the previous run status and chat history
         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)]
-        
+        st.session_state.chat_history = [("User", question)]  # Start with the current 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
         while not hasattr(st.session_state.run, 'status') or st.session_state.run.status not in ["completed", "failed"]:
             time.sleep(1)
@@ -147,30 +137,20 @@ def page1():
                 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
+    
+            # Get 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:
                 # Update chat history with the assistant's response
                 st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
-        
-    # 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
 
@@ -201,6 +181,7 @@ def page1():
             metadata={'session_id': st.session_state.session_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