Update app.py

app.py

@@ -31,22 +31,16 @@ def extend_prompt(input_text):
  gr.Warning("Input text is empty!")
  return None
  
- API_URL = API_BASE_URL + "Gustavosta/MagicPrompt-Stable-Diffusion"
+ API_URL = f"{API_BASE_URL}Gustavosta/MagicPrompt-Stable-Diffusion"
  
  payload = {"inputs": input_text}
 
  try:
- response = requests.post(API_URL, headers=HEADERS, json=payload)
- response.raise_for_status()
-
- output = response.json()[0]
- 
- prompt = output.get("generated_text", "")
- return prompt
- 
+ response = requests.post(API_URL, headers=HEADERS, json=payload).json()
+ return response[0].get("generated_text", "")
  except requests.exceptions.RequestException as e:
- gr.Warning("Cannot extend prompt: Please try again.")
- return input_text
+ gr.Error(f"Error in API request: {e}")
+ return None
 
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):
  if not prompt.strip():