Update app.py

app.py

@@ -18,35 +18,34 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 class BasicAgent:
-     def __init__(self, model_name: str = "gemini-2.0-flash-lite"):
-        """
-        Multi-modal agent powered by Google Gemini with:
-        - Web search
-        - Wikipedia access
-        - Document processing
-        """
-        self.model = genai.GenerativeModel(model_name)
-        self.wiki = wikipediaapi.Wikipedia('en')
-        self.searx_url = "https://searx.space/search"  # Public Searx instance
-        
+    def __init__(self, model="google/gemma-7b"):
+        self.api_url = f"https://api-inference.huggingface.co/models/{model}"
+        self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
         print("BasicAgent initialized.")
 
+        #usage
+        #agent = HuggingFaceAgent("google/gemma-7b")  # Same architecture as Gemini
+        #print(agent.generate("Explain quantum computing"))
+      
+    
      def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.generate_response(question)
+        fixed_answer = self.agent.generate(question)
         print(f"Agent returning answer: {fixed_answer}")
         return fixed_answer
 
 
-
+     # to check
      def generate_response(self, prompt: str) -> str:
-        """Get response from Gemini"""
+        """Get response from Gema"""
         try:
             response = self.model.generate_content(prompt)
             return response.text
         except Exception as e:
             return f"Error generating response: {str(e)}"
 
+
+
      def web_search(self, query: str) -> List[Dict]:
         """Use SearxNG meta-search engine"""
         params = {