Update app.py

app.py

@@ -1,15 +1,16 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, Response, stream_with_context
 from google import genai
 from google.genai import types
 import os
 from PIL import Image
 import io
 import base64
+import time
 
 app = Flask(__name__)
 
 # Remplacez par votre clé API réelle
-GOOGLE_API_KEY = "AIzaSyC_zxN9IHjEAxIoshWPzMfgb9qwMsu5t5Y"
+GOOGLE_API_KEY = "YOUR_API_KEY"
 
 client = genai.Client(
     api_key=GOOGLE_API_KEY,
@@ -24,34 +25,36 @@ def index():
 @app.route('/solve', methods=['POST'])
 def solve():
     try:
-        image_data = request.files['image'].read()  # Lire les données binaires de l'image
+        image_data = request.files['image'].read()
         img = Image.open(io.BytesIO(image_data))
 
-        # Convertir l'image en base64 pour l'envoyer à l'API Gemini
         buffered = io.BytesIO()
         img.save(buffered, format="PNG")
         img_str = base64.b64encode(buffered.getvalue()).decode()
 
-        # Utiliser l'API Gemini pour générer la réponse
-        response = client.models.generate_content(
+        response_stream = client.models.generate_content_stream(
             model="gemini-2.0-flash-thinking-exp-01-21",
             config={'thinking_config': {'include_thoughts': True}},
             contents=[
                 {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
-                "résous cette exercice"
+                "Résous ce problème?"
             ]
         )
 
-        # Extraire les pensées et la réponse
-        thoughts = ""
-        answer = ""
-        for part in response.candidates[0].content.parts:
-            if part.thought:
-                thoughts += part.text + "\n"
-            else:
-                answer += part.text + "\n"
-
-        return jsonify({'thoughts': thoughts, 'answer': answer})
+        def generate():
+            try:
+                for chunk in response_stream:
+                    for part in chunk.candidates[0].content.parts:
+                        if part.thought:
+                            yield f"data: {json.dumps({'thought': part.text})}\n\n"
+                        else:
+                            yield f"data: {json.dumps({'answer': part.text})}\n\n"
+                        time.sleep(0.05)  # Contrôler la vitesse de streaming
+            except Exception as e:
+                print(f"Error during generation: {e}")
+                yield f"data: {{ \"error\": \"{e}\" }}\n\n"
+
+        return Response(generate(), mimetype='text/event-stream')
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500