Update src/streamlit_app.py

升級為新版 SDK (google.genai)、修復了記憶功能問題，且包含了嚴格指令以防止模型印出內部草稿

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-import google.generativeai as genai
+from google import genai
+from google.genai import types # 引入 types 用來設定對話歷史與系統指令
 import os
 
 # 0. 頁面配置與 CSS 注入（隱藏側邊欄捲軸）
@@ -64,7 +65,8 @@ if not api_key:
     st.error("請確認已經在 Space 的 Settings 設定了 GEMINI_API_KEY")
     st.stop()
 
-genai.configure(api_key=api_key)
+# 【更新】使用新版 SDK 建立 Client
+client = genai.Client(api_key=api_key)
 
 # 1. 基礎設定與連結
 HOME_URL = "https://kidaid.org.tw/"
@@ -155,29 +157,50 @@ for col, ex in zip(example_cols, examples):
     if col.button(ex):
         st.session_state.example_prompt = ex
 
-# 5. 模型回覆邏輯（設定專業衛教師人設）
+# 5. 【更新】模型回覆邏輯（新版 SDK + 記憶功能 + 防止草稿輸出）
 def get_gemini_response(user_input):
     system_instruction = f"""
     你現在是『育兒成 | 全方位兒童發展整合照護平臺』的專業線上客服與育兒小助手。
     你的說話風格溫柔、有耐心、具備同理心，且充滿專業知識，就像一位溫暖的早期療育專家或兒童發展衛教師。
-    
+        
     以下是本平台最新的衛教資訊（包含認知、語言、粗大動作、精細動作、社會情緒、飲食、ASD早療對策等）：
     ---
     {st.session_state.knowledge}
     ---
+    
+    【核心任務】：
     請嚴格基於上述提供的資訊來回答家長的問題。將複雜的衛教知識轉化為容易理解的實作建議。
     如果家長問了超出了上述資訊範圍的醫療診斷問題，請溫和地告知：「小助手目前手邊沒有相關資訊。每個孩子的發展狀況不同，建議您可以諮詢專業的小兒科醫師或尋求早期療育評估喔！」
+    
+    【嚴格輸出限制（非常重要）】：
+    1. 請「直接」給出你要對家長說的回覆，開頭直接打招呼（例如：您好！...）。
+    2. 絕對禁止輸出任何你的內部思考過程、計畫草稿、英文標籤、或是角色設定分析（例如絕對禁止輸出 User:, Question:, Persona:, Constraints:, Core Development Points: 等內容）。
+    3. 所有的輸出內容必須 100% 都是直接面對家長溝通的繁體中文對話。
     """
     
     try:
-        model = genai.GenerativeModel(
-            # model_name="gemini-flash-lite-latest",
-            model_name="gemma-4-31b-it",
-            system_instruction=system_instruction
+        # 將 Streamlit 儲存的歷史紀錄，轉換為新版 SDK 規定的格式
+        history_format = []
+        for msg in st.session_state.messages:
+            role = "user" if msg["role"] == "user" else "model"
+            history_format.append(
+                types.Content(role=role, parts=[types.Part.from_text(text=msg["content"])])
+            )
+
+        # 使用新版 client 建立對話
+        chat = client.chats.create(
+            # 建議使用 gemini-2.5-flash，對話效果更好且不易出錯
+            model="gemini-2.5-flash", 
+            config=types.GenerateContentConfig(
+                system_instruction=system_instruction,
+                temperature=0.7, # 稍微調低溫度，讓回答更穩定、不易產生幻覺
+            ),
+            history=history_format
         )
-        chat = model.start_chat(history=[])
+        
         response = chat.send_message(user_input)
         return response.text
+        
     except Exception as e:
         error_msg = str(e)
         if "429" in error_msg or "quota" in error_msg.lower():