Update app.py

app.py

@@ -1,9 +1,15 @@
-import gradio as gr
 import yt_dlp
+import gradio as gr
 import os
 from glob import glob
 
+def convert_time_to_seconds(time_str):
+    hours, minutes, seconds = map(int, time_str.split(':'))
+    return hours * 3600 + minutes * 60 + seconds
+
 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': [{
@@ -13,32 +19,37 @@ def download_video(video_url, start_time, end_time):
         }],
         'outtmpl': '%(title)s.%(ext)s',
         'postprocessor_args': [
-            '-ss', start_time,  # 使用时间字符串直接传递
-            '-to', end_time
+            '-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
 
-iface = gr.Interface(
-    fn=download_video,
-    inputs=[
-        gr.Textbox(label="YouTube Video URL", placeholder="Enter YouTube video URL here..."),
-        gr.Textbox(label="Start Time (HH:MM:SS)", placeholder="00:00:00"),
-        gr.Textbox(label="End Time (HH:MM:SS)", placeholder="00:05:00")
-    ],
-    outputs=[
-        gr.Text(label="Status Message"),
-        gr.File(label="Downloaded MP3")
-    ],
-    title="YouTube Video Downloader",
-    description="Enter YouTube video URL and specify start and end times to download audio as MP3."
-)
+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
 
-iface.launch()
+if __name__ == "__main__":
+    iface = setup_interface()
+    iface.launch()