Update app.py

app.py

@@ -98,15 +98,8 @@ def chat():
         text = ""
         
         for item in content:
-            if item.get('type') == 'image':
-                if 'image_url' in item:
-                    image = process_image(item['image_url']['url'])
-                elif 'image' in item:
-                    image = process_image(item['image'])
-                else:
-                    return jsonify({
-                        "error": "Image content must have 'image_url' or 'image' field"
-                    }), 400
+            if item.get('type') == 'image_url' and 'image_url' in item and 'url' in item['image_url']:
+                image = process_image(item['image_url']['url'])
             elif item.get('type') == 'text':
                 text = item.get('text', '')
         
@@ -198,8 +191,5 @@ def upload_image():
         }), 500
 
 if __name__ == '__main__':
-    # Load model on startup
     load_model()
-    
-    # Run the Flask app on port 7860 for Hugging Face Spaces
     app.run(host='0.0.0.0', port=7860, debug=False)