Update main.py

main.py

@@ -1,36 +1,46 @@
-import os
-import google.generativeai as genai
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-
-app = FastAPI()
-
-# Получаем ключ из секретов HF
-genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
-
-model = genai.GenerativeModel("gemini-1.5-flash")
-
-class RequestData(BaseModel):
-    context: str
-    mood: str  # 'fun', 'serious', 'neutral', 'angry'
-
-@app.post("/complete")
-async def complete_text(data: RequestData):
-    try:
-        # Настройка личности
-        tone_instruction = ""
-        if data.mood == "fun":
-            tone_instruction = "Отвечай весело, используй сленг, эмодзи, будь на позитиве."
-        elif data.mood == "serious":
-            tone_instruction = "Отвечай максимально официально, сухо, деловой стиль, без воды."
-        elif data.mood == "angry":
-            tone_instruction = "Отвечай пассивно-агрессивно, кратко, с сарказмом. Ты устал от людей."
-        else: # neutral
-            tone_instruction = "Отвечай спокойно, нейтрально, как обычный помощник."
-
-        prompt = f"System: {tone_instruction}. Продолжи фразу пользователя. Не повторяй ввод. Только продолжение.\n\nUser Input: {data.context}\nOutput:"
-        
-        response = model.generate_content(prompt)
-        return {"completion": response.text.strip()}
-    except Exception as e:
+import os
+import google.generativeai as genai
+from fastapi import FastAPI, HTTPException, Request
+from pydantic import BaseModel
+import traceback # Добавили для отладки
+
+app = FastAPI()
+
+# Читаем ключ
+api_key = os.environ.get("GEMINI_API_KEY")
+
+if not api_key:
+    print("!!! ОШИБКА: GEMINI_API_KEY не найден в переменных окружения !!!")
+else:
+    print(f"Ключ найден, первые символы: {api_key[:5]}...")
+    genai.configure(api_key=api_key)
+
+# Попробуем gemini-pro, она доступна всем
+model = genai.GenerativeModel("gemini-1.5-flash") 
+
+class RequestData(BaseModel):
+    context: str
+    mood: str
+
+@app.post("/complete")
+async def complete_text(data: RequestData):
+    try:
+        print(f"Получен запрос: {data.context[:20]}...") # Лог запроса
+        
+        tone_instruction = "Отвечай нейтрально."
+        if data.mood == "fun": tone_instruction = "Отвечай весело."
+        elif data.mood == "serious": tone_instruction = "Отвечай серьезно."
+        elif data.mood == "angry": tone_instruction = "Отвечай злобно."
+
+        prompt = f"System: {tone_instruction}. Продолжи: {data.context}"
+        
+        response = model.generate_content(prompt)
+        print(f"Ответ от Gemini получен: {response.text[:20]}...")
+        
+        return {"completion": response.text.strip()}
+    except Exception as e:
+        # ВОТ ЭТО ПОКАЖЕТ ОШИБКУ В ЛОГАХ
+        error_msg = traceback.format_exc()
+        print("!!! КРИТИЧЕСКАЯ ОШИБКА !!!")
+        print(error_msg)
         raise HTTPException(status_code=500, detail=str(e))