Update app.py

app.py

@@ -8,7 +8,8 @@ import requests
 import wikipediaapi
 import pandas as pd
 from transformers import pipeline  # or HfAgent if you want the higher-level agent
-from huggingface_hub import HfAgent
+from huggingface_hub import InferenceClient  # Updated import
+from huggingface_hub.inference._text_generation import TextGenerationResponse
 
 load_dotenv()
 
@@ -20,20 +21,22 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 
 class BasicAgent:
-    def __init__(self, searx_url: str = "https://searx.example.org"):
-        # Initialize HF Agent
-        self.agent = HfAgent("https://api-inference.huggingface.co/models/bigcode/starcoder")
-        
-        # Initialize web search configuration
-        self.searx_url = searx_url  # You should replace with an actual SearxNG instance
-        print("BasicAgent initialized with HfAgent and web search capabilities")
+    def __init__(self, searx_url: str = "https://search.example.org"):
+        # Initialize HF Inference Client
+        self.client = InferenceClient(model="bigcode/starcoder")
+        self.searx_url = searx_url
+        print("BasicAgent initialized with Inference API")
 
     def __call__(self, question: str) -> str:
         print(f"Question: {question[:50]}...")
-        return self.agent.chat(question)
+        response: TextGenerationResponse = self.client.text_generation(
+            prompt=question,
+            max_new_tokens=100
+        )
+        return response
 
     def web_search(self, query: str) -> List[Dict]:
-        """Use SearxNG meta-search engine to search the web"""
+        """Use SearxNG meta-search engine"""
         params = {
             "q": query,
             "format": "json",
@@ -46,9 +49,6 @@ class BasicAgent:
         except requests.RequestException as e:
             print(f"Search error: {str(e)}")
             return []
-        except Exception as e:
-            print(f"Unexpected error: {str(e)}")
-            return []
             
     def wikipedia_search(self, query: str) -> str:
         """Get Wikipedia summary"""