配置自动保存功能

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-from utils import VideoUtils
+from utils import VideoUtils, SettingsManager
 from stt_module import STTManager
 from translator import Translator
 import tempfile
@@ -9,6 +9,9 @@ from dotenv import load_dotenv
 # 加载 .env 文件中的配置
 load_dotenv()
 
+# 加载本地保存的设置
+saved_settings = SettingsManager.load_settings()
+
 # 取消 Streamlit 上传限制（虽然主要通过命令行配置，但在脚本中提醒用户）
 # 实际上 Streamlit 的服务器配置需要在运行命令时指定，或者写在 config.toml 中
 # 我们这里先通过 UI 提醒用户
@@ -50,13 +53,13 @@ st.title("🎬 AI 媒体字幕提取与翻译")
 with st.sidebar:
     st.header("⚙️ 配置参数")
     
-    # 从环境变量读取默认值，如果没有则为空字符串或预设值
-    default_api_key = os.getenv("API_KEY", "")
-    default_base_url = os.getenv("BASE_URL", "https://api.siliconflow.cn/v1")
-    default_model = os.getenv("MODEL_NAME", "THUDM/glm-4-9b-chat")
-    default_stt_size = os.getenv("STT_MODEL_SIZE", "base")
-    default_lang = os.getenv("TARGET_LANG", "中文")
-    default_device = os.getenv("DEVICE", "cpu")
+    # 从本地保存的设置或环境变量读取默认值
+    default_api_key = saved_settings.get("API_KEY", os.getenv("API_KEY", ""))
+    default_base_url = saved_settings.get("BASE_URL", os.getenv("BASE_URL", "https://api.siliconflow.cn/v1"))
+    default_model = saved_settings.get("MODEL_NAME", os.getenv("MODEL_NAME", "THUDM/glm-4-9b-chat"))
+    default_stt_size = saved_settings.get("STT_MODEL_SIZE", os.getenv("STT_MODEL_SIZE", "base"))
+    default_lang = saved_settings.get("TARGET_LANG", os.getenv("TARGET_LANG", "中文"))
+    default_device = saved_settings.get("DEVICE", os.getenv("DEVICE", "cpu"))
 
     api_key = st.text_input("API Key", value=default_api_key, type="password", help="输入 API Key", disabled=st.session_state.processing)
     base_url = st.text_input("API Base URL", value=default_base_url, disabled=st.session_state.processing)
@@ -111,6 +114,17 @@ with st.sidebar:
     lang_index = lang_options.index(default_lang) if default_lang in lang_options else 0
     target_lang = st.selectbox("目标语言", lang_options, index=lang_index, disabled=st.session_state.processing)
 
+    # 保存当前配置到本地文件的辅助函数
+    def save_current_settings():
+        SettingsManager.save_settings({
+            "API_KEY": api_key,
+            "BASE_URL": base_url,
+            "MODEL_NAME": model_name,
+            "STT_MODEL_SIZE": model_size,
+            "TARGET_LANG": target_lang,
+            "DEVICE": device
+        })
+
     if st.session_state.processing:
         st.divider()
         if st.button("⏹️ 强制停止任务", type="secondary", use_container_width=True):
@@ -220,6 +234,9 @@ with tab1:
                     VideoUtils.write_srt(all_translated_segments, trans_srt_path)
 
                     status.update(label="处理完成！", state="complete", expanded=False)
+                    
+                    # 成功处理后自动保存配置
+                    save_current_settings()
 
                     # 保存结果到 session_state
                     st.session_state.process_results = {
@@ -344,6 +361,10 @@ with tab2:
                         "trans_srt": temp_trans_srt.name
                     }
                     st.success("翻译完成！")
+                    
+                    # 成功处理后自动保存配置
+                    save_current_settings()
+                    
                     st.session_state.processing = False
                     st.rerun()
             except Exception as e: