Update app.py

app.py

@@ -1,13 +1,28 @@
+from flask import Flask, request, jsonify
 from gradio_client import Client
 import gradio as gr
 
-# Create a client for the other space
+app = Flask(__name__)
+
 client = Client("radames/Enhance-This-HiDiffusion-SDXL")
 
-# Define your interface function
+@app.route('/predict', methods=['POST'])
+def my_interface():
+    data = request.get_json()
+
+    input_image = data['input_image']
+    prompt = data.get('prompt', "This is a beautiful scenery")
+    negative_prompt = data.get('negative_prompt', "blurry, ugly, duplicate, poorly drawn, deformed, mosaic")
+    seed = data.get('seed', 1415926535897932)
+    guidance_scale = data.get('guidance_scale', 8.5)
+    scale = data.get('scale', 2)
+    controlnet_conditioning_scale = data.get('controlnet_conditioning_scale', 0.5)
+    strength = data.get('strength', 1.0)
+    controlnet_start = data.get('controlnet_start', 0.0)
+    controlnet_end = data.get('controlnet_end', 1.0)
+    guassian_sigma = data.get('guassian_sigma', 2.0)
+    intensity_threshold = data.get('intensity_threshold', 3)
 
-def my_interface(input_image, prompt="This is a beautiful scenery", negative_prompt="blurry, ugly, duplicate, poorly drawn, deformed, mosaic", seed=1415926535897932, guidance_scale=8.5, scale=2, controlnet_conditioning_scale=0.5, strength=1.0, controlnet_start=0.0, controlnet_end=1.0, guassian_sigma=2.0, intensity_threshold=3):
-    # Call the other space's predict function
     result = client.predict(
         input_image=input_image, 
         prompt=prompt, 
@@ -23,13 +38,8 @@ def my_interface(input_image, prompt="This is a beautiful scenery", negative_pro
         intensity_threshold=intensity_threshold, 
         api_name="/predict"
     )
-    # Return only the first image from the result
-    return result[0][0]
 
-# Define your Gradio interface
-iface = gr.Interface(fn=my_interface, 
-                     inputs=gr.Image(), 
-                     outputs=gr.Image())
+    return jsonify(result[0][0])
 
-# Launch your Gradio interface
-iface.launch()
+if __name__ == '__main__':
+    app.run(debug=True)