Update app.py

app.py

@@ -5,12 +5,15 @@ from PIL import Image
 import io
 import json
 
-def call_api_and_generate_image(api_key, negative_prompt, positive_prompt, controlnet_image):
+def call_api_and_generate_image(api_url, api_key, negative_prompt, positive_prompt, controlnet_image):
     if not api_key:
         return None, "Error: API key is required. Please provide a valid API key."
 
+    if not api_url:
+        return None, "Error: API URL is required. Please provide a valid API URL."
+
     resp = requests.post(
-        "https://model-5qe9kjp3.api.baseten.co/development/predict",
+        api_url,
         headers={"Authorization": f"Api-Key {api_key}"},
         json={
             'workflow_values': {
@@ -30,9 +33,9 @@ def call_api_and_generate_image(api_key, negative_prompt, positive_prompt, contr
     image = Image.open(io.BytesIO(image_data))
     return image, json.dumps(result, indent=2)
 
-def generate_image(api_key, negative_prompt, positive_prompt, controlnet_image):
+def generate_image(api_url, api_key, negative_prompt, positive_prompt, controlnet_image):
     try:
-        return call_api_and_generate_image(api_key, negative_prompt, positive_prompt, controlnet_image)
+        return call_api_and_generate_image(api_url, api_key, negative_prompt, positive_prompt, controlnet_image)
     except Exception as e:
         return None, f"Error: {str(e)}"
 
@@ -40,6 +43,7 @@ def generate_image(api_key, negative_prompt, positive_prompt, controlnet_image):
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
+        gr.Textbox(label="API URL", placeholder="Enter the API URL here", value="https://model-5qe9kjp3.api.baseten.co/development/predict"),
         gr.Textbox(label="API Key", placeholder="Enter your API key here"),
         gr.Textbox(label="Negative Prompt", value="blurry, text, low quality", placeholder="Enter negative prompt here..."),
         gr.Textbox(label="Positive Prompt", value="An igloo on a snowy day, 4k, hd", placeholder="Enter positive prompt here..."),
@@ -50,7 +54,7 @@ iface = gr.Interface(
         gr.Textbox(label="API Response JSON", lines=10)
     ],
     title="Image Generation with API",
-    description="Enter your API key and prompts to generate an image using the provided API."
+    description="Enter the API URL, your API key, and prompts to generate an image using the provided API."
 )
 
 # Launch the interface