Update app.py

app.py

@@ -52,10 +52,13 @@ def generate():
             image.save(img_byte_arr, format='PNG')
             img_byte_arr = img_byte_arr.getvalue()
 
-            # Stocker la tâche de génération dans le dictionnaire
+            # Générer le contenu en streaming et obtenir l'itérateur
+            response_iterator = model.generate_content([mm, {"mime_type": "image/png", "data": img_byte_arr}], stream=True)
+
+            # Stocker l'itérateur dans le dictionnaire
             pending_responses[request_id] = {
                 'status': 'processing',
-                'response': model.generate_content([mm, {"mime_type": "image/png", "data": img_byte_arr}], stream=True)
+                'iterator': response_iterator
             }
 
             return jsonify({'request_id': request_id})
@@ -64,13 +67,14 @@ def generate():
         finally:
             # Nettoyer le fichier temporaire
             os.unlink(temp_file.name)
-    
+
 @app.route('/stream/<request_id>')
 def stream(request_id):
     def generate_stream():
         while request_id in pending_responses and pending_responses[request_id]['status'] == 'processing':
             try:
-                chunk = next(pending_responses[request_id]['response'])
+                # Utiliser l'itérateur stocké pour obtenir le prochain chunk
+                chunk = next(pending_responses[request_id]['iterator'])
                 yield f"data: {chunk.text}\n\n"
             except StopIteration:
                 pending_responses[request_id]['status'] = 'completed'