Update app.py

app.py

@@ -136,36 +136,36 @@ def upload_image(url):
 #         raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
 
 @app.post("/generate-image")
-async def generate_image(request: ImageRequest):
-    print("Image Generation Request", request)
-    try:
-         runware = Runware(api_key=RUNWARE_API_KEY)
-         await runware.connect()
+    async def generate_image(request: ImageRequest):
+        print("Image Generation Request", request)
+        try:
+            runware = Runware(api_key=RUNWARE_API_KEY)
+            await runware.connect()
         
-         request_image = IImageInference(
-             positivePrompt=request.prompt,
-             model=request.model,
-             numberResults=request.number_results,
-             height=request.height,
-             width=request.width,
-         )
+            request_image = IImageInference(
+                 positivePrompt=request.prompt,
+                 model=request.model,
+                 numberResults=request.number_results,
+                 height=request.height,
+                 width=request.width,
+             )
         
-         images = await runware.imageInference(requestImage=request_image)
-         image_urls = [image.imageURL for image in images]
-         print("Generated Images: ", image_urls)
+            images = await runware.imageInference(requestImage=request_image)
+            image_urls = [image.imageURL for image in images]
+            print("Generated Images: ", image_urls)
         
-         response = {
-             "batch": {
-                 "prompt": request.prompt,
-                 "width": request.width,
-                 "height": request.height,
-                 "model": request.model,
-                 "images": [{"url": url} for url in image_urls]
+            response = {
+                 "batch": {
+                     "prompt": request.prompt,
+                     "width": request.width,
+                     "height": request.height,
+                     "model": request.model,
+                     "images": [{"url": url} for url in image_urls]
+                 }
              }
-         }
-         return response
-     except Exception as e:
-         raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
+            return response
+        except Exception as e:
+            raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
 
 @app.get("/get-batches")
 async def get_batches():