# 安裝依賴庫
pip install streamlit streamlit-webrtc audio-recorder-streamlit openai-whisper python-dotx

import streamlit as st
from audio_recorder_streamlit import audio_recorder
import whisper
import os
from datetime import datetime

# 初始化模型
@st.cache_resource
def load_whisper_model():
    return whisper.load_model("base")  # 使用基礎(chǔ)版模型

model = load_whisper_model()

# 界面布局
st.title("?? 音頻處理工作流")
col1, col2 = st.columns(2)

with col1:
    # 音頻錄制組件
    audio_bytes = audio_recorder(
        text="點擊錄音",
        recording_color="#e87070",
        neutral_color="#6aa36f",
        icon_name="microphone",
        sample_rate=16000
    )
    
    # 保存錄音文件
    if audio_bytes:
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        audio_path = f"audio_{timestamp}.wav"
        with open(audio_path, "wb") as f:
            f.write(audio_bytes)
        st.session_state.audio_path = audio_path
        st.audio(audio_bytes, format="audio/wav")

with col2:
    # 語音轉(zhuǎn)文本功能
    if 'audio_path' in st.session_state and st.button("開始轉(zhuǎn)換"):
        progress_bar = st.progress(0)
        status_text = st.empty()
        
        try:
            status_text.text("加載音頻文件...")
            progress_bar.progress(20)
            
            # 使用Whisper進行轉(zhuǎn)換
            status_text.text("語音識別中...")
            result = model.transcribe(st.session_state.audio_path)
            progress_bar.progress(80)
            
            # 顯示結(jié)果
            st.subheader("轉(zhuǎn)換結(jié)果")
            st.code(result["text"], language="text")
            st.session_state.text_result = result["text"]
            
            # 生成下載按鈕
            with st.expander("下載選項"):
                st.download_button(
                    label="下載文本",
                    data=st.session_state.text_result,
                    file_name=f"transcript_{timestamp}.txt",
                    mime="text/plain"
                )
                
                with open(st.session_state.audio_path, "rb") as f:
                    st.download_button(
                        label="下載音頻",
                        data=f,
                        file_name=audio_path,
                        mime="audio/wav"
                    )
            
            progress_bar.progress(100)
            status_text.text("處理完成！")
            
        except Exception as e:
            st.error(f"處理失敗: {str(e)}")
            progress_bar.progress(0)

# 在HuggingFace Spaces部署時添加配置
STREAMLIT_SERVER_PORT = 8501

model = whisper.load_model("base", device="cuda")

import matplotlib.pyplot as plt
from scipy.io import wavfile

rate, data = wavfile.read(audio_path)
plt.specgram(data, Fs=rate)
st.pyplot(plt)

# requirements.txt
streamlit>=1.28
openai-whisper==20231106
audio-recorder-streamlit==0.1.7

# 監(jiān)控內(nèi)存使用
ps -o pid,user,%mem,command ax | grep streamlit