Upload 3 files

Dockerfile

@@ -0,0 +1,6 @@
+FROM python:3.9
+WORKDIR /code
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+COPY . .
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,36 @@
+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:
+        raise HTTPException(status_code=500, detail=str(e))

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+google-generativeai
+pydantic