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():