Create app.py

app.py

@@ -0,0 +1,59 @@
+from fastapi import FastAPI, File, UploadFile, Form
+import requests
+from gradio_client import Client, handle_file
+import shutil
+import os
+
+app = FastAPI()
+
+client = Client("yanze/PuLID-FLUX")
+BASE_URL = "https://yanze-pulid-flux.hf.space/file="
+
+@app.post("/generate_image")
+async def generate_image(
+    prompt: str = Form(...),
+    image_url: str = Form(None),
+    image_file: UploadFile = File(None)
+):
+    # Determine input image
+    id_image = None
+    if image_url:
+        id_image = handle_file(image_url)
+    elif image_file:
+        file_location = f"temp_{image_file.filename}"
+        with open(file_location, "wb") as buffer:
+            shutil.copyfileobj(image_file.file, buffer)
+        id_image = handle_file(file_location)
+        os.remove(file_location)
+    
+    try:
+        result = client.predict(
+            prompt=prompt,
+            id_image=id_image,
+            start_step=0,
+            guidance=4,
+            seed="-1",
+            true_cfg=1,
+            width=896,
+            height=1152,
+            num_steps=20,
+            id_weight=1,
+            neg_prompt="bad quality, worst quality, text, signature, watermark, extra limbs",
+            timestep_to_start_cfg=1,
+            max_sequence_length=128,
+            api_name="/generate_image"
+        )
+
+        file_path = result[0]
+        full_url = f"{BASE_URL}{file_path}"
+
+        # Download and save image
+        response = requests.get(full_url)
+        if response.status_code == 200:
+            with open("output_image.jpg", "wb") as f:
+                f.write(response.content)
+            return {"message": "Image saved", "url": full_url}
+        
+        return {"message": "Image generated", "url": full_url}
+    except Exception as e:
+        return {"error": str(e)}