Update app.py

app.py

@@ -7,7 +7,7 @@ import tempfile
 from urllib.parse import urljoin
 from bs4 import BeautifulSoup
 
-def process_url(url):
+def process_url(url, num_episodes):
     try:
         headers = {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
@@ -16,12 +16,12 @@ def process_url(url):
         response = requests.get(url, headers=headers)
         response.raise_for_status()
     except Exception as e:
-        return None, f"Erreur de connexion : {str(e)}"
+        return None, f"Erreur : {str(e)}"
 
     soup = BeautifulSoup(response.text, 'html.parser')
     mp3_links = []
 
-    # Nouvelle méthode de détection
+    # Extraction des liens MP3
     scripts = soup.find_all('script', type='application/ld+json')
     for script in scripts:
         if script.string:
@@ -31,27 +31,23 @@ def process_url(url):
                 if full_url not in mp3_links:
                     mp3_links.append(full_url)
 
-    # Fallback pour les URLs dans les attributs data
+    # Fallback si nécessaire
     if not mp3_links:
-        for tag in soup.find_all(attrs={"data-url": re.compile(r"\.mp3")}):
-            mp3_url = urljoin(url, tag['data-url'].split('?')[0])
-            mp3_links.append(mp3_url)
+        matches = re.findall(r'(https?://media\.radiofrance-podcast\.net[^\s"\']+?\.mp3)', response.text)
+        mp3_links = list(dict.fromkeys(matches))
 
-    # Dernier recours : recherche globale
-    if not mp3_links:
-        matches = re.findall(r'(https?://[^\s"\']+?\.mp3)', response.text)
-        for match in matches:
-            clean_url = urljoin(url, match.split('?')[0])
-            if clean_url not in mp3_links and 'podcast' in clean_url:
-                mp3_links.append(clean_url)
-
-    # Filtrage final
-    mp3_links = list(dict.fromkeys(mp3_links))  # Supprime les doublons
+    # Application du nombre d'épisodes demandé
+    try:
+        num_episodes = int(num_episodes)
+        if num_episodes > 0:
+            mp3_links = mp3_links[:num_episodes]
+    except:
+        pass  Si valeur invalide, on prend tout
 
     if not mp3_links:
-        return None, "Aucun épisode trouvé - Structure de page inconnue"
+        return None, "Aucun épisode trouvé"
 
-    # Téléchargement et création ZIP
+    # Téléchargement
     temp_dir = tempfile.mkdtemp()
     filenames = []
     
@@ -72,6 +68,7 @@ def process_url(url):
     if not filenames:
         return None, "Échec du téléchargement"
 
+    # Création du ZIP
     zip_path = os.path.join(temp_dir, 'podcast.zip')
     with zipfile.ZipFile(zip_path, 'w') as zipf:
         for file in filenames:
@@ -79,33 +76,43 @@ def process_url(url):
 
     return zip_path, None
 
-def download_podcast(url):
-    zip_path, error = process_url(url)
+def download_podcast(url, num_episodes):
+    zip_path, error = process_url(url, num_episodes)
     if error:
         raise gr.Error(error)
     return zip_path
 
 with gr.Blocks() as app:
-    gr.Markdown("## 🎧 Téléchargeur Radio France")
+    gr.Markdown("## 🎧 Téléchargeur Radio France - Contrôle des épisodes")
+    
     with gr.Row():
         url_input = gr.Textbox(
             label="URL du podcast",
             placeholder="Ex: https://www.radiofrance.fr/...",
             max_lines=1
         )
-    btn = gr.Button("Télécharger les épisodes", variant="primary")
-    output = gr.File(label="Fichier ZIP")
+        num_input = gr.Number(
+            label="Nombre d'épisodes à télécharger (0 = tous)",
+            value=0,
+            minimum=0,
+            step=1,
+            precision=0
+        )
+    
+    btn = gr.Button("Télécharger", variant="primary")
+    output = gr.File(label="Fichier ZIP résultant")
     
     examples = gr.Examples(
         examples=[[
-            "https://www.radiofrance.fr/franceculture/podcasts/serie-le-secret-de-la-licorne-les-aventures-de-tintin"
+            "https://www.radiofrance.fr/franceculture/podcasts/serie-le-secret-de-la-licorne-les-aventures-de-tintin",
+            4
         ]],
-        inputs=[url_input]
+        inputs=[url_input, num_input]
     )
 
     btn.click(
         fn=download_podcast,
-        inputs=url_input,
+        inputs=[url_input, num_input],
         outputs=output,
         api_name="download"
     )