Update app.py

app.py

@@ -8,33 +8,31 @@ import urllib.request
 import re
 import time
 
-# Configuration des paramètres
+# Configuration des headers HTTP
 HEADERS = {
     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36',
     'Accept-Language': 'en-US,en;q=0.5'
 }
 
-# Fonction pour Bing
+# Téléchargement via Bing
 def download_bing_images(search_query, limit, adult_filter_off):
     try:
-        adult_filter = not adult_filter_off
+        # Ici, adult_filter_off est True si le mode sécurisé est désactivé ("Off")
         downloader.download(
             search_query,
             limit=limit,
-            adult_filter_off=adult_filter,
+            adult_filter_off=adult_filter_off,
             force_replace=False,
             timeout=60,
             filter_type='photo'
         )
-        
         output_dir = os.path.join('dataset', search_query)
         return get_image_paths(output_dir)
-    
     except Exception as e:
         print(f"Erreur Bing : {str(e)}")
         return []
 
-# Fonction pour Google (méthode sécurisée)
+# Téléchargement via Google (méthode sécurisée)
 def download_google_images(search_query, limit):
     try:
         output_dir = os.path.join('dataset', f'google_{search_query}')
@@ -62,42 +60,52 @@ def download_google_images(search_query, limit):
         print(f"Erreur Google : {str(e)}")
         return []
 
-# Téléchargement et sauvegarde des images
+# Téléchargement et sauvegarde des images depuis une liste d'URLs
 def download_and_save(urls, output_dir):
     saved_paths = []
     for idx, url in enumerate(urls):
         try:
             filename = f"image_{idx+1}_{int(time.time())}.jpg"
             full_path = os.path.join(output_dir, filename)
-            
             req = urllib.request.Request(url, headers=HEADERS)
             with urllib.request.urlopen(req, timeout=10) as response:
                 with open(full_path, 'wb') as f:
                     f.write(response.read())
-                    saved_paths.append(full_path)
+            saved_paths.append(full_path)
         except Exception as e:
             print(f"Erreur téléchargement {url} : {str(e)}")
     return saved_paths
 
+# Récupérer les chemins des images dans un dossier
 def get_image_paths(directory):
     if os.path.exists(directory):
-        return [os.path.join(directory, f) for f in os.listdir(directory) 
+        return [os.path.join(directory, f) for f in os.listdir(directory)
                 if f.lower().endswith(('png', 'jpg', 'jpeg'))]
     return []
 
-# Fonction principale
+# Fonction principale appelée par l'interface Gradio
 def download_handler(source, query, limit, safe_mode):
-    limit = max(1, min(limit, 100))  # Limite entre 1-100
+    # S'assurer que le nombre d'images est entre 1 et 100
+    limit = max(1, min(limit, 100))
     try:
         if source == "Bing":
-            return download_bing_images(query, limit, safe_mode == "Off")
+            # Si safe_mode est "Off", alors le filtre est désactivé (adult_filter_off=True)
+            image_paths = download_bing_images(query, limit, safe_mode == "Off")
         elif source == "Google":
-            return download_google_images(query, limit)
+            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."
+        return image_paths, status_msg
     except Exception as e:
         print(f"Erreur globale : {str(e)}")
-        return []
+        return [], f"Erreur: {str(e)}"
 
-# Interface Gradio améliorée
+# Création de l'interface Gradio avec Blocks
 with gr.Blocks(theme=gr.themes.Soft(), title="Image Downloader") as app:
     gr.Markdown("# 📸 Téléchargeur d'Images Multi-Sources")
     gr.Markdown("Téléchargez des images depuis Bing ou Google (max 100)")
@@ -110,15 +118,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Image Downloader") as app:
     
     submit_btn = gr.Button("🚀 Lancer le téléchargement", variant="primary")
     
-    gallery = gr.Gallery(
-        label="Résultats",
-        columns=5,
-        object_fit="contain",
-        height="auto"
-    )
-    
+    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],
@@ -135,4 +138,5 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Image Downloader") as app:
     )
 
 if __name__ == "__main__":
-    app.launch(server_port=7860, show_error=True)
+    # 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)