Update app.py

app.py

@@ -1,5 +1,6 @@
 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
@@ -10,7 +11,7 @@ import base64
 
 app = Flask(__name__)
 
-# --- Convstantes ---
+# --- Constantes ---
 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}}"
 
@@ -78,11 +79,44 @@ 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 utilisant l'exécution de code.
+    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
@@ -90,17 +124,44 @@ def generate_complete_latex(client, image_bytes):
         
         prompt = f"""
         # ROLE & OBJECTIF
-        Agis en tant qu'expert en mathématiques et tuteur pédagogue. Ton objectif est de créer une correction détaillée et 
-        irréprochable pour l'exercice mathématique présenté dans l'image fournie. 
-        
-        # TÂCHE PRINCIPALE
-        1. Analyse l'image pour comprendre l'énoncé de l'exercice.
-        2. Utilise Python pour effectuer des calculs complexes si nécessaire.
-        3. Résous l'exercice mathématique étape par étape.
-        4. Génère ensuite la solution complète en LaTeX.
-        
-        # SPÉCIFICATIONS TECHNIQUES DU CODE LATEX
-        [vos spécifications LaTeX actuelles...]
+Agis en tant qu'expert en mathématiques et tuteur pédagogue. Ton objectif est de créer une correction détaillée et irréprochable pour l'exercice mathématique présenté dans l'image fournie. La correction doit être sous forme de document LaTeX complet et directement compilable.
+
+# CONTEXTE
+- **Input:** Une image contenant un exercice de mathématiques.
+- **Niveau Cible:** Élève de Terminale S (Lycée, filière scientifique française).
+- **Output Attendu:** Un fichier source LaTeX (.tex) autonome.
+
+# 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.
+
+# SPÉCIFICATIONS TECHNIQUES DU CODE LATEX
+1.  **Structure du Document:** Commence **strictement** par `\\documentclass{{article}}` et se termine **strictement** par `\\end{{document}}`.
+2.  **Packages Requis:** Inclus impérativement les packages suivants via `\\usepackage{{...}}`: `amsmath`, `amssymb`, `geometry`, `hyperref`, `url`. Assure-toi qu'ils sont déclarés dans le préambule.
+3.  **Compilabilité:** Le code généré doit être valide et compilable sans erreur avec `pdflatex`.
+4.  **Formatage du Code:** Produis un code LaTeX propre, bien indenté et lisible.
+5.  **Environnements Mathématiques:** Utilise les environnements LaTeX appropriés (`align`, `equation`, `gather`, etc.) pour présenter les calculs et les équations de manière claire et standard.
+6.  **AUCUN Marqueur de Code:** Le résultat doit être **uniquement** le code LaTeX brut. N'inclus **JAMAIS** de marqueurs de code comme ```latex ... ``` ou ``` ... ``` au début ou à la fin.
+
+# STYLE & CONTENU DE LA SOLUTION
+1.  **Pédagogie:** La correction doit être claire, aérée et facile à comprendre pour un élève de Terminale S.
+2.  **Justifications:** Justifie **chaque** étape clé du raisonnement mathématique. Explique *pourquoi* une certaine méthode est utilisée ou *comment* on passe d'une étape à l'autre.
+3.  **Rigueur:** Assure l'exactitude mathématique complète de la solution.
+4.  **Structure Logique:** Organise la solution de manière logique. Utilise des sections (`\\section*{{...}}`, `\\subsection*{{...}}`) si cela améliore la clarté pour des problèmes longs ou multi-parties.
+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.
+3.  **Traduction en LaTeX:** Convertis ta résolution raisonnée en code LaTeX, en appliquant méticuleusement toutes les spécifications de formatage et de style demandées
         """
 
         response = client.models.generate_content(
@@ -108,57 +169,34 @@ def generate_complete_latex(client, image_bytes):
             contents=[image, prompt],
             config=genai.types.GenerateContentConfig(
                 temperature=0.3,
-                tools=[genai.types.Tool(
-                    code_execution=genai.types.ToolCodeExecution()
-                )]
+                tools=[types.Tool(code_execution=types.ToolCodeExecution)]
             )
         )
 
-        # Traitement de la réponse avec gestion de l'exécution de code
-        latex_content_raw = None
-        thinking_content = None
-        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:
-                if getattr(part, 'thought', None):
-                    thinking_content = part.text
-                elif getattr(part, 'executable_code', None):
-                    # Capture le code exécutable et son résultat
-                    code = part.executable_code.code
-                    result = getattr(part, 'code_execution_result', None)
-                    result_output = result.output if result else "Pas de résultat d'exécution"
-                    code_execution_results.append({
-                        'code': code,
-                        'result': result_output
-                    })
-                elif part.text:
-                    # Le contenu texte contient probablement le LaTeX
-                    if '\\documentclass' in part.text:
-                        latex_content_raw = part.text
-
-        # Si aucun contenu LaTeX n'a été trouvé dans les parties, chercher dans le texte global
-        if not latex_content_raw and response.text:
-            if '\\documentclass' in response.text:
-                latex_content_raw = response.text
+        latex_content_raw, thinking_content, code_execution_results = extract_content_from_response(response)
 
         if latex_content_raw:
             latex_content_cleaned = clean_latex(latex_content_raw)
             
-            # Vérifier et ajouter la mention obligatoire
+            # Vérification supplémentaire : est-ce que la mention obligatoire est présente ?
             if LATEX_MENTION not in latex_content_cleaned:
+                # Tente de l'insérer avant \end{document}
                 end_doc_tag = "\\end{document}"
                 if end_doc_tag in latex_content_cleaned:
                     latex_content_cleaned = latex_content_cleaned.replace(end_doc_tag, f"{LATEX_MENTION}\n{end_doc_tag}")
                 else:
+                    # Si \end{document} n'est pas là, ajoute les deux à la fin
                     latex_content_cleaned += f"\n{LATEX_MENTION}\n{end_doc_tag}"
                     
-            return latex_content_cleaned, thinking_content, None, code_execution_results
+            return latex_content_cleaned, thinking_content, None
         else:
-            return None, thinking_content, "Aucun contenu LaTeX n'a été généré par le modèle.", code_execution_results
+            return None, thinking_content, "Aucun contenu LaTeX n'a été généré par le modèle."
 
+    except genai.types.StopCandidateException as e:
+        return None, None, "❌ Génération stoppée (possible contenu inapproprié)"
     except Exception as e:
-        return None, None, f"❌ Erreur lors de la génération du LaTeX: {str(e)}", []
+        return None, None, f"❌ Erreur lors de la génération du LaTeX: {str(e)}"
+
 # --- Fonction de Compilation LaTeX ---
 
 def compile_latex_to_pdf(latex_content):
@@ -285,8 +323,8 @@ def process():
     if client is None:
         return jsonify({'success': False, 'message': message})
     
-    # Générer le LaTeX avec exécution de code
-    latex_content, thinking_content, error, code_execution_results = generate_complete_latex(client, image_bytes)
+    # Générer le LaTeX
+    latex_content, thinking_content, error = generate_complete_latex(client, image_bytes)
     if error:
         return jsonify({'success': False, 'message': error})
         
@@ -297,8 +335,7 @@ def process():
             'success': False, 
             'message': message,
             'latex': latex_content,
-            'thinking': thinking_content,
-            'code_execution': code_execution_results
+            'thinking': thinking_content
         })
     
     # Encoder le PDF en base64 pour le retourner au front-end
@@ -310,11 +347,9 @@ def process():
         'message': 'PDF généré avec succès',
         'latex': latex_content,
         'thinking': thinking_content,
-        'code_execution': code_execution_results,
         'pdf_base64': pdf_base64
     })
 
-    
 @app.route('/download-pdf', methods=['POST'])
 def download_pdf():
     if 'pdf_data' not in request.json: