Update app.py

app.py

@@ -6,10 +6,10 @@ from PIL import Image
 import io
 import base64
 import json
-#4
+
 app = Flask(__name__)
 
-GOOGLE_API_KEY =  os.environ.get("GEMINI_API_KEY")
+GOOGLE_API_KEY = os.environ.get("GEMINI_API_KEY")
 
 client = genai.Client(
     api_key=GOOGLE_API_KEY,
@@ -17,15 +17,11 @@ client = genai.Client(
 
 @app.route('/')
 def index():
-    #return "La plateforme est en maintenance."
     return render_template('index.html')
 
-
 @app.route('/free')
 def indexx():
     return render_template('maj.html')
-    
-
 
 @app.route('/solve', methods=['POST'])
 def solve():
@@ -44,28 +40,42 @@ def solve():
                     model="gemini-2.5-pro-exp-03-25",
                     contents=[
                         {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
-                       """Résous ça en français with rendering latex""" 
-                    
+                        """Résous cet exercice en français avec du LaTeX. 
+                        Si nécessaire, utilise du code Python pour effectuer les calculs complexes.
+                        Présente ta solution de façon claire et espacée."""
                     ],
                     config=types.GenerateContentConfig(
-    thinking_config=types.ThinkingConfig(
-      thinking_budget=8000
-    )
-  )
+                        thinking_config=types.ThinkingConfig(
+                            thinking_budget=8000
+                        ),
+                        tools=[types.Tool(
+                            code_execution=types.ToolCodeExecution()
+                        )]
+                    )
                 )
-#Resous cette exercice. ça doit être bien présentable et espacé. je veux un jolie rendu
+
                 for chunk in response:
                     for part in chunk.candidates[0].content.parts:
-                        if part.thought:
+                        if hasattr(part, 'thought') and part.thought:
                             if mode != "thinking":
                                 yield f'data: {json.dumps({"mode": "thinking"})}\n\n'
                                 mode = "thinking"
+                        elif hasattr(part, 'executable_code') and part.executable_code:
+                            if mode != "executing_code":
+                                yield f'data: {json.dumps({"mode": "executing_code"})}\n\n'
+                                mode = "executing_code"
+                            yield f'data: {json.dumps({"content": f"```python\n{part.executable_code.code}\n```"})}\n\n'
+                        elif hasattr(part, 'code_execution_result') and part.code_execution_result:
+                            if mode != "code_result":
+                                yield f'data: {json.dumps({"mode": "code_result"})}\n\n'
+                                mode = "code_result"
+                            yield f'data: {json.dumps({"content": f"Résultat d'exécution:\n```\n{part.code_execution_result.output}\n```"})}\n\n'
                         else:
                             if mode != "answering":
                                 yield f'data: {json.dumps({"mode": "answering"})}\n\n'
                                 mode = "answering"
-                        
-                        yield f'data: {json.dumps({"content": part.text})}\n\n'
+                            if hasattr(part, 'text') and part.text:
+                                yield f'data: {json.dumps({"content": part.text})}\n\n'
 
             except Exception as e:
                 print(f"Error during generation: {e}")
@@ -83,8 +93,6 @@ def solve():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-
-
 @app.route('/solved', methods=['POST'])
 def solved():
     try:
@@ -102,22 +110,39 @@ def solved():
                     model="gemini-2.5-flash-preview-04-17",
                     contents=[
                         {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
-                        """ Résous ça en français with rendering latex"""
-                    ]
+                        """Résous cet exercice en français avec du LaTeX.
+                        Si nécessaire, utilise du code Python pour effectuer les calculs complexes.
+                        Présente ta solution de façon claire et espacée."""
+                    ],
+                    config=types.GenerateContentConfig(
+                        tools=[types.Tool(
+                            code_execution=types.ToolCodeExecution()
+                        )]
+                    )
                 )
-#Resous cette exercice. ça doit être bien présentable et espacé. je veux un jolie rendu
+
                 for chunk in response:
                     for part in chunk.candidates[0].content.parts:
-                        if part.thought:
+                        if hasattr(part, 'thought') and part.thought:
                             if mode != "thinking":
                                 yield f'data: {json.dumps({"mode": "thinking"})}\n\n'
                                 mode = "thinking"
+                        elif hasattr(part, 'executable_code') and part.executable_code:
+                            if mode != "executing_code":
+                                yield f'data: {json.dumps({"mode": "executing_code"})}\n\n'
+                                mode = "executing_code"
+                            yield f'data: {json.dumps({"content": f"```python\n{part.executable_code.code}\n```"})}\n\n'
+                        elif hasattr(part, 'code_execution_result') and part.code_execution_result:
+                            if mode != "code_result":
+                                yield f'data: {json.dumps({"mode": "code_result"})}\n\n'
+                                mode = "code_result"
+                            yield f'data: {json.dumps({"content": f"Résultat d'exécution:\n```\n{part.code_execution_result.output}\n```"})}\n\n'
                         else:
                             if mode != "answering":
                                 yield f'data: {json.dumps({"mode": "answering"})}\n\n'
                                 mode = "answering"
-                        
-                        yield f'data: {json.dumps({"content": part.text})}\n\n'
+                            if hasattr(part, 'text') and part.text:
+                                yield f'data: {json.dumps({"content": part.text})}\n\n'
 
             except Exception as e:
                 print(f"Error during generation: {e}")
@@ -134,9 +159,6 @@ def solved():
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
-        
-
-
 
 if __name__ == '__main__':
     app.run(debug=True)