Update app.py

app.py

@@ -14,13 +14,16 @@ class BasicAgent:
     def __init__(self):
         self.api_key = os.getenv("GEMINI_API_KEY")
         if not self.api_key:
-            raise ValueError("❌ GEMINI_API_KEY не найден в переменных окружения.")
-        # Используем актуальную версию v1 и модель gemini-1.5-pro
-        self.api_url = f"https://generativelanguage.googleapis.com/v1/models/gemini-1.5-pro:generateContent?key={self.api_key}"
-        print("✅ BasicAgent with Gemini API initialized.")
+            raise ValueError("❌ Переменная окружения GEMINI_API_KEY не найдена.")
+        
+        # Модель и версия API
+        self.model_name = "gemini-1.5-pro"  # Gemini 2.5 = Gemini 1.5 Pro
+        self.api_url = f"https://generativelanguage.googleapis.com/v1/models/{self.model_name}:generateContent?key={self.api_key}"
+        
+        print(f"✅ BasicAgent инициализирован с моделью: {self.model_name}")
 
     def __call__(self, question: str) -> str:
-        print(f"🔍 Обрабатывается вопрос: {question[:50]}...")
+        print(f"🧠 Вопрос получен: {question[:80]}")
 
         headers = {
             "Content-Type": "application/json"
@@ -37,15 +40,23 @@ class BasicAgent:
         }
 
         try:
-            response = requests.post(self.api_url, headers=headers, json=payload, timeout=20)
+            response = requests.post(self.api_url, headers=headers, json=payload, timeout=30)
             response.raise_for_status()
             result = response.json()
-            reply = result["candidates"][0]["content"]["parts"][0]["text"]
-            print(f"✅ Ответ: {reply[:80]}...")
-            return reply
+            
+            # Извлечение текста ответа
+            answer = result["candidates"][0]["content"]["parts"][0]["text"]
+            print(f"✅ Ответ: {answer[:80]}")
+            return answer
+
+        except requests.exceptions.RequestException as e:
+            print(f"❌ Ошибка HTTP: {e}")
+            return "Ошибка при подключении к Gemini API."
+
         except Exception as e:
-            print(f"⚠️ Ошибка при вызове Gemini API: {e}")
-            print("Ответ сервера:", response.text if 'response' in locals() else 'нет ответа')
+            print(f"❌ Общая ошибка: {e}")
+            if 'response' in locals():
+                print("Ответ сервера:", response.text[:500])
             return "Ошибка при получении ответа от Gemini."