� Patch gestion erreurs + timeouts pour stabilité Hugging Face

app.py

@@ -1,8 +1,7 @@
 # app.py – Hugging Face Spaces avec liste dynamique depuis EC2
 
-import sys
-import os
 import requests
+from requests.exceptions import RequestException
 import gradio as gr
 import base64
 from PIL import Image
@@ -30,37 +29,46 @@ def segment_image(image: Image.Image):
         image.save(buffered, format="PNG")
         buffered.seek(0)
         files = {"file": ("input.png", buffered, "image/png")}
-        response = requests.post(API_URL, files=files)
+
+        response = requests.post(API_URL, files=files, timeout=10)
         response.raise_for_status()
+
         data = response.json()
         mask_bytes = base64.b64decode(data["mask_base64"])
         mask_image = Image.open(io.BytesIO(mask_bytes)).convert("RGB")
         overlay_img = overlay_mask_on_image(image, mask_image)
         return image, mask_image, overlay_img, f"{data['inference_time']} sec"
+
+    except RequestException as e:
+        print(f"[ERREUR API EC2] {e}")
+        return None, None, None, "Erreur : EC2 injoignable ou lente"
+
     except Exception as e:
+        print(f"[ERREUR GRADIO] {e}")
         return None, None, None, f"Erreur : {str(e)}"
 
 def get_remote_image_names():
     try:
-        response = requests.get(IMAGE_LIST_URL)
+        response = requests.get(IMAGE_LIST_URL, timeout=5)
         response.raise_for_status()
-        return response.json()["files"]
-    except:
+        return response.json().get("files", [])
+    except Exception as e:
+        print(f"[ERREUR LISTE FICHIERS] {e}")
         return ["Erreur lors du chargement des noms"]
 
 def load_image_from_url(filename):
     try:
         url = IMAGE_BASE_URL + filename
-        response = requests.get(url)
+        response = requests.get(url, timeout=5)
         response.raise_for_status()
         return Image.open(io.BytesIO(response.content)).convert("RGB")
     except Exception as e:
-        print(f"Erreur chargement image URL {filename} : {e}")
+        print(f"[ERREUR CHARGEMENT IMAGE {filename}] {e}")
         return None
 
 with gr.Blocks(title="Segmentation d'Images Urbaines") as demo:
     gr.Markdown("# 🧠 Segmentation d'Images Urbaines")
-    gr.Markdown("Upload une image ou sélectionne un exemple distant hébergé sur EC2 (33 images dispo)")
+    gr.Markdown("Upload une image ou sélectionne une image distante depuis EC2")
 
     with gr.Row():
         input_image = gr.Image(type="pil", label="Image d'entrée")
@@ -82,7 +90,5 @@ with gr.Blocks(title="Segmentation d'Images Urbaines") as demo:
     with gr.Accordion("🎨 Légende des classes", open=False):
         gr.Image(value=legend_img, label="Légende (fictive ici)", interactive=False)
 
-
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()