Update app.py

app.py

@@ -32,7 +32,7 @@ def download_bing_images(search_query, limit, adult_filter_off):
         print(f"Erreur Bing : {str(e)}")
         return []
 
-# Téléchargement via Google (méthode sécurisée)
+# Téléchargement via Google
 def download_google_images(search_query, limit):
     try:
         output_dir = os.path.join('dataset', f'google_{search_query}')
@@ -85,21 +85,17 @@ def get_image_paths(directory):
 
 # Fonction principale appelée par l'interface Gradio
 def download_handler(source, query, limit, safe_mode):
-    # S'assurer que le nombre d'images est entre 1 et 100
     limit = max(1, min(limit, 100))
     try:
         if source == "Bing":
-            # Si safe_mode est "Off", alors le filtre est désactivé (adult_filter_off=True)
+            # Si safe_mode est "Off", le filtre est désactivé (adult_filter_off=True)
             image_paths = download_bing_images(query, limit, safe_mode == "Off")
         elif source == "Google":
             image_paths = download_google_images(query, limit)
         else:
             image_paths = []
         
-        if image_paths:
-            status_msg = f"{len(image_paths)} image(s) téléchargée(s)."
-        else:
-            status_msg = "Aucune image téléchargée."
+        status_msg = f"{len(image_paths)} image(s) téléchargée(s)." if image_paths else "Aucune image téléchargée."
         return image_paths, status_msg
     except Exception as e:
         print(f"Erreur globale : {str(e)}")
@@ -121,7 +117,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Image Downloader") as app:
     gallery = gr.Gallery(label="Résultats", columns=5, object_fit="contain", height="auto")
     status = gr.Textbox(label="Statut", interactive=False)
 
-    # Liaison du bouton au gestionnaire de téléchargement (2 sorties attendues)
     submit_btn.click(
         fn=download_handler,
         inputs=[source, query, limit, safe_mode],
@@ -138,5 +133,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Image Downloader") as app:
     )
 
 if __name__ == "__main__":
-    # Utilisation de server_name="0.0.0.0" pour Hugging Face Spaces
-    app.launch(server_name="0.0.0.0", server_port=7860, show_error=True)
+    # Récupère le port depuis la variable d'environnement (nécessaire pour Hugging Face Spaces)
+    port = int(os.environ.get("PORT", 7860))
+    app.launch(server_name="0.0.0.0", server_port=port, show_error=True)
+