Update app.py

app.py

@@ -33,22 +33,22 @@ class BasicAgent:
         return answer.strip()
 
 
+
     def __call__(self, question: str) -> str:
         print(f"Question: {question[:100]}")
     
         q = question.lower()
     
-        # ✅ 1. Detectar preguntas simples
-        if "how many" in q:
-            try:
-                import re
-                nums = re.findall(r"\d+", question)
-                if nums:
-                    return nums[0]
-            except:
-                pass
+        # ✅ fallback SIEMPRE (evita blanks)
+        fallback = "unknown"
     
-        # ✅ 2. Matemáticas simples
+        # ✅ 1. detectar números simples
+        import re
+        nums = re.findall(r"\d+", question)
+        if "how many" in q and nums:
+            return nums[-1]
+    
+        # ✅ 2. matemáticas simples
         if any(x in q for x in ["sum", "add", "multiply", "divide"]):
             try:
                 expr = re.findall(r"[0-9\+\-\*\/\.\(\) ]+", question)[0]
@@ -56,35 +56,35 @@ class BasicAgent:
             except:
                 pass
     
-        # ✅ 3. Respuestas cortas (nombres, lugares)
-        
-        prompt = f"""
-        Give ONLY a single short answer (one word or number).
-        
-        Question: {question}
-        Answer:
-        """
-
-    
+        # ✅ 3. llamada HF con protección
         try:
             response = self.client.text_generation(
                 model="google/flan-t5-large",
-                prompt=prompt,
+                prompt=f"Answer with one word or number: {question}",
                 max_new_tokens=20
             )
     
-            answer = self.clean_answer(response)
+            # ✅ controlar respuesta vacía
+            if not response or response.strip() == "":
+                print("Empty HF response → fallback")
+                return fallback
+    
+            answer = response.strip()
     
-            # cortar cosas largas
-            answer = answer.split(",")[0]
+            # limpiar
+            answer = answer.split("\n")[0]
             answer = answer.split(".")[0]
+            answer = answer.split(",")[0].strip()
+    
+            if answer == "":
+                return fallback
+    
+            return answer
     
         except Exception as e:
             print(f"HF error: {e}")
-            answer = ""
-    
-        print(f"Final answer: {answer}")
-        return answer
+            return fallback
+