Update app.py

app.py

@@ -26,7 +26,11 @@ def select_model(model_name):
         return API_BASE_URL + model_name
 
 def extend_prompt(input_text):
-    API_URL = "https://api-inference.huggingface.co/models/Gustavosta/MagicPrompt-Stable-Diffusion"
+    if not input_text.strip():
+        gr.Warning("Input text is empty!")
+        return None
+        
+    API_URL = API_BASE_URL + "Gustavosta/MagicPrompt-Stable-Diffusion"
     
     payload = {"inputs": input_text}
 
@@ -37,6 +41,10 @@ def extend_prompt(input_text):
     return extended_prompt
 
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):
+    if not input_text.strip():
+        gr.Warning("Input text is empty!")
+        return None
+        
     model_url = select_model(selected_model)
     API_URL = f"{model_url}"
 
@@ -80,15 +88,6 @@ with gr.Blocks(theme="soft") as playground:
     with gr.Accordion("Advanced settings", open=False):
         negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
 
-    gr.HTML(
-        """
-        <h2>Notes:</h2>
-        <p>
-          SDXL based models may take longer to generate than older models.
-        </p>
-        """
-    )
-
     gen_button.click(generate_image, inputs=[text_prompt, model_dropdown, negative_prompt], outputs=image_output)
     extend_button.click(extend_prompt, inputs=text_prompt, outputs=text_prompt)