Update app.py

app.py

@@ -14,20 +14,21 @@ class BasicAgent:
     def __init__(self):
         self.api_key = os.getenv("GEMINI_API_KEY")
         if not self.api_key:
-            raise ValueError("❌ Переменная окружения GEMINI_API_KEY не найдена.")
-
-        self.model_name = "gemini-1.5-pro"
-        self.api_url = f"https://generativelanguage.googleapis.com/v1/models/{self.model_name}:generateContent?key={self.api_key}"
-        print(f"✅ Gemini API URL: {self.api_url}")
+            raise ValueError("❌ Переменная GEMINI_API_KEY не найдена. Установи её в Hugging Face Secrets.")
+        
+        self.model = "gemini-1.5-pro"
+        self.api_url = f"https://generativelanguage.googleapis.com/v1/models/{self.model}:generateContent?key={self.api_key}"
+        print(f"✅ Используется Gemini API: {self.api_url}")
 
     def __call__(self, question: str) -> str:
-        print(f"🤖 Вопрос: {question[:80]}")
-
-        headers = {"Content-Type": "application/json"}
+        headers = {
+            "Content-Type": "application/json"
+        }
 
         payload = {
             "contents": [
                 {
+                    "role": "user",
                     "parts": [{"text": question}]
                 }
             ]
@@ -35,21 +36,26 @@ class BasicAgent:
 
         try:
             response = requests.post(self.api_url, headers=headers, json=payload, timeout=30)
+
+            if response.status_code == 429:
+                print("❗️Превышен лимит запросов (429). Пауза 10 секунд...")
+                time.sleep(10)
+                return "Превышен лимит запросов. Попробуйте позже."
+
+            if response.status_code == 400:
+                print("❌ Ошибка 400. Проверь формат запроса.")
+                print("Ответ:", response.text[:500])
+                return "Ошибка 400: Неверный формат запроса."
+
             response.raise_for_status()
 
             data = response.json()
             answer = data["candidates"][0]["content"]["parts"][0]["text"]
-            print(f"✅ Ответ Gemini: {answer[:100]}")
             return answer
 
-        except requests.exceptions.HTTPError as e:
-            print("❌ HTTP ошибка:", e)
-            print("Ответ сервера:", response.text[:500])
-            return f"Ошибка HTTP: {e}"
-
         except Exception as e:
-            print("❌ Другая ошибка:", e)
-            return "Произошла ошибка при подключении к Gemini API."
+            print("❌ Ошибка при вызове Gemini:", e)
+            return f"Ошибка при вызове Gemini API: {e}"