Update app.py

app.py

@@ -10,7 +10,7 @@ from smolagents import CodeAgent, LiteLLMModel, VisitWebpageTool, DuckDuckGoSear
 # --- CONSTANTS ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- AGENT CLASS (MISTRAL FOCUSED) ---
+# --- AGENT CLASS (Fixed Name) ---
 class BasicAgent:
     def __init__(self):
         print("🚀 Initializing Mistral-Powered Agent...")
@@ -18,69 +18,55 @@ class BasicAgent:
         # --- 1. API KEY CHECK ---
         mistral_key = os.getenv("MISTRAL_API_KEY")
         if not mistral_key:
-            raise ValueError("⚠️ MISTRAL_API_KEY environment variable mein nahi mili! Settings me add karo.")
+            # Agar Mistral nahi hai to error mat do, Qwen try karo (Fallback)
+            print("⚠️ Mistral Key not found. Please set MISTRAL_API_KEY for best results.")
+            # Fallback logic if needed, but for now we raise error to alert user
+            raise ValueError("⚠️ MISTRAL_API_KEY missing! Settings -> Secrets me add karo.")
 
-        # --- 2. MODEL SETUP (Mistral Large) ---
-        # Mistral Large logic aur code ke liye best hai
+        # --- 2. MODEL SETUP ---
         model = LiteLLMModel(
             model_id="mistral/mistral-large-latest",
             api_key=mistral_key
         )
 
         # --- 3. TOOLS ---
-        # Web Search aur Page Visit tools
         search_tool = DuckDuckGoSearchTool()
         visit_tool = VisitWebpageTool()
 
         # --- 4. CREATE AGENT ---
-        # CodeAgent use kar rahe hain jo seedha Python likhta hai
         self.agent = CodeAgent(
             tools=[search_tool, visit_tool],
             model=model,
-            # GAIA ke liye ye libraries zaroori hain
             additional_authorized_imports=[
                 "numpy", "pandas", "math", "datetime", "re", "csv", "json", "random", "itertools"
             ],
-            max_steps=25,  # Mushkil sawalon ke liye steps badhaye
+            max_steps=25,
             verbosity_level=2,
-            name="Mistral-Gaia-Solver"
+            # 👇 YAHAN CHANGE KIYA HAI (Hyphen hata ke Underscore lagaya)
+            name="Mistral_Gaia_Solver" 
         )
 
-    def __call__(self, question: str, file_path: Optional[str] = None) -> str:
-        """
-        Ye function agent ko run karta hai aur answer clean karta hai.
-        """
-        # --- PROMPT ENGINEERING FOR EXACT MATCH ---
+    def __call__(self, question: str, file_path: str = None) -> str:
+        # Prompt Logic
         prompt = f"""
-        Current Task: {question}
+        Task: {question}
         
         INSTRUCTIONS:
-        1. You are an expert Python coding agent. Solve this step-by-step.
-        2. If a file is attached, YOU MUST READ IT using Python code immediately.
-        3. Do not assume values. Calculate them using the file or web search.
-        
-        OUTPUT FORMAT RULES:
-        - Once you find the answer, output ONLY the final value.
-        - Do not write "The answer is...".
-        - Do not add units (like $, kg, years) unless specifically asked.
-        - Example: If the answer is 42, just print 42.
+        1. Use Python code to solve this step-by-step.
+        2. If a file is attached, YOU MUST READ IT using Python immediately.
+        3. Output ONLY the final answer value.
         """
         
         if file_path:
-            prompt += f"\n\n⚠️ ATTACHED FILE: A file is available at path '{file_path}'. Read it now."
+            prompt += f"\n\n⚠️ ATTACHED FILE: '{file_path}'"
 
         try:
-            # Agent Run
-            print(f"🤖 Agent thinking on: {question[:50]}...")
+            print(f"🤖 Agent working on: {question[:30]}...")
             response = self.agent.run(prompt)
             
-            # --- OUTPUT CLEANING ---
-            final_answer = str(response)
-            
-            # Common text removal
-            final_answer = final_answer.replace("Final Answer:", "").strip()
+            # Output Cleaning
+            final_answer = str(response).replace("Final Answer:", "").strip()
             
-            # Remove trailing dot (e.g., "100." -> "100")
             if final_answer.endswith(".") and len(final_answer) < 20:
                 final_answer = final_answer[:-1]