Update app.py

app.py

@@ -58,13 +58,23 @@ elif hasattr(st.session_state.run, 'status') and st.session_state.run.status ==
         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:
-                    message_text = content_part.text.value
-                    st.markdown(message_text)
+                    # Check if the content part is a text message
+                    if hasattr(content_part, 'text'):
+                        message_text = content_part.text.value
+                        st.markdown(message_text)
+                    # Check if the content part is an image file
+                    elif hasattr(content_part, 'image'):
+                        # Handle the image file. For example, display it in Streamlit.
+                        # You'll need to write the appropriate code to handle the image based on your requirements.
+                        image_url = content_part.image.url  # Example: Get the image URL
+                        st.image(image_url, caption="Generated Image")
+
 
 if prompt := st.chat_input("Wie kann ich dir helfen?"):
     with st.chat_message('user'):