Update app.py

app.py

@@ -1,4 +1,3 @@
-
 from flask import Flask, render_template, request, jsonify, Response, stream_with_context
 from google import genai
 from google.genai import types
@@ -90,7 +89,7 @@ Provide the improved version of the solution, adhering to the original problem's
 **2. How to Handle Issues in the Solution**
 When you identify an issue in a step, you MUST first classify it into one of the following two categories and then follow the specified procedure.
 *   **a. Critical Error:**
-    This is any error that breaks the logical chain of the proof. This includes both **logical fallacies** (e.g., claiming that ‘A>B, C>D’ implies ‘A-C>B-D’) and **factual errors** (e.g., a calculation error like '2+3=6').
+    This is any error that breaks the logical chain of the proof. This includes both **logical fallacies** (e.g., claiming that 'A>B, C>D' implies 'A-C>B-D') and **factual errors** (e.g., a calculation error like '2+3=6').
     *   **Procedure:**
         *   Explain the specific error and state that it **invalidates the current line of reasoning**.
         *   Do NOT check any further steps that rely on this error.
@@ -209,43 +208,61 @@ def solve():
                 current_solution = ""
                 
                 # --- Étape 0: Extraction du problème ---
-                yield f'data: {json.dumps({"content": "### Étape 0 : Extraction de l\'énoncé\nAnalyse de l’image...", "type": "header"})}\n\n'
+                # Fixed: moved the string outside the f-string expression
+                step0_msg = "### Étape 0 : Extraction de l'énoncé\nAnalyse de l'image..."
+                yield f'data: {json.dumps({"content": step0_msg, "type": "header"})}\n\n'
+                
                 problem_text = agent_system.extract_problem_text(img_str)
-                yield f'data: {json.dumps({"content": f"**Énoncé détecté :**\n\n{problem_text}\n\n---\n", "type": "text"})}\n\n'
+                problem_msg = f"**Énoncé détecté :**\n\n{problem_text}\n\n---\n"
+                yield f'data: {json.dumps({"content": problem_msg, "type": "text"})}\n\n'
 
                 # --- Pipeline de raisonnement ---
                 # Étape 1: Solution Initiale
-                yield f'data: {json.dumps({"content": "### Étape 1 : Génération de la solution initiale\nLe modèle prépare une première version de la preuve...", "type": "header"})}\n\n'
+                step1_msg = "### Étape 1 : Génération de la solution initiale\nLe modèle prépare une première version de la preuve..."
+                yield f'data: {json.dumps({"content": step1_msg, "type": "header"})}\n\n'
+                
                 step1_prompt = agent_system.prompts["step1_initial_solution"] + f"\n\n### Problem ###\n{problem_text}"
                 current_solution = agent_system.run_agent_step("1 (Initial)", step1_prompt)
-                yield f'data: {json.dumps({"content": f"**Solution Initiale :**\n\n{current_solution}\n\n---\n", "type": "text"})}\n\n'
+                
+                solution1_msg = f"**Solution Initiale :**\n\n{current_solution}\n\n---\n"
+                yield f'data: {json.dumps({"content": solution1_msg, "type": "text"})}\n\n'
 
                 # Itération de Vérification-Correction (ici, 1 seule itération pour la démo)
                 MAX_ITERATIONS = 1 
                 for i in range(MAX_ITERATIONS):
                     # Étape 3: Vérification
-                    yield f'data: {json.dumps({"content": f"### Étape 3.{i+1} : Vérification par l\'expert\nAnalyse critique de la solution pour détecter les erreurs...", "type": "header"})}\n\n'
+                    step3_msg = f"### Étape 3.{i+1} : Vérification par l'expert\nAnalyse critique de la solution pour détecter les erreurs..."
+                    yield f'data: {json.dumps({"content": step3_msg, "type": "header"})}\n\n'
+                    
                     step3_prompt = agent_system.prompts["step3_verification"] + f"\n\n### Problem ###\n{problem_text}\n\n### Solution to be verified ###\n{current_solution}"
                     verification_report = agent_system.run_agent_step(f"3.{i+1} (Verification)", step3_prompt)
-                    yield f'data: {json.dumps({"content": f"**Rapport de vérification :**\n\n{verification_report}\n\n---\n", "type": "text"})}\n\n'
+                    
+                    verification_msg = f"**Rapport de vérification :**\n\n{verification_report}\n\n---\n"
+                    yield f'data: {json.dumps({"content": verification_msg, "type": "text"})}\n\n'
 
                     # Vérifier si une correction est nécessaire
                     if "critical error" in verification_report.lower() or "justification gap" in verification_report.lower():
                         # Étape 5: Correction
-                        yield f'data: {json.dumps({"content": f"### Étape 5.{i+1} : Correction de la solution\nLe modèle utilise le rapport pour corriger sa preuve...", "type": "header"})}\n\n'
+                        step5_msg = f"### Étape 5.{i+1} : Correction de la solution\nLe modèle utilise le rapport pour corriger sa preuve..."
+                        yield f'data: {json.dumps({"content": step5_msg, "type": "header"})}\n\n'
+                        
                         step5_prompt = agent_system.prompts["step5_correction"].replace(
                             "[The full verification report will be inserted here]", verification_report
                         ).replace(
                             "[The previous solution attempt will be inserted here]", current_solution
                         )
                         current_solution = agent_system.run_agent_step(f"5.{i+1} (Correction)", step5_prompt)
-                        yield f'data: {json.dumps({"content": f"**Solution Corrigée :**\n\n{current_solution}\n\n---\n", "type": "text"})}\n\n'
+                        
+                        correction_msg = f"**Solution Corrigée :**\n\n{current_solution}\n\n---\n"
+                        yield f'data: {json.dumps({"content": correction_msg, "type": "text"})}\n\n'
                     else:
-                        yield f'data: {json.dumps({"content": "✅ Le rapport de vérification n\'a trouvé aucune erreur critique. La solution est considérée comme valide.", "type": "header"})}\n\n'
+                        success_msg = "✅ Le rapport de vérification n'a trouvé aucune erreur critique. La solution est considérée comme valide."
+                        yield f'data: {json.dumps({"content": success_msg, "type": "header"})}\n\n'
                         break
 
                 # --- Affichage Final ---
-                yield f'data: {json.dumps({"content": "# Solution Finale Validée\nVoici la version finale de la correction.", "type": "header"})}\n\n'
+                final_header_msg = "# Solution Finale Validée\nVoici la version finale de la correction."
+                yield f'data: {json.dumps({"content": final_header_msg, "type": "header"})}\n\n'
                 yield f'data: {json.dumps({"content": current_solution, "type": "final"})}\n\n'
                 
             except Exception as e:
@@ -261,4 +278,4 @@ def solve():
         return jsonify({'error': error_message}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True)