Update app.py

app.py

@@ -160,8 +160,6 @@ Provide a new, complete, and rigorously justified solution. Ensure that every er
             print(f"Error loading prompts: {e}")
         return prompts
 
-
-
     def extract_problem_text(self, img_str):
         try:
             response = client.models.generate_content(
@@ -239,37 +237,47 @@ def solve():
                     agent_system = AgentSystem()
 
                     # Étape 0: Extraction
-                    yield f'data: {json.dumps({"mode": "thinking"})}\n\n'
-                    yield f'data: {json.dumps({"content": "# 🔍 EXTRACTION DU PROBLÈME\n\nAnalyse de l’image pour extraire l’énoncé du problème...\n\n", "type": "text"})}\n\n'
+                    yield 'data: ' + json.dumps({"mode": "thinking"}) + '\n\n'
+                    
+                    # Pre-define strings with newlines to avoid f-string backslash issue
+                    extraction_msg = "# 🔍 EXTRACTION DU PROBLÈME\n\nAnalyse de l'image pour extraire l'énoncé du problème...\n\n"
+                    yield 'data: ' + json.dumps({"content": extraction_msg, "type": "text"}) + '\n\n'
 
                     problem_text = agent_system.extract_problem_text(img_str)
-                    yield f'data: {json.dumps({"content": f"**Problème identifié:**\n{problem_text}\n\n", "type": "text"})}\n\n'
+                    problem_msg = f"**Problème identifié:**\n{problem_text}\n\n"
+                    yield 'data: ' + json.dumps({"content": problem_msg, "type": "text"}) + '\n\n'
 
                     # Étape 1
-                    yield f'data: {json.dumps({"content": "# 📝 ÉTAPE 1: SOLUTION INITIALE\n\n", "type": "text"})}\n\n'
+                    step1_msg = "# 📝 ÉTAPE 1: SOLUTION INITIALE\n\n"
+                    yield 'data: ' + json.dumps({"content": step1_msg, "type": "text"}) + '\n\n'
+                    
                     step1_prompt = agent_system.prompts["step1_initial_solution"].replace(
                         "[The mathematical problem will be inserted here]", problem_text
                     )
                     initial_solution = agent_system.run_agent_step("step1", step1_prompt, use_calculator)
-                    yield f'data: {json.dumps({"content": initial_solution, "type": "text"})}\n\n'
+                    yield 'data: ' + json.dumps({"content": initial_solution, "type": "text"}) + '\n\n'
 
                     # Étape 2
-                    yield f'data: {json.dumps({"content": "# 🔧 ÉTAPE 2: AUTO-AMÉLIORATION\n\n", "type": "text"})}\n\n'
+                    step2_msg = "# 🔧 ÉTAPE 2: AUTO-AMÉLIORATION\n\n"
+                    yield 'data: ' + json.dumps({"content": step2_msg, "type": "text"}) + '\n\n'
+                    
                     step2_prompt = agent_system.prompts["step2_self_improvement"].replace(
                         "[The initial solution attempt will be inserted here]", initial_solution
                     )
                     improved_solution = agent_system.run_agent_step("step2", step2_prompt, use_calculator)
-                    yield f'data: {json.dumps({"content": improved_solution, "type": "text"})}\n\n'
+                    yield 'data: ' + json.dumps({"content": improved_solution, "type": "text"}) + '\n\n'
 
                     # Étape 3
-                    yield f'data: {json.dumps({"content": "# ✅ ÉTAPE 3: VÉRIFICATION\n\n", "type": "text"})}\n\n'
+                    step3_msg = "# ✅ ÉTAPE 3: VÉRIFICATION\n\n"
+                    yield 'data: ' + json.dumps({"content": step3_msg, "type": "text"}) + '\n\n'
+                    
                     step3_prompt = agent_system.prompts["step3_verification"].replace(
                         "[The mathematical problem will be inserted here]", problem_text
                     ).replace(
                         "[The solution to be verified will be inserted here]", improved_solution
                     )
                     verification_result = agent_system.run_agent_step("step3", step3_prompt, False)
-                    yield f'data: {json.dumps({"content": verification_result, "type": "text"})}\n\n'
+                    yield 'data: ' + json.dumps({"content": verification_result, "type": "text"}) + '\n\n'
 
                     needs_correction = (
                         "Critical Error" in verification_result
@@ -278,7 +286,9 @@ def solve():
                     )
 
                     if needs_correction:
-                        yield f'data: {json.dumps({"content": "# 🛠️ ÉTAPE 5: CORRECTION\n\n", "type": "text"})}\n\n'
+                        step5_msg = "# 🛠️ ÉTAPE 5: CORRECTION\n\n"
+                        yield 'data: ' + json.dumps({"content": step5_msg, "type": "text"}) + '\n\n'
+                        
                         step5_prompt = agent_system.prompts["step5_correction"].replace(
                             "[The full verification report will be inserted here]", verification_result
                         ).replace(
@@ -286,14 +296,17 @@ def solve():
                         )
                         corrected_solution = agent_system.run_agent_step("step5", step5_prompt, use_calculator)
                         final_solution = corrected_solution
-                        yield f'data: {json.dumps({"content": corrected_solution, "type": "text"})}\n\n'
+                        yield 'data: ' + json.dumps({"content": corrected_solution, "type": "text"}) + '\n\n'
                     else:
                         final_solution = improved_solution
-                        yield f'data: {json.dumps({"content": "✅ La solution a été validée sans correction.\n\n", "type": "text"})}\n\n'
+                        validated_msg = "✅ La solution a été validée sans correction.\n\n"
+                        yield 'data: ' + json.dumps({"content": validated_msg, "type": "text"}) + '\n\n'
 
-                    yield f'data: {json.dumps({"mode": "answering"})}\n\n'
-                    yield f'data: {json.dumps({"content": "# 📋 SOLUTION FINALE\n\n", "type": "text"})}\n\n'
-                    yield f'data: {json.dumps({"content": final_solution, "type": "text"})}\n\n'
+                    yield 'data: ' + json.dumps({"mode": "answering"}) + '\n\n'
+                    
+                    final_msg = "# 📋 SOLUTION FINALE\n\n"
+                    yield 'data: ' + json.dumps({"content": final_msg, "type": "text"}) + '\n\n'
+                    yield 'data: ' + json.dumps({"content": final_solution, "type": "text"}) + '\n\n'
 
                 else:
                     prompt = BASE_PROMPT
@@ -316,11 +329,11 @@ def solve():
                     for chunk in response:
                         for part in chunk.candidates[0].content.parts:
                             if hasattr(part, 'text') and part.text:
-                                yield f'data: {json.dumps({"content": part.text, "type": "text"})}\n\n'
+                                yield 'data: ' + json.dumps({"content": part.text, "type": "text"}) + '\n\n'
 
             except Exception as e:
                 print(f"Error during generation: {e}")
-                yield f'data: {json.dumps({"error": "Erreur inattendue"})}\n\n'
+                yield 'data: ' + json.dumps({"error": "Erreur inattendue"}) + '\n\n'
 
         return Response(
             stream_with_context(generate()),