Update app.py

app.py

@@ -101,6 +101,27 @@ def page1():
             metadata={'session_id': st.session_state.session_id}
         )
 
+    # Check if 'run' object has 'status' attribute and display messages
+    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
+            )
+
+            # 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':
+                                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
@@ -120,9 +141,8 @@ def page1():
             time.sleep(1)  # Wait before checking run status
             st.rerun()
 
-    # Check if 'run' object has 'status' attribute
+    # Handle the run's different statuses (running, failed, etc.)
     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 ......")
@@ -130,7 +150,6 @@ def page1():
                 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'):
@@ -141,7 +160,6 @@ 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":
             st.session_state.run = client.beta.threads.runs.retrieve(
                 thread_id=st.session_state.thread.id,
@@ -151,25 +169,6 @@ def page1():
                 time.sleep(3)
                 st.rerun()
 
-        # 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
-            )
-
-            # 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")
 
 
 if __name__ == "__main__":