Update app.py

app.py

@@ -17,33 +17,30 @@ class BasicAgent:
         self.generator = pipeline(
             "text2text-generation",
             model="google/flan-t5-large",
-            max_new_tokens=32,
-            temperature=0.0,      # VERY IMPORTANT: deterministic
+            max_new_tokens=64,
+            temperature=0.0,
             do_sample=False
         )
 
     def __call__(self, question: str) -> str:
         prompt = (
-            "Answer the question with ONLY the final answer.\n"
-            "No explanation. No extra words. No punctuation.\n\n"
-            f"Question: {question}\n"
+            "Answer the question EXACTLY as expected.\n"
+            "Give the final answer only.\n"
+            "Do not explain.\n\n"
+            f"Question:\n{question}\n\n"
             "Answer:"
         )
 
         result = self.generator(prompt)[0]["generated_text"]
 
-        # ---------- STRONG CLEANUP ----------
+        # ✅ MINIMAL CLEANUP (CRITICAL)
         answer = result.strip()
-        answer = re.sub(r"(?i)^(the answer is|answer:)\s*", "", answer)
-        answer = re.split(r"[\n\.]", answer)[0]
-        answer = answer.strip()
-        answer = answer.rstrip(".,;:")
-
-        # Normalize boolean answers
-        if answer.lower() == "true":
-            answer = "True"
-        elif answer.lower() == "false":
-            answer = "False"
+
+        # Remove only obvious prefixes
+        answer = re.sub(r"(?i)^(answer:|the answer is)\s*", "", answer)
+
+        # Keep full line, keep units, keep words
+        answer = answer.split("\n")[0].strip()
 
         print(f"\nQ: {question}\nA: {answer}\n")
         return answer