Update app.py

app.py

@@ -39,7 +39,7 @@ class ReasoningAgent:
         )
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question}")
+        print(f"\n=== New Question ===\n{question}\n")
 
         # Prompt structuré
         prompt = f"""
@@ -59,34 +59,40 @@ Question: {question}
         inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
         output = model.generate(**inputs, max_new_tokens=300)
         text_output = tokenizer.decode(output[0], skip_special_tokens=True)
-        print(f"Raw model output: {text_output[:300]}...")  # pour debug
+        print(f"Raw model output:\n{text_output[:1000]}...\n")  # on affiche un gros morceau du texte
 
         # Parser le JSON
         try:
             parsed = json.loads(text_output)
-            action = parsed.get("action", "None")
-            observation = ""
-            if action.startswith("AddTwoNumbers"):
-                # extraire les deux nombres
-                try:
-                    numbers = action[action.find("(")+1:action.find(")")].split(",")
-                    a, b = int(numbers[0].strip()), int(numbers[1].strip())
-                    observation = AddTwoNumbers(a, b)
-                    parsed["observation"] = str(observation)
-                    print(f"Performed tool action: {action} = {observation}")
-                except Exception as e:
-                    observation = f"Error parsing numbers: {e}"
-                    parsed["observation"] = observation
-
-            final_answer = parsed.get("answer", "No answer returned.")
-            reasoning = parsed.get("thought", "")
-        except json.JSONDecodeError:
-            final_answer = text_output
-            reasoning = ""
-
-        print(f"Thought: {reasoning}")
-        print(f"Answer: {final_answer}")
-        return final_answer
+        except json.JSONDecodeError as e:
+            print(f"⚠️ JSON decode error: {e}")
+            parsed = {"thought": "", "action": "None", "observation": "", "answer": text_output}
+
+        thought = parsed.get("thought", "")
+        action = parsed.get("action", "None")
+        observation = parsed.get("observation", "")
+        answer = parsed.get("answer", "No answer returned.")
+
+        # Exécution de l'outil si nécessaire
+        if action.startswith("AddTwoNumbers"):
+            try:
+                numbers = action[action.find("(")+1:action.find(")")].split(",")
+                a, b = int(numbers[0].strip()), int(numbers[1].strip())
+                observation = AddTwoNumbers(a, b)
+                parsed["observation"] = str(observation)
+                print(f"✅ Tool executed: {action} -> {observation}")
+            except Exception as e:
+                observation = f"⚠️ Error executing tool: {e}"
+                parsed["observation"] = observation
+                print(observation)
+
+        # Print du raisonnement actuel
+        print(f"💭 Thought: {thought}")
+        print(f"🔧 Action: {action}")
+        print(f"👀 Observation: {observation}")
+        print(f"📝 Answer: {answer}\n{'-'*50}\n")
+
+        return answer
 
 # --- Main Evaluation & Submission Function ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):