Update app.py

app.py

@@ -1,21 +1,25 @@
-# Let's re-evaluate the implementation using the guide for making predictions with the Gradio client
 import gradio as gr
-from gradio.client import Client
+from gradio_client import Client
 
-# Define the function to call the existing Gradio space API using the Gradio client
-def enhance_image(input_image):
-    client = Client("radames/Enhance-This-HiDiffusion-SDXL")
-    result = client.predict(input_image, api_name="/predict")
+# Initialize the client
+client = Client("radames/Enhance-This-HiDiffusion-SDXL")
+
+def enhance_image(image, prompt):
+    # Use the client to predict the result
+    result = client.predict(image, prompt, api_name="/predict")
     return result
 
 # Create the Gradio interface
-interface = gr.Interface(
+iface = gr.Interface(
     fn=enhance_image,
-    inputs=gr.Image(type="pil", label="Input Image"),
+    inputs=[
+        gr.Image(type="pil", label="Input Image"),
+        gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
+    ],
     outputs=gr.Image(type="pil", label="Enhanced Image"),
-    title="Enhance This HiDiffusion SDXL",
-    description="Upload an image to enhance it using HiDiffusion SDXL."
+    title="Image Enhancement with Text Prompt",
+    description="Upload an image and provide a text prompt to enhance the image using HiDiffusion SDXL."
 )
 
-# Launch the app
-interface.launch()
+# Launch the interface
+iface.launch()