Adaptive API Base Url

app.py

@@ -41,17 +41,31 @@ def save_pdf(file):
     except Exception as e:
         return f"❌ 出错: {str(e)}", get_debug_info()
 
-def save_api_key(api_key):
+def save_api_settings(api_key, api_base_url=None):
     if not api_key: return "❌ Key 不能为空", get_debug_info()
     try:
         config = {}
         if os.path.exists(CONFIG_PATH):
             with open(CONFIG_PATH, "r", encoding="utf-8") as f:
                 config = yaml.safe_load(f) or {}
+                
+        # 保存 API Key
         config.setdefault("api_keys", {})["gemini_api_key"] = api_key
+        
+        # 如果 api_base_url 不为空（不是 None 且不是空字符串），则保存该值
+        if api_base_url:
+            config["api_base_url"] = api_base_url
+            
+        # 写入 YAML 文件
         with open(CONFIG_PATH, "w", encoding="utf-8") as f:
             yaml.dump(config, f, allow_unicode=True)
-        return "✅ Key 已保存", get_debug_info()
+            
+        # 动态生成成功提示语
+        success_msg = "✅ Key 已保存"
+        if api_base_url:
+            success_msg += "，Base URL 已更新"
+            
+        return success_msg, get_debug_info()
     except Exception as e:
         return f"❌ 出错: {str(e)}", get_debug_info()
 
@@ -195,14 +209,16 @@ def run_final_generation(task_type="all"):
 
 # --- UI ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📑 Mineru PDF 解析调试器--testv2")
+    gr.Markdown("# 📑 Mineru PDF 解析调试器")
     
     with gr.Row():
         with gr.Column(scale=1):
             with gr.Group():
                 gr.Markdown("### 1. 配置 & 上传")
-                key_input = gr.Textbox(label="API Key", type="password")
-                key_btn = gr.Button("保存 Key")
+                key_input = gr.Textbox(label="API Key", type="password", placeholder="请输入您的 API Key")
+                # 👇 这里新增了 Base URL 的输入框
+                api_base_url_input = gr.Textbox(label="API Base URL (可选)", placeholder="例如: https://api.example.com/v1")
+                key_btn = gr.Button("保存配置") # 稍微改了下名字更贴切
                 gr.Markdown("---")
                 pdf_input = gr.File(label="选择 PDF", file_types=[".pdf"])
                 pdf_btn = gr.Button("保存 PDF")
@@ -238,7 +254,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             refresh_btn = gr.Button("🔄 刷新状态")
 
     # ================= 逻辑绑定 =================
-    key_btn.click(save_api_key, inputs=key_input, outputs=[parse_status, debug_view])
+    # 👇 这里绑定了两个输入框
+    key_btn.click(save_api_settings, inputs=[key_input, api_base_url_input], outputs=[parse_status, debug_view])
     pdf_btn.click(save_pdf, inputs=pdf_input, outputs=[parse_status, debug_view])
     
     parse_btn.click(
@@ -246,7 +263,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[parse_status, debug_view, cmd_logs]
     )
 
-    # 最终生成逻辑绑定 (使用 lambda 传递固定的 task_type 参数)
+    # 最终生成逻辑绑定
     gen_ppt_btn.click(fn=lambda: run_final_generation("ppt"), outputs=[gen_status, debug_view, cmd_logs, download_file])
     gen_poster_btn.click(fn=lambda: run_final_generation("poster"), outputs=[gen_status, debug_view, cmd_logs, download_file])
     gen_pr_btn.click(fn=lambda: run_final_generation("pr"), outputs=[gen_status, debug_view, cmd_logs, download_file])