Create app.py

app.py

@@ -0,0 +1,43 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel, Field
+from typing import List
+
+from shashwat_work.main import get_images
+
+app = FastAPI(title="Images API", version="1.0.0")
+
+# CORS (open by default; tighten if needed)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"],
+)
+class ImageItem(BaseModel):
+    ref_url: str
+    num: int
+
+class GenerateRequest(BaseModel):
+    items: List[ImageItem]
+
+class ImageResult(BaseModel):
+    ref_url: "str"
+    urls: List["str"]
+
+class GenerateResponse(BaseModel):
+    urls: List[ImageResult]
+
+@app.get("/health")
+def health():
+    return {"ok": True}
+
+@app.post("/generate", response_model=GenerateResponse)
+def generate(req: GenerateRequest):
+    try:
+        urls = get_images(req.items)
+        # Basic sanity check
+        if not isinstance(urls, list):
+            raise ValueError("get_images must return a List[str]")
+        return {"urls": urls}
+    except Exception as e:
+        # Log in real app; return sanitized error
+        raise HTTPException(status_code=500, detail=str(e))