Update src/streamlit_app.py

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 requests
 import os
 
@@ -66,7 +67,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. 基礎設定與連結
 # 抓取 Deep Learning 101 的 Logo (你也可以換成 SHIELD 專屬 Logo)
@@ -138,7 +140,7 @@ for col, ex in zip(example_cols, examples):
     if col.button(ex):
         st.session_state.example_prompt = ex
 
-# 5. 模型回覆邏輯（設定資安 AI 架構師人設）
+# 5. 【更新】模型回覆邏輯（新版 SDK + 記憶功能 + 防止草稿輸出）
 def get_gemini_response(user_input):
     system_instruction = f"""
     你現在是 Deep Learning 101 開發的『S.H.I.E.L.D. 戰情小助手』。
@@ -148,18 +150,40 @@ def get_gemini_response(user_input):
     ---
     {st.session_state.knowledge}
     ---
+    
+    【核心任務】：
     請嚴格基於上述提供的資訊來回答使用者的問題。
     如果使用者問了超出 S.H.I.E.L.D. 白皮書範圍的問題，請禮貌地告知：「目前的戰情資料庫尚未收錄此資訊，建議您查閱 Deep Learning 101 的其他專案或直接聯繫維護團隊。」
+    
+    【嚴格輸出限制】：
+    1. 請直接給出專業、精確的繁體中文回覆。
+    2. 絕對禁止輸出任何你的內部思考過程、計畫草稿、英文標籤、或是角色設定分析。
+    3. 所有的輸出內容必須 100% 都是直接面對使用者溝通的對話內容。
     """
+    
     try:
-        # 建議使用 1.5-flash，對於技術長文的理解力較強
-        model = genai.GenerativeModel(
-            model_name="gemini-flash-lite-latest",
-            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="gemma-4-31b-it", 
+            config=types.GenerateContentConfig(
+                system_instruction=system_instruction,
+                temperature=0.3, # 針對技術戰情，調低溫度讓回答精準客觀，減少幻覺
+            ),
+            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():