Update app.py

app.py

@@ -12,24 +12,41 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        from openai import OpenAI
-        self.llm = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-        print("BasicAgent initialized.")
+         self.api_key = os.getenv("GEMINI_API_KEY")
+        if not self.api_key:
+            raise ValueError("❌ GEMINI_API_KEY не найден в переменных окружения.")
+        self.api_url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
+        print("✅ BasicAgent with Gemini API initialized.")
+
     def __call__(self, question: str) -> str:
-        prompt = f"""Answer the following concisely and exactly:
-{question}"""
+        print(f"🔍 Обрабатывается вопрос: {question[:50]}...")
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}",
+        }
+
+        payload = {
+            "contents": [
+                {
+                    "parts": [
+                        {"text": question}
+                    ]
+                }
+            ]
+        }
+
         try:
-            response = self.llm.chat.completions.create(
-                model="gpt-4o",
-                 messages=[
-                    {"role": "system", "content": "You are an expert AI assistant."},
-                    {"role": "user", "content": prompt}
-                ],
-                temperature=0.2,
-            )
-            return response.choices[0].message.content.strip()
+            response = requests.post(self.api_url, headers=headers, json=payload, timeout=20)
+            response.raise_for_status()
+            result = response.json()
+            # Получение текста ответа
+            reply = result["candidates"][0]["content"]["parts"][0]["text"]
+            print(f"✅ Ответ: {reply[:50]}...")
+            return reply
         except Exception as e:
-            return f"ERROR: {e}"
+            print(f"⚠️ Ошибка при вызове Gemini API: {e}")
+            return "Ошибка при получении ответа от Gemini."
         
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):