Create upload reference

upload reference

@@ -0,0 +1,87 @@
+import os
+import datetime
+import shutil
+import gradio as gr
+import tempfile # 假設這是用於處理臨時檔案的庫
+
+# --- 自動儲存設定 ---
+# 儲存上傳的參考語音檔案的資料夾
+SAVE_UPLOADED_REFERENCES_DIR = "uploaded_references" 
+
+# 確保儲存資料夾存在
+os.makedirs(SAVE_UPLOADED_REFERENCES_DIR, exist_ok=True)
+# --- 結束自動儲存設定 ---
+
+# --- generate_speech 函數中的相關部分 ---
+# 假設這是 generate_speech 函數的定義
+# def generate_speech(text, language, uploaded_speaker_audio_path):
+#     # ... 其他邏輯 ...
+
+    # --- 決定使用哪個語音參考檔案 ---
+    speaker_wav_to_use = uploaded_speaker_audio_path
+    
+    # 如果用戶上傳了參考語音，則儲存它
+    if uploaded_speaker_audio_path:
+        try:
+            timestamp_ref = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+            # 獲取原始檔案的副檔名，例如 .wav
+            original_ext = os.path.splitext(uploaded_speaker_audio_path)[1]
+            # 構建儲存檔案的名稱，包含時間戳和原始副檔名
+            saved_ref_file_name = f"{timestamp_ref}_uploaded_ref{original_ext}"
+            # 構建儲存檔案的完整路徑
+            saved_ref_file_path = os.path.join(SAVE_UPLOADED_REFERENCES_DIR, saved_ref_file_name)
+            # 將上傳的臨時檔案複製到指定的儲存目錄
+            shutil.copy(uploaded_speaker_audio_path, saved_ref_file_path)
+            print(f"上傳的參考語音已儲存到：{saved_ref_file_path}")
+        except Exception as e:
+            print(f"儲存上傳的參考語音時發生錯誤: {e}")
+            # 即使儲存參考語音失敗，也繼續生成語音
+        
+        print(f"使用上傳的語音參考檔案: {speaker_wav_to_use}")
+    else: # 如果用戶沒有上傳檔案，則使用預設檔案
+        # ... 處理預設檔案的邏輯 ...
+        pass # 這裡省略了預設檔案的處理，因為重點是上傳參考語音的儲存
+    # --- 結束決定 ---
+
+    # ... 後續的語音生成和生成的語音儲存邏輯 ...
+
+# --- 結束 generate_speech 函數中的相關部分 ---
+
+
+# --- 用於查看已上傳參考語音的函數 ---
+def list_uploaded_reference_files():
+    """掃描上傳參考語音資料夾，返回所有 .wav 檔案的完整路徑列表。"""
+    ref_files = []
+    if os.path.exists(SAVE_UPLOADED_REFERENCES_DIR) and os.path.isdir(SAVE_UPLOADED_REFERENCES_DIR):
+        for filename in os.listdir(SAVE_UPLOADED_REFERENCES_DIR):
+            if filename.lower().endswith(".wav"):
+                # 返回完整路徑，以便 Gradio 可以正確顯示和提供下載
+                ref_files.append(os.path.join(SAVE_UPLOADED_REFERENCES_DIR, filename))
+    # 按修改時間倒序排列，最新檔案在前
+    ref_files.sort(key=os.path.getmtime, reverse=True) 
+    return ref_files
+# --- 結束查看已上傳參考語音的函數 ---
+
+
+# --- Gradio 介面中的相關部分 ---
+# 假設這是 gr.Blocks(title="...") as demo: 內部
+# with gr.Blocks(title="Coqui TTS XTTS-v2 語音生成") as demo:
+#     # ... 其他 Tab ...
+
+    with gr.Tab("查看已上傳參考語音"):
+        gr.Markdown("### 已儲存的上傳參考語音檔案")
+        gr.Markdown("這些是您上傳的語音參考檔案。")
+        
+        saved_uploaded_ref_files_output = gr.File(
+            label="上傳的參考語音檔案",
+            file_count="multiple", # 允許顯示多個檔案
+            interactive=False # 不允許用戶上傳，只用於顯示和下載
+        )
+        refresh_uploaded_ref_button = gr.Button("刷新參考語音列表")
+
+        # 應用程式載入時，自動載入檔案列表
+        # demo.load(list_uploaded_reference_files, outputs=[saved_uploaded_ref_files_output])
+        # 點擊刷新按鈕時，重新載入檔案列表
+        refresh_uploaded_ref_button.click(list_uploaded_reference_files, outputs=[saved_uploaded_ref_files_output])
+
+# --- 結束 Gradio 介面中的相關部分 ---