PodMagic / app.py
Ribot's picture
Update app.py
09d051e verified
raw
history blame
3.03 kB
import gradio as gr
import requests
import re
import os
import zipfile
import tempfile
from urllib.parse import urljoin
def process_url(url):
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'}
response = requests.get(url, headers=headers)
response.raise_for_status()
except Exception as e:
return None, f"Erreur de connexion : {str(e)}"
# Extraction des URLs MP3
mp3_links = []
pattern = r'(?:contentUrl|url)"\s*:\s*"([^"]+?\.mp3)'
matches = re.findall(pattern, response.text)
for match in matches:
clean_url = match.split('";')[0] if '";' in match else match
absolute_url = urljoin(response.url, clean_url)
if absolute_url not in mp3_links:
mp3_links.append(absolute_url)
if not mp3_links:
return None, "Aucun lien MP3 détecté dans le code source"
# Téléchargement
temp_dir = tempfile.mkdtemp()
filenames = []
for idx, mp3_url in enumerate(mp3_links, 1):
try:
filename = f"{idx:02d}_{os.path.basename(mp3_url).split('?')[0]}"
filepath = os.path.join(temp_dir, filename)
with requests.get(mp3_url, headers=headers, stream=True) as r:
r.raise_for_status()
with open(filepath, 'wb') as f:
for chunk in r.iter_content(chunk_size=8192):
if chunk:
f.write(chunk)
filenames.append(filepath)
except Exception as e:
continue
if not filenames:
return None, "Échec des téléchargements"
# Création ZIP
zip_path = os.path.join(temp_dir, 'podcast.zip')
with zipfile.ZipFile(zip_path, 'w') as zipf:
for file in filenames:
zipf.write(file, arcname=os.path.basename(file))
return zip_path, None
def download_podcast(url):
zip_path, error = process_url(url)
if error:
raise gr.Error(error)
return zip_path
# Configuration Gradio compatible Hugging Face
with gr.Blocks() as app:
gr.Markdown("# 📻 Téléchargeur de Podcasts Radio France")
with gr.Row():
url_input = gr.Textbox(
label="URL de la série podcast",
placeholder="Collez l'URL Radio France ici..."
)
download_btn = gr.Button("Télécharger les épisodes")
output_file = gr.File(label="Fichier ZIP à télécharger")
error_output = gr.Textbox(visible=False)
examples = gr.Examples(
examples=[[
"https://www.radiofrance.fr/franceculture/podcasts/serie-le-secret-de-la-licorne-les-aventures-de-tintin"
]],
inputs=[url_input]
)
download_btn.click(
fn=download_podcast,
inputs=url_input,
outputs=output_file,
api_name="download"
)
# Configuration spécifique pour Hugging Face
app.launch(debug=False, show_error=True)