update app.py

app.py

@@ -21,44 +21,36 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 
 class BasicAgent:
-    def __init__(self, model="google/gemma-7b"):
-        # Initialize the HF Agent with your model endpoint
-        self.agent = HfAgent(
-            endpoint=f"https://api-inference.huggingface.co/models/{model}",
-            headers={"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
-        )
-        print("BasicAgent initialized.")
+    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 __call__(self, question: str) -> str:
         print(f"Question: {question[:50]}...")
-        return self.agent.run(question)  # For full agent workflow
-        # OR for simple chat:
-        # return self.agent.chat(question)
-
-    def generate_response(self, prompt: str) -> str:
-        """Get response from model using chat interface"""
-        try:
-            response = self.agent.chat(prompt)
-            return response
-        except Exception as e:
-            return f"Error generating response: {str(e)}"
-
-
+        return self.agent.chat(question)
 
     def web_search(self, query: str) -> List[Dict]:
-        """Use SearxNG meta-search engine"""
+        """Use SearxNG meta-search engine to search the web"""
         params = {
             "q": query,
             "format": "json",
             "engines": "google,bing,duckduckgo"
         }
         try:
-            response = requests.get(self.searx_url, params=params)
+            response = requests.get(self.searx_url, params=params, timeout=10)
             response.raise_for_status()
             return response.json().get("results", [])
-        except requests.RequestException:
+        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"""
         page = self.wiki.page(query)