Update app.py

app.py

@@ -201,27 +201,17 @@ def generate_image_from_parameters(prompt, guidance_scale, controlnet_scale, con
         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)
-
-        # Specify the desired output directory for saving generated images
-        output_directory = "/home/user/app/generated_files"
-
-        # Create the output directory if it doesn't exist
-        os.makedirs(output_directory, exist_ok=True)
-
-        # Generate a unique filename for the saved image
-        filename = f"generated_image_{int(time.time())}.png"
-
-        # Save the generated image to the permanent location
-        output_path = os.path.join(output_directory, filename)
-        generated_image.save(output_path, format="PNG")
+        generated_image, _, _, _ = inference(control_image, prompt, "", guidance_scale, controlnet_scale, 0, controlnet_end, upscaler_strength, seed, sampler_type)
 
-        # Open the saved image file and read its binary data
-        with open(output_path, "rb") as image_file:
-            image_data = image_file.read()
+        # 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)
+        output_image_binary = output_image_io.read()
 
-        # Return the image data to the client
-        return image_data
+        # Return the generated image binary data
+        logger.debug("Output Values: generated_image=<binary data>")
+        return output_image_binary
 
     except Exception as e:
         # Handle exceptions and return an error message if something goes wrong