Update app.py

app.py

@@ -101,33 +101,11 @@ def page1():
             metadata={'session_id': st.session_state.session_id}
         )
 
-    # If the run is completed, display the messages
-    elif 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
-        )
-
-        end_time = time.time()  # Stop the timer
-        duration = end_time - start_time  # Calculate duration
-        st.write(f"Response time: {duration:.2f} seconds")
-
-        # Display messages
-        for message in reversed(st.session_state.messages.data):
-            if message.role in ["user", "assistant"]:
-                with st.chat_message(message.role):
-                    for content_part in message.content:
-                        if content_part.type == 'text':  # For text responses
-                            message_text = content_part.text.value
-                            st.markdown(message_text)
-                        elif hasattr(content_part, 'image_file') and content_part.image_file:  
-                            image = handle_and_display_image(content_part.image_file.file_id)
-                            if image:
-                                st.image(image, caption="Generated Image")
-
-    start_time = time.time()  # Start the timer
+    # Initialize start_time to None
+    start_time = None
 
     if prompt := st.chat_input("Wie kann ich dir helfen?"):
+        start_time = time.time()  # Start the timer when the user submits a query
         with st.chat_message('user'):
             st.write(prompt)
 
@@ -168,18 +146,29 @@ def page1():
                 else:
                     st.error("FAILED: The OpenAI API is currently processing too many requests. Please try again later ......")
 
-        # Handle any status that is not 'completed'
-        elif st.session_state.run.status != "completed":
-            # Attempt to retrieve the run again
-            st.session_state.run = client.beta.threads.runs.retrieve(
-                thread_id=st.session_state.thread.id,
-                run_id=st.session_state.run.id,
+        # When the run is completed
+        elif hasattr(st.session_state.run, 'status') and st.session_state.run.status == "completed":
+            end_time = time.time()  # Stop the timer when the response is received
+            if start_time:
+                duration = end_time - start_time  # Calculate duration
+                st.write(f"Response time: {duration:.2f} seconds")
+
+            # Retrieve the list of messages and display them
+            st.session_state.messages = client.beta.threads.messages.list(
+                thread_id=st.session_state.thread.id
             )
-            if st.session_state.retry_error < 3:
-                time.sleep(3)
-                st.rerun()
-          
 
+            for message in reversed(st.session_state.messages.data):
+                if message.role in ["user", "assistant"]:
+                    with st.chat_message(message.role):
+                        for content_part in message.content:
+                            if content_part.type == 'text':  # For text responses
+                                message_text = content_part.text.value
+                                st.markdown(message_text)
+                            elif hasattr(content_part, 'image_file') and content_part.image_file:  
+                                image = handle_and_display_image(content_part.image_file.file_id)
+                                if image:
+                                    st.image(image, caption="Generated Image")