Update ai_service.py

ai_service.py

@@ -1,4 +1,4 @@
-# ai_service.py (Final version, added None placeholder for Gradio button input)
+# ai_service.py (Modified system_instruction to prefer detailed tool output)
 import json
 from datetime import datetime
 import google.generativeai as genai
@@ -45,10 +45,7 @@ def call_mcp_pws_search() -> str:
     try:
         print(f"--- 正在呼叫遠端 PWS MCP 伺服器 (由 Gemini 觸發) ---")
         client = Client("cwadayi/MCP-pws")
-        
-        # [核心修正] 傳入一個 None 作為伺服器端按鈕事件所需的佔位符輸入
         result = client.predict(None, api_name="/get_disaster_warnings")
-        
         return result[0] if isinstance(result, tuple) and len(result) > 0 else str(result)
     except Exception as e:
         return f"工具執行失敗，錯誤訊息: {e}"
@@ -56,25 +53,17 @@ def call_mcp_pws_search() -> str:
 # --- 3. 向 Gemini 定義工具 (Tool Declarations) ---
 
 earthquake_search_tool_declaration = {
-    "name": "call_earthquake_search_tool",
-    "description": "從台灣中央氣象署的資料庫中搜尋地震事件。",
-    "parameters": {
-        "type": "OBJECT",
-        "properties": {
+    "name": "call_earthquake_search_tool", "description": "從台灣中央氣象署的資料庫中搜尋地震事件。",
+    "parameters": { "type": "OBJECT", "properties": {
             "start_date": {"type": "STRING", "description": "搜尋的開始日期 (格式 'YYYY-MM-DD')。"},
             "end_date": {"type": "STRING", "description": "搜尋的結束日期 (格式 'YYYY-MM-DD')。"},
-        },
-        "required": ["start_date", "end_date"]
+        }, "required": ["start_date", "end_date"]
     }
 }
 
 pws_search_tool_declaration = {
-    "name": "call_mcp_pws_search",
-    "description": "查詢最新的 PWS (Public Weather Service) 公共天氣服務發布情形。",
-    "parameters": {
-        "type": "OBJECT",
-        "properties": {}
-    }
+    "name": "call_mcp_pws_search", "description": "查詢最新的 PWS (Public Weather Service) 公共天氣服務發布情形。",
+    "parameters": {"type": "OBJECT", "properties": {}}
 }
 
 available_tools = {
@@ -86,7 +75,15 @@ available_tools = {
 model = None
 if GEMINI_API_KEY and "YOUR_GEMINI_API_KEY" not in GEMINI_API_KEY:
     try:
-        system_instruction = "You are a helpful AI assistant. You must answer in Traditional Chinese. You have access to tools. When a tool returns data, you must analyze the data to fully answer the user's question."
+        # [*** 核心修正 ***]
+        # 修改系統指令，要求 AI 優先顯示工具的原始詳細回傳內容
+        system_instruction = (
+            "You are a helpful AI assistant. You must answer in Traditional Chinese."
+            "You have access to tools. When a tool returns data, "
+            "you MUST prioritize showing the detailed, raw data returned from the tool. "
+            "Do not summarize unless the user explicitly asks for a summary. "
+            "Format the tool's output clearly for the user."
+        )
         model = genai.GenerativeModel(
             model_name="gemini-1.5-flash",
             tools=[earthquake_search_tool_declaration, pws_search_tool_declaration],
@@ -97,12 +94,10 @@ if GEMINI_API_KEY and "YOUR_GEMINI_API_KEY" not in GEMINI_API_KEY:
 
 # --- 5. 主要的 AI 文字生成函式 ---
 def generate_ai_text(user_prompt: str) -> str:
-    if not model:
-        return "🤖 AI (Gemini) 服務尚未設定 API 金鑰，或金鑰無效。"
+    if not model: return "🤖 AI (Gemini) 服務尚未設定 API 金鑰，或金鑰無效。"
     try:
         chat = model.start_chat()
         response = chat.send_message(user_prompt)
-        
         for part in response.parts:
             if part.function_call:
                 function_call = part.function_call
@@ -118,7 +113,7 @@ def generate_ai_text(user_prompt: str) -> str:
                 return final_response.text
         
         return response.text
-
     except Exception as e:
         print(f"與 Gemini AI 互動時發生錯誤: {e}")
         return f"🤖 AI 服務發生錯誤: {e}"
+