# 關(guān)鍵庫版本要求
moviepy==1.0.3  # 音視頻處理核心庫

# 導(dǎo)入 moviepy 庫中的 AudioFileClip 類，用于處理視頻文件
from moviepy.audio.io.AudioFileClip import AudioFileClip
# 導(dǎo)入 os 模塊，用于處理文件路徑相關(guān)操作
import os

def extract_audio_from_video(video_path, audio_path=None):
    """
    從視頻中提取音頻。
    
    參數(shù):
    video_path (str): 視頻文件的路徑。
    audio_path (str, 可選): 提取后音頻文件的保存路徑，若未提供，則默認(rèn)與視頻文件同名，擴(kuò)展名為 .mp3。
    
    返回:
    str: 提取后音頻文件的路徑。
    """
    # 檢查視頻文件是否存在
    if not os.path.isfile(video_path):
        raise FileNotFoundError(f'指定的視頻文件不存在: {video_path}')
    
    # 若未提供音頻保存路徑，則默認(rèn)使用視頻文件同名且擴(kuò)展名為 .mp3 的路徑
    if audio_path is None:
        audio_path = os.path.splitext(video_path)[0] + ".mp3"
    
    print(f"提取的音頻路徑: {audio_path}")
    
    try:
        # 創(chuàng)建 AudioFileClip 對(duì)象，用于提取視頻中的音頻
        with AudioFileClip(video_path) as audio_clip:
            # 將提取的音頻保存到指定路徑
            audio_clip.write_audiofile(audio_path, codec='mp3')
        return audio_path
    except Exception as e:
        # 打印提取音頻時(shí)出現(xiàn)的錯(cuò)誤信息
        print(f"提取音頻時(shí)出錯(cuò): {e}")
        # 拋出異常，讓調(diào)用者處理
        raise

import streamlit as st
import os
from datetime import datetime
from video_audio_extractor import extract_audio_from_video
from dotenv import load_dotenv

# 創(chuàng)建輸入和輸出目錄
def create_directories():
    # 獲取當(dāng)前日期，并格式化為 yyyymmdd 的字符串
    today = datetime.now().strftime('%Y%m%d')
    # 定義輸入目錄，將其設(shè)置為 'input' 文件夾下以當(dāng)前日期命名的子文件夾
    input_dir = os.path.join('input', today)
    # 定義輸出目錄，將其設(shè)置為 'output' 文件夾下以當(dāng)前日期命名的子文件夾
    output_dir = os.path.join('output', today)
    # 創(chuàng)建輸入目錄，如果目錄已存在則不會(huì)報(bào)錯(cuò)
    os.makedirs(input_dir, exist_ok=True)
    # 創(chuàng)建輸出目錄，如果目錄已存在則不會(huì)報(bào)錯(cuò)
    os.makedirs(output_dir, exist_ok=True)
    # 返回創(chuàng)建好的輸入目錄和輸出目錄路徑
    return input_dir, output_dir

# 主函數(shù)
def main():
    # 加載環(huán)境變量文件
    # 若加載成功，繼續(xù)執(zhí)行后續(xù)操作；若失敗，顯示錯(cuò)誤信息并終止程序
    if not load_dotenv():
        st.error("讀取配置文件失敗，請(qǐng)檢查配置文件是否存在或路徑是否正確")
        return
    
    # 創(chuàng)建輸入和輸出目錄
    input_dir, output_dir = create_directories()

    # 設(shè)置應(yīng)用標(biāo)題
    st.title('視頻提取音頻工具')

    # 創(chuàng)建文件上傳組件，僅允許上傳指定類型的視頻文件
    uploaded_file = st.file_uploader('請(qǐng)上傳視頻文件', type=['mp4', 'mov', 'avi'])
    # 創(chuàng)建轉(zhuǎn)換按鈕
    convert_button = st.button('轉(zhuǎn)換為音頻文件')

    if uploaded_file is not None and convert_button:
        try:
            # 初始化進(jìn)度條
            progress_bar = st.progress(0)
            # 獲取上傳文件的擴(kuò)展名
            file_extension = os.path.splitext(uploaded_file.name)[1]
            # 生成帶有時(shí)間戳的新文件名
            new_file_name = f'video_audio_extractor_{datetime.now().strftime("%Y%m%d_%H%M%S")}'
            # 拼接視頻文件保存路徑
            video_path = os.path.join(input_dir, f"{new_file_name}{file_extension}")
            # 拼接音頻文件保存路徑
            audio_path = os.path.join(output_dir, f"{new_file_name}.mp3")
            print(audio_path)
            # 保存上傳的視頻文件
            with open(video_path, 'wb') as f:
                f.write(uploaded_file.getbuffer())
            # 更新進(jìn)度條到 25%
            progress_bar.progress(25)

            # 調(diào)用提取音頻的函數(shù)
            audio_path = extract_audio_from_video(video_path, audio_path)
            # 更新進(jìn)度條到 100%
            progress_bar.progress(100)

            # 顯示音頻保存成功的消息
            st.success(f'音頻已保存至 {audio_path}')
            # 提供音頻文件下載鏈接
            with open(audio_path, 'rb') as audio_file:
                st.download_button('下載音頻', audio_file, file_name=os.path.basename(audio_path))
            # 顯示音頻播放組件
            with open(audio_path, 'rb') as audio_file:
                st.audio(audio_file, format='audio/mp3')

        except Exception as e:
            # 捕獲并顯示處理過程中出現(xiàn)的錯(cuò)誤
            st.error(f'處理過程中出錯(cuò): {e}')

if __name__ == '__main__':
    main()