Update app.py

app.py

@@ -43,15 +43,11 @@ def chat_with_assistant(message, history):
             messages = client.beta.threads.messages.list(thread_id=thread.id)
             content_block = messages.data[0].content[0]
             
-            # Print the attributes of content_block for debugging
-            print("Content block attributes:", dir(content_block))
-            
+            # Check if the response is text or an image
             if hasattr(content_block, 'text'):
                 assistant_response = content_block.text.value
-            elif hasattr(content_block, 'image'):
-                assistant_response = f"<img src='{content_block.image.url}' alt='Image' />"
-            elif hasattr(content_block, 'graph'):
-                assistant_response = f"<div>{content_block.graph.url}</div>"  # Assuming graph content
+            elif hasattr(content_block, 'image_file'):
+                assistant_response = f"<img src='{content_block.image_file.url}' alt='Image' />"
             else:
                 assistant_response = "[Unknown content type]"
             break