Update app.py

app.py

@@ -4,6 +4,8 @@ import os
 from PIL import Image
 import tempfile
 import io
+import uuid
+import time
 
 app = Flask(__name__)
 
@@ -25,37 +27,60 @@ model = genai.GenerativeModel(
     safety_settings=safety_settings
 )
 
+# Dictionnaire pour stocker les réponses en cours de génération
+pending_responses = {}
+
 @app.route('/')
 def home():
     return render_template('index.html')
 
-@app.route('/generate', methods=['POST','GET'])
+@app.route('/generate', methods=['POST'])
 def generate():
     if 'image' not in request.files:
         return jsonify({'error': 'No image uploaded'}), 400
 
     image_file = request.files['image']
+    request_id = str(uuid.uuid4())  # Générer un identifiant unique
+
+    # Sauvegarder temporairement l'image
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        image_file.save(temp_file.name)
+        try:
+            image = Image.open(temp_file.name)
+            # Convertir l'image en bytes pour le streaming
+            img_byte_arr = io.BytesIO()
+            image.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
 
-    # Fonction pour générer des chunks de réponse
+            # Stocker la tâche de génération dans le dictionnaire
+            pending_responses[request_id] = {
+                'status': 'processing',
+                'response': model.generate_content([mm, {"mime_type": "image/png", "data": img_byte_arr}], stream=True)
+            }
+
+            return jsonify({'request_id': request_id})
+        except Exception as e:
+            return jsonify({'error': str(e)}), 500
+        finally:
+            # Nettoyer le fichier temporaire
+            os.unlink(temp_file.name)
+    
+@app.route('/stream/<request_id>')
+def stream(request_id):
     def generate_stream():
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
-            image_file.save(temp_file.name)
+        while request_id in pending_responses and pending_responses[request_id]['status'] == 'processing':
             try:
-                image = Image.open(temp_file.name)
-                # Convertir l'image en bytes pour le streaming
-                img_byte_arr = io.BytesIO()
-                image.save(img_byte_arr, format='PNG')
-                img_byte_arr = img_byte_arr.getvalue()
-
-                # Générer le contenu en streaming
-                response = model.generate_content([mm, {"mime_type": "image/png", "data": img_byte_arr}], stream=True)
-                for chunk in response:
-                    yield f"data: {chunk.text}\n\n"
+                chunk = next(pending_responses[request_id]['response'])
+                yield f"data: {chunk.text}\n\n"
+            except StopIteration:
+                pending_responses[request_id]['status'] = 'completed'
             except Exception as e:
                 yield f"data: Error: {str(e)}\n\n"
-            finally:
-                # Nettoyer le fichier temporaire
-                os.unlink(temp_file.name)
+                pending_responses[request_id]['status'] = 'error'
+            time.sleep(0.1)  # Attendre un peu avant de vérifier à nouveau
+
+        if request_id in pending_responses:
+            del pending_responses[request_id]
 
     return Response(generate_stream(), mimetype='text/event-stream')