Update app.py

app.py

@@ -10,37 +10,41 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
     def __init__(self):
-        print("WolfarmAlpha Agent Initialized")
-        self.app_id = os.getenv("WOLFARM_APP_ID")
-        
+        print("FLAN-T5 API Agent initialized.")
+        self.api_token = os.getenv("HF_API_TOKEN")
+        self.api_url = "https://api-inference.huggingface.co/models/google/flan-t5-small"
+        self.headers = {
+            "Authorization": f"Bearer {self.api_token}",
+            "Content-Type": "application/json"
+        }
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
-        if not self.app_id:
-            print("❌ WolframAlpha App ID not found.")
-            return "Error: WolframAlpha App ID not set."
+        if not self.api_token:
+            return "Error: Missing Hugging Face API token."
+
+        prompt = f"Answer this question clearly and briefly:\n{question.strip()}"
+
+        payload = {
+            "inputs": prompt
+        }
 
         try:
-            url = "http://api.wolframalpha.com/v1/result"
-            params = {
-                "i": question,
-                "appid": self.app_id
-            }
-            response = requests.get(url, params=params, timeout=10)
-
-            print(f"🔁 Request URL: {response.url}")
-
-            if response.status_code == 200:
-                return response.text
-            elif response.status_code == 501:
-                return "Error: WolframAlpha API endpoint doesn't support this query type (501)."
+            response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=15)
+            response.raise_for_status()
+            output = response.json()
+
+            if isinstance(output, list) and "generated_text" in output[0]:
+                return output[0]["generated_text"]
             else:
-                return f"Couldn't retrieve a response from WolframAlpha. Status {response.status_code}"
+                print(f"Unexpected API response: {output}")
+                return "Model did not return a valid response."
 
         except Exception as e:
-            return f"Exception occurred: {str(e)}"
+            print(f"Exception during inference: {e}")
+            return f"API Error: {str(e)}"
         
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):