Update app.py

app.py

@@ -1,29 +1,25 @@
-from fastapi import FastAPI
-from gradio_client import Client, handle_file
+from fastapi import FastAPI, Query
+import subprocess
+import json
 
 app = FastAPI()
 
-@app.post("/generate_image")
-async def generate_image():
-    client = Client("yanze/PuLID-FLUX")
-    result = client.predict(
-        prompt="portrait, color, cinematic",
-        id_image=handle_file('https://d3lzcn6mbbadaf.cloudfront.net/media/details/ANI-20241109170223.jpg'),
-        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"
-    )
-    # The result is a tuple containing:
-    # [0] filepath (str): the generated image path,
-    # [1] used seed (str): the seed value used,
-    # [2] output gallery (list of dicts): additional output information.
-    return {"result": result}
+@app.get("/generate")
+async def generate(
+    prompt: str, 
+    image_url: str, 
+    format: str = "landscape", 
+    numberofimages: int = 1
+):
+    # Ensure number of images doesn't exceed the limit
+    if numberofimages > 5:
+        numberofimages = 5
+
+    # Run the Python script
+    command = f'python flux1.py --prompt "{prompt}" --image_url "{image_url}" --format {format} --numberofimages {numberofimages}'
+    result = subprocess.run(command, shell=True, capture_output=True, text=True)
+    
+    # Extract URLs from the output
+    urls = [line.split(": ")[1].strip() for line in result.stdout.split("\n") if "Generated Image" in line]
+    
+    return {"urls": urls}