Update app.py

app.py

@@ -258,32 +258,31 @@ async def generate_image(
 ):
     async def generate_image_task():
         try:
-        # Save the uploaded image to a temporary file
-        temp_image_path = f"/tmp/{int(time.time())}_{image.filename}"
-        with open(temp_image_path, "wb") as temp_image:
-            temp_image.write(image.file.read())
-
-        # Open the uploaded image using PIL
-        control_image = Image.open(temp_image_path)
-
-        # Call existing inference function with the provided parameters
-        generated_image = inference(control_image, prompt, "", guidance_scale, controlnet_scale, 0, controlnet_end, upscaler_strength, seed, sampler_type)
-        if generated_image is None:
+            # Save the uploaded image to a temporary file
+            temp_image_path = f"/tmp/{int(time.time())}_{image.filename}"
+            with open(temp_image_path, "wb") as temp_image:
+                temp_image.write(image.file.read())
+
+            # Open the uploaded image using PIL
+            control_image = Image.open(temp_image_path)
+
+            # Call existing inference function with the provided parameters
+            generated_image = inference(control_image, prompt, "", guidance_scale, controlnet_scale, 0, controlnet_end, upscaler_strength, seed, sampler_type)
+            if generated_image is None:
+                return "Failed to generate image"
+
+            # Save the generated image as binary data
+            output_image_io = io.BytesIO()
+            generated_image.save(output_image_io, format="PNG")
+            output_image_io.seek(0)
+            
+            # Return the image as a streaming response
+            return StreamingResponse(content=output_image_io, media_type="image/png")
+
+        except Exception as e:
+            logger.error("Error occurred during image generation: %s", str(e))
             return "Failed to generate image"
 
-        # Save the generated image as binary data
-        output_image_io = io.BytesIO()
-        generated_image.save(output_image_io, format="PNG")
-        output_image_io.seek(0)
-        
-        # Return the image as a streaming response
-        return StreamingResponse(content=output_image_io, media_type="image/png")
-
-    except Exception as e:
-        logger.error("Error occurred during image generation: %s", str(e))
-        return "Failed to generate image"
-
-        pass
     try:
         await queue_manager.add_task(generate_image_task)
         position_in_queue = queue_manager.queue.qsize()
@@ -292,9 +291,7 @@ async def generate_image(
 
     except Exception as e:
         logger.error("Error occurred during image generation: %s", str(e))
-        return "Failed to add task to the queue"        
-
-
+        return "Failed to add task to the queue"
 
 async def start_fastapi():
     # Get internal IP address