Update app.py

app.py

@@ -468,12 +468,12 @@ def reflect_post(api_key, api_key2, user_input):
     sub_query_texts = [q["sub_query"] for q in direct_queries]
 
     # Step 2：批次匹配表名 (API Call #2)
-    table_map = batch_find_relevant_tables(api_key, sub_query_texts)
+    table_map = batch_find_relevant_tables(api_key2, sub_query_texts)
     if not table_map:
         return all_querys_summary, "⚠️ 系統無法為您的查詢匹配到合適的資料表。"
         
     # Step 3: 批次解析所有子問題的參數 (API Call #3)
-    params_map = batch_parse_sub_queries_with_gemini(api_key2, sub_query_texts)
+    params_map = batch_parse_sub_queries_with_gemini(api_key3, sub_query_texts)
     if not params_map:
         return all_querys_summary, "⚠️ 系統無法解析您問題中的查詢參數。"
 
@@ -509,7 +509,7 @@ def reflect_post(api_key, api_key2, user_input):
 
     # Step 5：整合分析 (API Call #4)
     integration_prompt = f"使用者問題：{user_input}\n\n查詢資料如下：\n{combined_context}"
-    integration_result = reply(api_key2, system_integration, integration_prompt)
+    integration_result = reply(api_key4, system_integration, integration_prompt)
     return all_querys_summary, integration_result
 
 # =======================================================================
@@ -520,6 +520,9 @@ def gradio_interface(user_input):
     """Gradio 主要處理函式"""
     api_key = os.getenv('Gemini')
     api_key2 = os.getenv('Gemini2')
+    api_key3 = os.getenv('Gemini3')
+    api_key4 = os.getenv('Gemini4')
+    
     if not api_key:
         return "❌ 查詢失敗", "錯誤：未在伺服器環境中設定 'Gemini' API 金鑰。"