Update main.py

main.py

@@ -1,46 +1,46 @@
 import os
 import google.generativeai as genai
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-import traceback # Добавили для отладки
 
 app = FastAPI()
 
-# Читаем ключ
-api_key = os.environ.get("GEMINI_API_KEY")
+# Получаем ключ
+genai.configure(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") 
+# ВАЖНО: Убедись, что в requirements.txt версия >=0.7.2, иначе flash не заработает
+model = genai.GenerativeModel("gemini-1.5-flash")
 
 class RequestData(BaseModel):
     context: str
-    mood: 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 = "Отвечай злобно."
+        # Настройка личности
+        tone_instruction = ""
+        if data.mood == "fun":
+            tone_instruction = "Отвечай весело, используй сленг, эмодзи."
+        elif data.mood == "serious":
+            tone_instruction = "Отвечай официально, сухо, деловой стиль."
+        elif data.mood == "angry":
+            tone_instruction = "Отвечай пассивно-агрессивно, с сарказмом."
+        else: 
+            tone_instruction = "Отвечай спокойно и нейтрально."
 
-        prompt = f"System: {tone_instruction}. Продолжи: {data.context}"
+        # ТВОЙ НОВЫЙ ПРОМПТ
+        prompt = (
+            f"System: {tone_instruction}. "
+            f"User input might be in wrong keyboard layout (e.g. 'ghbdtn' instead of 'привет'). "
+            f"If input looks like encoded Russian, decode it internally and complete the Russian phrase. "
+            f"If input is English, just complete it. "
+            f"Do not repeat the input. Output ONLY the completion text.\n\n"
+            f"User Input: {data.context}\nOutput:"
+        )
         
         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))