Update app.py

app.py

@@ -6,15 +6,14 @@ from PIL import Image
 import io
 import base64
 import time
+import json
 
 app = Flask(__name__)
 
-# Remplacez par votre clé API réelle
 GOOGLE_API_KEY = "AIzaSyC_zxN9IHjEAxIoshWPzMfgb9qwMsu5t5Y"
 
 client = genai.Client(
     api_key=GOOGLE_API_KEY,
-    # Use `v1alpha` so you can see the `thought` flag.
     http_options={'api_version': 'v1alpha'},
 )
 
@@ -32,29 +31,37 @@ def solve():
         img.save(buffered, format="PNG")
         img_str = base64.b64encode(buffered.getvalue()).decode()
 
-        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 ce problème?"
-            ]
-        )
-
         def generate():
             try:
+                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 ce problème?"
+                    ]
+                )
+
                 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"
+                        if hasattr(part, 'thought') and 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
+                            yield f'data: {json.dumps({"answer": part.text})}\n\n'
+                    time.sleep(0.1)  # Légère pause pour contrôler le flux
+                
             except Exception as e:
                 print(f"Error during generation: {e}")
-                yield f"data: {{ \"error\": \"{e}\" }}\n\n"
+                yield f'data: {json.dumps({"error": str(e)})}\n\n'
 
-        return Response(generate(), mimetype='text/event-stream')
+        return Response(
+            stream_with_context(generate()),
+            mimetype='text/event-stream',
+            headers={
+                'Cache-Control': 'no-cache',
+                'X-Accel-Buffering': 'no'
+            }
+        )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500