Update app.py

app.py

@@ -34,11 +34,15 @@ def extend_prompt(input_text):
     
     payload = {"inputs": input_text}
 
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
-    response.raise_for_status()
+    try:
+        response = requests.post(API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
 
-    extended_prompt = response.content
-    return str(extended_prompt)
+        extended_prompt = response.content
+        return str(extended_prompt)
+        
+    except requests.exceptions.RequestException as e:
+        return None
 
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):
     if not prompt.strip():
@@ -55,12 +59,16 @@ def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale
         "seed": seed if seed is not None else random.randint(-1, 2147483647)
     }
 
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
-    response.raise_for_status()
+    try:
+        response = requests.post(API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
     
-    image_bytes = io.BytesIO(response.content)
-    image = Image.open(image_bytes)
-    return image
+        image_bytes = io.BytesIO(response.content)
+        image = Image.open(image_bytes)
+        return image
+        
+    except requests.exceptions.RequestException as e:
+        raise gr.Error(e)
 
 with gr.Blocks(theme="soft") as playground:
     gr.HTML(