Update app.py

app.py

@@ -3,13 +3,12 @@ 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 convert_time_to_seconds(hours, minutes, seconds):
+    return int(hours) * 3600 + int(minutes) * 60 + int(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)
+def download_video(video_url, start_hour, start_minute, start_second, end_hour, end_minute, end_second):
+    start_seconds = convert_time_to_seconds(start_hour, start_minute, start_second)
+    end_seconds = convert_time_to_seconds(end_hour, end_minute, end_second)
     options = {
         'format': 'bestaudio/best',
         'postprocessors': [{
@@ -27,27 +26,34 @@ def download_video(video_url, start_time, end_time):
     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():
+    # 定义输入组件
+    start_time_hours = gr.inputs.Dropdown(list(range(24)), label="Start Hour")
+    start_time_minutes = gr.inputs.Dropdown(list(range(60)), label="Start Minute")
+    start_time_seconds = gr.inputs.Number(default=0, label="Start Seconds", maximum=59)
+    end_time_hours = gr.inputs.Dropdown(list(range(24)), label="End Hour")
+    end_time_minutes = gr.inputs.Dropdown(list(range(60)), label="End Minute")
+    end_time_seconds = gr.inputs.Number(default=0, label="End Seconds", maximum=59)
+    
     interface = gr.Interface(
         fn=download_video,
         inputs=[
-            gr.Textbox(lines=2, placeholder="Enter YouTube video URL here...", label="YouTube Video URL"),
-            start_time_hours = gr.inputs.Dropdown(range(24), label="Start Hour")
-            start_time_minutes = gr.inputs.Dropdown(range(60), label="Start Minute")
-            start_time_seconds = gr.inputs.Number(default=0, label="Start Seconds", maximum=59)
-            end_time_hours = gr.inputs.Dropdown(range(24), label="End Hour")
-            end_time_minutes = gr.inputs.Dropdown(range(60), label="End Minute")
-            end_time_seconds = gr.inputs.Number(default=0, label="End Seconds", maximum=59)
+            gr.inputs.Textbox(lines=2, placeholder="Enter YouTube video URL here...", label="YouTube Video URL"),
+            start_time_hours,
+            start_time_minutes,
+            start_time_seconds,
+            end_time_hours,
+            end_time_minutes,
+            end_time_seconds
         ],
         outputs=[
-            gr.Text(label="Status Message"),
-            gr.File(label="Download MP3")
+            gr.outputs.Text(label="Status Message"),
+            gr.outputs.File(label="Downloaded MP3")
         ],
         title="YouTube Video Downloader",
         description="Enter YouTube video URL and specify start and end times to download audio as MP3."