Update app.py

app.py

@@ -13,12 +13,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class EnhancedGAIAAgentFree:
     """
     GAIA Agent for free HuggingFace models.
-    Avoids PRO credits and multi-modal content.
+    Returns 'unknown' for images, videos, code, Excel, or unsupported questions.
     """
 
     def __init__(self):
         print("🚀 GAIAAgent initializing... (FREE version)")
-        
         hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
         if not hf_token:
             print("⚠️ HF_TOKEN not found! Add it to Space secrets.")
@@ -27,64 +26,48 @@ class EnhancedGAIAAgentFree:
             return
 
         self.client = InferenceClient(token=hf_token)
-        self.model = "TheBloke/guanaco-7B-GPTQ"  # free HF model
+        self.model = "TheBloke/guanaco-7B-GPTQ"  # free model
         print(f"✅ Model loaded: {self.model}")
 
     def __call__(self, question: str) -> str:
         """
-        Answer a question with free LLM.
+        Answer a question or return 'unknown' for unsupported content.
         """
-        print(f"\nQ: {question[:150]}...")
-        if not self.client or not self.model:
-            return "ERROR: HF_TOKEN not configured"
-
-        # Skip unsupported questions
-        if any(word in question.lower() for word in ["image", "video", "file", "attached", "excel", "code"]):
-            return "unknown"
-
         try:
-            answer = self._generate_answer(question)
-            print(f"A: {answer[:150]}...")
-            return answer
-        except Exception as e:
-            print(f"❌ Agent error: {e}")
-            return "unknown"
+            # Filter unsupported content
+            unsupported = ["image", "video", "file", "attached", "excel", "python code", "chart", "csv"]
+            if any(word in question.lower() for word in unsupported):
+                return "unknown"
 
-    def _generate_answer(self, question: str) -> str:
-        """
-        Free LLM answer generator with safe prompt.
-        """
-        prompt = f"""
+            # Very short or empty questions
+            if not question.strip() or len(question.strip()) < 5:
+                return "unknown"
+
+            # Free text generation for supported questions
+            prompt = f"""
 You are an expert for GAIA benchmark.
 Answer concisely. ONLY provide the final answer.
 If you cannot determine the answer, write "unknown".
 Question: {question}
 FINAL ANSWER:"""
 
-        response = self.client.text_generation(
-            model=self.model,
-            prompt=prompt,
-            max_new_tokens=128,
-            temperature=0.1,
-            do_sample=False,
-            return_full_text=False
-        )
-        return self._clean_answer(response)
+            response = self.client.text_generation(
+                model=self.model,
+                prompt=prompt,
+                max_new_tokens=128,
+                temperature=0.1,
+                do_sample=False,
+                return_full_text=False
+            )
+
+            # Clean response
+            answer = response.strip() if response else "unknown"
+            return answer
 
-    def _clean_answer(self, text: str) -> str:
-        """
-        Keep only the final answer text.
-        """
-        if not text:
+        except Exception as e:
+            print(f"❌ Agent error: {e}")
             return "unknown"
 
-        # Remove prefixes
-        prefixes = ["Answer:", "The answer is", "A:", "FINAL ANSWER:", "Result:"]
-        for p in prefixes:
-            if text.lower().startswith(p.lower()):
-                text = text[len(p):].strip()
-        return text.strip() if text.strip() else "unknown"
-
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """