Upload yt_dip(個人版).py

yt_dip(個人版).py

@@ -1,57 +1,58 @@
-# -*- coding: utf-8 -*-
-"""YT-DIP(個人版).ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1BR2TvWOPERSBX29RxIuc-4oAiRuFy6cM
-
-## YT-DIP 專案說明：
-本專案採用 "YT-DIP" 庫實現視頻下載與轉換功能，該庫提供了一種高效且可靠的方法來從YouTube下載視頻並轉換成音訊格式。相比傳統的 "youtomp3" 項目的PYtube，本專案的實現方式更為簡潔，欲處理時間得以減少至原項目的1/3。
-
-### 主要功能
-- **音訊轉換**：將下載的視頻檔轉換為高品質的 MP3 格式。
-- **時間剪輯**：支援對下載的視頻進行時間段剪輯，只提取使用者感興趣的部分。
-- **使用者介面**：簡潔明瞭的使用者參數設置介面，支援通過表單直接輸入視頻URL和剪輯時間。
-
-### 開發環境
-- **主要依賴**：YT-DIP, ffmpeg
-- **語言**：Python
-"""
-
-#@title 必要的庫
-# 安裝 yt-dlp 和 ffmpeg
-!pip install -U yt-dlp
-!apt-get install ffmpeg
+import yt_dlp
+import gradio as gr
+import os
+from glob import glob
 
-#@title 設定下載和剪輯參數 { run: "auto", display-mode: "form" }
-#@markdown ### 輸入 影片 URL
-video_url = 'https://www.youtube.com/watch?v=_Em7aKFOJjA' #@param {type:"string"}
-#@markdown ### 剪輯開始時間
-start_time = "00:00:00" #@param {type:"string"}
-#@markdown ### 剪輯結束時間
-end_time = "00:05:30" #@param {type:"string"}
+# 安装库的操作在 Hugging Face Spaces 会在 requirements.txt 中处理，这里不再需要
+# requirements.txt 内容应包括 yt-dlp 和 gradio
 
-# Helper function to convert HH:MM:SS to seconds
 def convert_time_to_seconds(time_str):
     hours, minutes, seconds = map(int, time_str.split(':'))
     return hours * 3600 + minutes * 60 + seconds
 
-# 使用帮助函数转换时间
-start_seconds = convert_time_to_seconds(start_time)
-end_seconds = convert_time_to_seconds(end_time)
-
-#@title 下載並剪輯視頻
-!yt-dlp -x --audio-format mp3 --audio-quality 128K --ppa "ExtractAudio:-ss {start_seconds} -to {end_seconds}" {video_url} -o "%(title)s.%(ext)s"
-from google.colab import files
-import os
-import glob
-# 使用 glob 模块来更可靠地匹配文件名，尤其是包含特殊字符和空格的情况
-download_files = glob.glob('*.mp3')
-if download_files:
-    # 确保正确处理文件名，尤其是包含特殊字符的情况
-    download_filename = download_files[0]  # 下载第一个找到的 MP3 文件
-    print(f"Downloading file: {download_filename}")  # 打印将要下载的文件名
-    files.download(download_filename)
-else:
-    print("No MP3 file found.")
+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()