fix path

server.py

@@ -6,6 +6,8 @@ from huggingface_hub import hf_hub_download, login
 from safetensors.torch import load_file
 from io import BytesIO
 import os
+import base64  # Added for encoding images as base64
+from typing import List  # Added for type hinting the list of prompts
 
 app = FastAPI()
 
@@ -74,6 +76,38 @@ async def generate_image(prompt: str):
     except Exception as e:
         return {"error": str(e)}
 
+# New endpoint to handle a list of prompts
+@app.get("/generate_multiple")
+async def generate_multiple_images(prompts: List[str]):
+    try:
+        # List to store base64-encoded images
+        generated_images = []
+        
+        # Generate an image for each prompt
+        for prompt in prompts:
+            image = pipe(
+                prompt,
+                num_inference_steps=4,
+                guidance_scale=0
+            ).images[0]
+            
+            # Save image to buffer
+            buffer = BytesIO()
+            image.save(buffer, format="PNG")
+            buffer.seek(0)
+            
+            # Encode the image as base64
+            image_base64 = base64.b64encode(buffer.getvalue()).decode("utf-8")
+            generated_images.append({
+                "prompt": prompt,
+                "image_base64": image_base64
+            })
+        
+        return {"images": generated_images}
+    
+    except Exception as e:
+        return {"error": str(e)}
+
 @app.get("/health")
 async def health_check():
     return {"status": "healthy"}