Update app.py

app.py

@@ -1,31 +1,35 @@
 import gradio as gr
-import speedtest
+from pytube import YouTube
+import os
 
-def run_speedtest():
-    st = speedtest.Speedtest()
-    st.get_best_server()  # Selecciona el mejor servidor basado en la latencia
-    download_speed = st.download()
-    upload_speed = st.upload()
-    ping = st.results.ping
-
-    # Convertir de bits por segundo a megabits por segundo y redondear
-    download_mbps = round(download_speed / 1e6, 2)
-    upload_mbps = round(upload_speed / 1e6, 2)
-    
-    result_text = (
-        f"**Resultados del Speedtest:**\n\n"
-        f"Descarga: **{download_mbps} Mbps**\n"
-        f"Subida: **{upload_mbps} Mbps**\n"
-        f"Ping: **{ping} ms**"
-    )
-    return result_text
+def download_youtube_video(url):
+    try:
+        # Crea el objeto YouTube a partir de la URL
+        yt = YouTube(url)
+        # Selecciona el stream progresivo con extensión mp4, ordenado de mayor a menor resolución
+        stream = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
+        if stream is None:
+            return "No se encontró un stream adecuado para descargar."
+        
+        # Define un directorio de descarga y asegúrate de que exista
+        download_folder = "downloads"
+        os.makedirs(download_folder, exist_ok=True)
+        
+        # Descarga el video en el directorio especificado
+        output_path = stream.download(output_path=download_folder)
+        return f"Video descargado: {output_path}"
+    except Exception as e:
+        return f"Error al descargar el vídeo: {str(e)}"
 
+# Crea la interfaz Gradio
 iface = gr.Interface(
-    fn=run_speedtest,
-    inputs=[],
-    outputs="markdown",
-    title="SpeedTest de Internet",
-    description="Ejecuta un test de velocidad de Internet usando speedtest-cli y muestra los resultados en una interfaz interactiva."
+    fn=download_youtube_video,
+    inputs=gr.Textbox(label="URL del vídeo de YouTube", placeholder="Ingresa la URL aquí..."),
+    outputs=gr.Textbox(label="Resultado"),
+    title="Descargador de Vídeos de YouTube",
+    description="Ingresa la URL de un vídeo de YouTube y descarga la versión en MP4 de mayor resolución disponible."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()
+