Update app.py

app.py

@@ -8,7 +8,7 @@ import requests
 from typing import List, Dict, Union
 import pandas as pd
 import wikipediaapi
-#from docx import Document
+from google import genai
 
 load_dotenv()
 
@@ -20,7 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self, model="google/gemma-7b"):
-        self.api_url = f"https://api-inference.huggingface.co/models/{model}"
+        #self.api_url = f"https://api-inference.huggingface.co/models/{model}"
         self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
         # Wikipedia setup (with proper User-Agent)
         self.wiki = wikipediaapi.Wikipedia(
@@ -34,10 +34,22 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.generate_response(question)
+        fixed_answer = self.answer_question(question)
         print(f"Agent returning answer: {fixed_answer}")
         return fixed_answer
 
+    def answer_question(self, prompt: str) -> str:
+        from google import genai
+
+        client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        
+        response = client.models.generate_content(
+            model="gemini-2.0-flash",
+            contents=prompt,
+        )
+        
+        return response.text
+
 
     def generate_response(self, prompt: str) -> str:
         """Get response from HuggingFace model"""