Create main.py

main.py

@@ -0,0 +1,29 @@
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from typing import Optional
+import random
+
+app = FastAPI()
+
+class ImageRequest(BaseModel):
+    prompt: str
+    width: Optional[int] = 720
+    height: Optional[int] = 1280
+    n: Optional[int] = 1
+
+@app.post("/v1/images/generations")
+async def generate_image(request: ImageRequest):
+    images = []
+    for _ in range(request.n):
+        seed = random.randint(0, 1000000)
+        url = (
+            f"https://image.pollinations.ai/prompt/{request.prompt}"
+            f"?width={request.width}&height={request.height}"
+            f"&seed={seed}&enhance=true&nologo=true&model=flux"
+        )
+        images.append({"url": url})
+
+    return {
+        "created": int(random.time()),
+        "data": images
+    }