Update app.py

app.py

@@ -9,7 +9,7 @@ import wikipediaapi
 import pandas as pd
 from transformers import pipeline  # or HfAgent if you want the higher-level agent
 from huggingface_hub import InferenceClient  # Updated import
-from huggingface_hub.inference._text_generation import TextGenerationResponse
+
 
 load_dotenv()
 
@@ -21,33 +21,27 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 
 class BasicAgent:
-    def __init__(self, searx_url: str = "https://search.example.org"):
-        # Initialize HF Inference Client
-        self.client = InferenceClient(model="bigcode/starcoder")
+    def __init__(self, model: str = "google/gemma-7b", searx_url: str = "https://search.example.org"):
+        self.client = InferenceClient(model=model)
         self.searx_url = searx_url
-        print("BasicAgent initialized with Inference API")
+        print(f"Initialized with model: {model}")
 
     def __call__(self, question: str) -> str:
-        print(f"Question: {question[:50]}...")
-        response: TextGenerationResponse = self.client.text_generation(
+        print(f"Processing: {question[:50]}...")
+        return self.client.text_generation(
             prompt=question,
-            max_new_tokens=100
+            max_new_tokens=100,
+            temperature=0.7
         )
-        return response
 
     def web_search(self, query: str) -> List[Dict]:
-        """Use SearxNG meta-search engine"""
-        params = {
-            "q": query,
-            "format": "json",
-            "engines": "google,bing,duckduckgo"
-        }
+        """Search using SearxNG"""
+        params = {"q": query, "format": "json", "engines": "google,bing"}
         try:
             response = requests.get(self.searx_url, params=params, timeout=10)
-            response.raise_for_status()
             return response.json().get("results", [])
-        except requests.RequestException as e:
-            print(f"Search error: {str(e)}")
+        except Exception as e:
+            print(f"Search failed: {e}")
             return []
             
     def wikipedia_search(self, query: str) -> str: