Update app.py

app.py

@@ -1,29 +1,28 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import pipeline
-import os
 import asyncio
 from contextlib import asynccontextmanager
 
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    global classifier
-    classifier = pipeline("sentiment-analysis")
-    yield
+class ImageRequest(BaseModel):  # Новая модель для запроса
+    prompt: str
+    model: str = "flux_1_schnell"
+    size: str = "1_1_HD"
+    is_public: bool = False
 
-app = FastAPI(lifespan=lifespan)
-
-class TextRequest(BaseModel):
-    text: str
-
-@app.post("/analyze")
-async def analyze(request: TextRequest):
+@app.post("/generate-image")
+async def generate_image(request: ImageRequest):
+    url = "https://api.fastflux.co/v1/images/generate"
+    headers = {"Content-Type": "application/json"}
+    json_data = {
+        "prompt": request.prompt,
+        "model": request.model,
+        "size": request.size,
+        "isPublic": request.is_public
+    }
+    
     try:
-        result = classifier(request.text)[0]
-        return {
-            "text": request.text,
-            "sentiment": result['label'],
-            "confidence": round(result['score'], 4)
-        }
+        response = requests.post(url, headers=headers, json=json_data)
+        response.raise_for_status()
+        return response.json()
     except Exception as e:
         return {"error": str(e)}