Update app.py

app.py

@@ -55,13 +55,26 @@ 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)
+                    if content_part.type == 'text':  # For text responses
+                        message_text = content_part.text.value
+                        st.markdown(message_text)
+                    elif content_part.type == 'image_url':  # For image URLs
+                        image_url = content_part.image_url  # Adjust based on actual response structure
+                        st.image(image_url)
+                    elif content_part.type == 'base64_image':  # For base64-encoded images
+                        base64_data = content_part.base64_image  # Adjust based on actual response structure
+                        image_data = base64.b64decode(base64_data)
+                        image = Image.open(io.BytesIO(image_data))
+                        st.image(image)
+                   
+    
+
 
 if prompt := st.chat_input("Wie kann ich dir helfen?"):
     with st.chat_message('user'):