Update app.py

app.py

@@ -5,7 +5,6 @@ import os
 from PIL import Image
 import io
 import base64
-import time
 import json
 
 app = Flask(__name__)
@@ -32,24 +31,30 @@ def solve():
         img_str = base64.b64encode(buffered.getvalue()).decode()
 
         def generate():
+            mode = 'starting'
             try:
-                response_stream = client.models.generate_content_stream(
+                response = 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?"
+                        "Résous ce problème mathématique étape par étape."
                     ]
                 )
 
-                for chunk in response_stream:
+                for chunk in response:
                     for part in chunk.candidates[0].content.parts:
-                        if hasattr(part, 'thought') and part.thought:
-                            yield f'data: {json.dumps({"thought": part.text})}\n\n'
+                        if part.thought:
+                            if mode != "thinking":
+                                yield f'data: {json.dumps({"mode": "thinking"})}\n\n'
+                                mode = "thinking"
                         else:
-                            yield f'data: {json.dumps({"answer": part.text})}\n\n'
-                    time.sleep(0.1)  # Légère pause pour contrôler le flux
-                
+                            if mode != "answering":
+                                yield f'data: {json.dumps({"mode": "answering"})}\n\n'
+                                mode = "answering"
+                        
+                        yield f'data: {json.dumps({"content": part.text})}\n\n'
+
             except Exception as e:
                 print(f"Error during generation: {e}")
                 yield f'data: {json.dumps({"error": str(e)})}\n\n'