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)