Update app.py

app.py

@@ -78,4 +78,56 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None,
         
         # Check if the image is completely black
         black = Image.new('RGB', image.size, (0, 0, 0))
-        if ImageChops.difference(image, black).get
+        if ImageChops.difference(image, black).getbbox() is None:
+            return None, 'The image is completely black. There may be a parameter that cannot be specified, or an error may have occurred internally.'
+
+        return image, None
+
+    except Exception as e:
+        print("Error generating image:", e)
+        return None, f"Error generating image: {e}"
+
+@app.route('/', methods=['GET'])
+def home():
+    prompt = request.args.get('prompt', '')
+    model = request.args.get('model', '')
+    negative_prompt = request.args.get('Nprompt', None)
+    noise = request.args.get('noise', None)
+    cfg_scale = request.args.get('cfg_scale', None)
+    num_inference_steps = request.args.get('steps', None)
+    
+    try:
+        if cfg_scale is not None:
+            cfg_scale = float(cfg_scale)
+    except ValueError:
+        return 'Invalid "cfg_scale" parameter. It should be a number.', 400
+
+    try:
+        if num_inference_steps is not None:
+            num_inference_steps = int(num_inference_steps)
+    except ValueError:
+        return 'Invalid "steps" parameter. It should be an integer.', 400
+    
+    if not model:
+        return 'Please provide a "model" query parameter in the URL.', 400
+
+    if not prompt:
+        return 'Please provide a "prompt" query parameter in the URL.', 400
+    
+    # Generate the image
+    image, error_message = gen_fn(model, prompt, negative_prompt, noise, cfg_scale, num_inference_steps)
+    if error_message:
+        return error_message, 400
+    
+    if isinstance(image, Image.Image):  # Ensure the result is a PIL image
+        # Save image to BytesIO object
+        img_io = BytesIO()
+        image.save(img_io, format='PNG')
+        img_io.seek(0)
+        return send_file(img_io, mimetype='image/png', as_attachment=False)
+    
+    return 'Failed to generate image.', 500
+
+if __name__ == '__main__':
+    # Launch Flask app
+    app.run(host='0.0.0.0', port=7860)  # Run Flask app