Update app.py

app.py

@@ -1,6 +1,5 @@
 from flask import Flask, render_template, request, send_file, jsonify
 from google import genai
-from google.genai import types
 from PIL import Image
 import subprocess
 import tempfile
@@ -11,7 +10,7 @@ import base64
 
 app = Flask(__name__)
 
-# --- Constantes ---
+# --- Convstantes ---
 MODEL_SINGLE_GENERATION = "gemini-2.5-pro-exp-03-25"  # Modèle avec "thinking" pour la génération complète
 LATEX_MENTION = r"\vspace{1cm}\noindent\textit{Ce devoir a été généré par Mariam AI. \url{https://mariam-241.vercel.app}}"
 
@@ -79,44 +78,11 @@ def clean_latex(raw_latex_text):
 
     return cleaned.strip()  # Retourne le résultat nettoyé
 
-def extract_content_from_response(response):
-    """
-    Extrait le contenu LaTeX et le raisonnement (thought) de la réponse Gemini.
-    Prend également en compte les résultats d'exécution de code.
-    """
-    latex_content_raw = None
-    thinking_content = []
-    code_execution_results = []
-
-    if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
-        for part in response.candidates[0].content.parts:
-            # Extraire le thinking
-            if getattr(part, 'thought', None):
-                thinking_content.append(part.text)
-            
-            # Extraire le code et les résultats d'exécution
-            if getattr(part, 'executable_code', None):
-                thinking_content.append(f"Code généré:\n{part.executable_code.code}")
-            
-            if getattr(part, 'code_execution_result', None):
-                code_execution_results.append(part.code_execution_result.output)
-                thinking_content.append(f"Résultat d'exécution:\n{part.code_execution_result.output}")
-            
-            # Extraire le texte (contenu LaTeX)
-            if getattr(part, 'text', None) and not latex_content_raw:
-                latex_content_raw = part.text
-
-    # Combiner les éléments de thinking_content
-    thinking_content = "\n\n".join(thinking_content)
-    
-    return latex_content_raw, thinking_content, code_execution_results
-
 # --- Fonction Principale (API GenAI) ---
 
 def generate_complete_latex(client, image_bytes):
     """
     Génère une solution complète en LaTeX à partir de l'image en une seule étape.
-    Utilise l'exécution de code pour améliorer les réponses mathématiques.
     """
     try:
         # Convertir les bytes en image PIL
@@ -134,8 +100,7 @@ Agis en tant qu'expert en mathématiques et tuteur pédagogue. Ton objectif est
 # TÂCHE PRINCIPALE
 1.  Analyse l'image pour comprendre parfaitement l'énoncé de l'exercice.
 2.  Résous l'exercice de manière rigoureuse, étape par étape.
-3.  Utilise l'exécution de code Python quand nécessaire pour vérifier tes calculs, résoudre des équations ou générer des visualisations.
-4.  Rédige la solution complète directement en code LaTeX, en respectant **toutes** les spécifications ci-dessous.
+3.  Rédige la solution complète directement en code LaTeX, en respectant **toutes** les spécifications ci-dessous.
 
 # SPÉCIFICATIONS TECHNIQUES DU CODE LATEX
 1.  **Structure du Document:** Commence **strictement** par `\\documentclass{{article}}` et se termine **strictement** par `\\end{{document}}`.
@@ -153,11 +118,6 @@ Agis en tant qu'expert en mathématiques et tuteur pédagogue. Ton objectif est
 5.  **Mention Obligatoire:** Insère la ligne suivante **exactement** telle quelle, juste avant la ligne `\\end{{document}}`:
     {LATEX_MENTION}
 
-# UTILISATION DU CODE PYTHON
-1. Utilise Python pour vérifier tes calculs, résoudre des équations complexes ou générer des visualisations.
-2. Lorsque tu utilises l'exécution de code, assure-toi que les résultats sont correctement intégrés dans ta solution LaTeX.
-3. N'affiche pas le code Python lui-même dans le document LaTeX final, utilise-le uniquement comme outil pour améliorer la qualité de ta correction.
-
 # PROCESSUS INTERNE RECOMMANDÉ (Pour l'IA)
 1.  **Analyse Approfondie:** Décompose le problème en sous-étapes logiques.
 2.  **Résolution Étape par Étape:** Effectue la résolution mathématique complète en interne.
@@ -169,11 +129,23 @@ Agis en tant qu'expert en mathématiques et tuteur pédagogue. Ton objectif est
             contents=[image, prompt],
             config=genai.types.GenerateContentConfig(
                 temperature=0.3,
-                tools=[types.Tool(code_execution=types.ToolCodeExecution)]
-            )
+            ),
+            tools=[types.Tool(
+            code_execution=types.ToolCodeExecution
+            )]
+            
         )
 
-        latex_content_raw, thinking_content, code_execution_results = extract_content_from_response(response)
+        latex_content_raw = None
+        thinking_content = None
+
+        # Extrait le contenu LaTeX et le raisonnement (thought)
+        if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
+            for part in response.candidates[0].content.parts:
+                if getattr(part, 'thought', None):
+                    thinking_content = part.text
+                else:
+                    latex_content_raw = part.text
 
         if latex_content_raw:
             latex_content_cleaned = clean_latex(latex_content_raw)