Update app.py

app.py

@@ -1,5 +1,5 @@
-import gradio as gr
 import yt_dlp
+import gradio as gr
 import os
 from glob import glob
 
@@ -7,52 +7,49 @@ def convert_time_to_seconds(time_str):
     hours, minutes, seconds = map(int, time_str.split(':'))
     return hours * 3600 + minutes * 60 + seconds
 
-def download_videos(text_input):
-    lines = text_input.strip().split('\n')
-    output_files = []
-    
-    for line in lines:
-        parts = line.split(',')
-        if len(parts) == 3:
-            video_url = parts[0].strip()
-            start_time = parts[1].strip()
-            end_time = parts[2].strip()
-
-            start_seconds = convert_time_to_seconds(start_time)
-            end_seconds = convert_time_to_seconds(end_time)
-
-            options = {
-                'format': 'bestaudio/best',
-                'postprocessors': [{
-                    'key': 'FFmpegExtractAudio',
-                    'preferredcodec': 'mp3',
-                    'preferredquality': '128',
-                }],
-                'outtmpl': '%(title)s.%(ext)s',
-                'postprocessor_args': [
-                    '-ss', str(start_seconds),
-                    '-to', str(end_seconds)
-                ]
-            }
-
-            with yt_dlp.YoutubeDL(options) as ydl:
-                ydl.download([video_url])
-
-            download_files = glob('*.mp3')
-            for file in download_files:
-                os.rename(file, os.path.join('downloaded', file))
-                output_files.append(os.path.join('downloaded', file))
-
-    if output_files:
-        return output_files
-    return "No MP3 file found."
-
-interface = gr.Interface(
-    fn=download_videos,
-    inputs=gr.Textbox(label="Enter video URL, start time, end time per line (comma separated)", placeholder="URL, Start Time (HH:MM:SS), End Time (HH:MM:SS)\nExample:\nhttps://youtu.be/xxx, 00:00:00, 00:05:00", lines=10),
-    outputs=gr.File(label="Downloaded MP3 Files"),
-    title="Batch YouTube Video Downloader",
-    description="Enter each video's URL, start time, and end time on a new line, separated by commas."
-)
-
-interface.launch()
+def download_video(video_url, start_time, end_time):
+    start_seconds = convert_time_to_seconds(start_time)
+    end_seconds = convert_time_to_seconds(end_time)
+    options = {
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '128',
+        }],
+        'outtmpl': '%(title)s.%(ext)s',
+        'postprocessor_args': [
+            '-ss', str(start_seconds),
+            '-to', str(end_seconds)
+        ]
+    }
+
+    with yt_dlp.YoutubeDL(options) as ydl:
+        ydl.download([video_url])
+
+    # 查找和提供下载的文件
+    download_files = glob('*.mp3')
+    if download_files:
+        return f"Download successful: {download_files[0]}", download_files[0]
+    return "No MP3 file found.", None
+
+def setup_interface():
+    interface = gr.Interface(
+        fn=download_video,
+        inputs=[
+            gr.Textbox(lines=2, placeholder="Enter YouTube video URL here...", label="YouTube Video URL"),
+            gr.Textbox(placeholder="Start Time (e.g., 00:00:00)", label="Start Time (HH:MM:SS)"),
+            gr.Textbox(placeholder="End Time (e.g., 00:05:30)", label="End Time (HH:MM:SS)")
+        ],
+        outputs=[
+            gr.Text(label="Status Message"),
+            gr.File(label="Download MP3")
+        ],
+        title="YouTube Video Downloader",
+        description="Enter YouTube video URL and specify start and end times to download audio as MP3."
+    )
+    return interface
+
+if __name__ == "__main__":
+    iface = setup_interface()
+    iface.launch()