Update app.py

app.py

@@ -101,31 +101,31 @@ 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
+        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:
+            with st.chat_message('assistant'):
+                for content_part in last_assistant_message.content:
+                    if content_part.type == 'text':
+                        st.markdown(content_part.text.value)
+                    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")
+
     
-    if "run" in st.session_state and hasattr(st.session_state.run, 'status'):
-        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
-            )
 
-            # Filter for assistant messages
-            assistant_messages = [m for m in st.session_state.messages.data if m.role == "assistant"]
-
-            # Get the last assistant message
-            if assistant_messages:
-                final_assistant_message = assistant_messages[-1]
-                with st.chat_message('assistant'):
-                    for content_part in final_assistant_message.content:
-                        if content_part.type == 'text':
-                            st.markdown(content_part.text.value)
-                        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")
-    # User input and submission
     prompt = st.text_input("Wie kann ich dir helfen?")
     if st.button("Submit"):
+
+
         # Add message to the thread
         st.session_state.messages = client.beta.threads.messages.create(
             thread_id=st.session_state.thread.id,
@@ -138,30 +138,34 @@ def page1():
             thread_id=st.session_state.thread.id,
             assistant_id=st.session_state.assistant.id,
         )
-
         if st.session_state.retry_error < 3:
-            time.sleep(1)  # Wait before checking run status
+            time.sleep(1) # Wait 1 second before checking run status
             st.rerun()
-    
-        # Handle the run's different statuses (running, failed, etc.)
+
+    # Check if 'run' object has 'status' attribute
     if hasattr(st.session_state.run, 'status'):
+        # Handle the 'running' status
         if st.session_state.run.status == "running":
             with st.chat_message('assistant'):
                 st.write("Thinking ......")
-            time.sleep(15)  # Increased wait time to 5 seconds
-            st.rerun()
-    
+            if st.session_state.retry_error < 3:
+                time.sleep(1)  # Short delay to prevent immediate rerun
+                st.rerun()
+
+        # Handle the 'failed' status
         elif st.session_state.run.status == "failed":
             st.session_state.retry_error += 1
             with st.chat_message('assistant'):
                 if st.session_state.retry_error < 3:
-                    time.sleep(5)  # Increased retry delay to 5 seconds
+                    st.write("Run failed, retrying ......")
+                    time.sleep(3)  # Longer delay before retrying
                     st.rerun()
                 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,
@@ -170,8 +174,5 @@ def page1():
                 time.sleep(3)
                 st.rerun()
 
-
-
-
 if __name__ == "__main__":
-    main()
+    main()