Update app.py

app.py

@@ -2,8 +2,7 @@ from flask import Flask, render_template, request, jsonify
 import google.generativeai as genai
 import os
 from PIL import Image
-import io
-import base64
+import tempfile
 
 app = Flask(__name__)
 
@@ -39,24 +38,27 @@ def home():
 
 @app.route('/generate', methods=['POST'])
 def generate():
-    try:
-        # Récupérer l'image depuis la requête
-        image_data = request.json.get('image')
-        if not image_data:
-            return jsonify({"result": "djo"})
+    if 'image' not in request.files:
+        return jsonify({'error': 'No image uploaded'}), 400
+    
+    image_file = request.files['image']
+    
+    # Sauvegarder temporairement l'image
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        image_file.save(temp_file.name)
+        try:
+            image = Image.open(temp_file.name)
         
-        # Convertir l'image base64 en image PIL
-        image_data = image_data.split(',')[1]
-        image_bytes = base64.b64decode(image_data)
-        image = Image.open(io.BytesIO(image_bytes))
-        
-        # Générer le contenu
-        response = model.generate_content([mm, image])
-        result = response.text
-        
-        return jsonify({"result": result})
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+            # Générer le contenu
+            response = model.generate_content([mm, image])
+            result = response.text
+            
+            return jsonify({"result": result})
+        except Exception as e:
+             return jsonify({'error': str(e)}), 500
+        finally:
+        # Nettoyer le fichier temporaire
+            os.unlink(temp_file.name)
 
 if __name__ == '__main__':
     app.run(debug=True)