Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,7 +4,7 @@ import requests
 import os
 
 # 0. 頁面配置與 CSS 注入（隱藏側邊欄捲軸）
-st.set_page_config(page_title="CIE CIE TAIPEI 線上客服", page_icon="🍸", layout="wide")
+st.set_page_config(page_title="S.H.I.E.L.D. 戰情小助手", page_icon="🛡️", layout="wide")
 
 st.markdown(
     """
@@ -18,12 +18,19 @@ st.markdown(
         scrollbar-width: none;
     }
     
-    /* 2. 調整範例按鈕樣式 */
+    /* 2. 調整範例按鈕樣式 (改為科技藍風格) */
     .stButton button {
         width: auto;
         padding: 5px 15px;
         border-radius: 20px;
-        border: 1px solid #D4AF37; /* 加入一點微奢華的金色邊框 */
+        border: 1px solid #58a6ff; 
+        color: #58a6ff;
+        background-color: transparent;
+        transition: all 0.3s ease;
+    }
+    .stButton button:hover {
+        background-color: #58a6ff;
+        color: #0d1117;
     }
     
     /* 3. 極度壓縮頂部空白與兩側邊距，專為小視窗優化 */
@@ -52,39 +59,23 @@ if not api_key:
 genai.configure(api_key=api_key)
 
 # 1. 基礎設定與連結
-# ⚠️ 注意：這裡預設你把 md 檔放在 github 的 ai-knowledge 資料夾下。如果是放在根目錄，請把 ai-knowledge/ 刪除。
-MD_BASE_URL = "https://raw.githubusercontent.com/ciecietaipei/ciecietaipei.github.io/main/ai-knowledge/"
-# 從你們的 assets 資料夾抓取 Logo
-LOGO_URL = "https://raw.githubusercontent.com/ciecietaipei/ciecietaipei.github.io/main/assets/ciecie_logo_circle.png"
-HOME_URL = "https://ciecietaipei.github.io"
+# 抓取 Deep Learning 101 的 Logo (你也可以換成 SHIELD 專屬 Logo)
+LOGO_URL = "https://raw.githubusercontent.com/Deep-Learning-101/deep-learning-101.github.io/main/DeepLearning101-LOGO.png"
+HOME_URL = "https://deep-learning-101.github.io/SHIELD/"
 
+# 將知識庫指向 S.H.I.E.L.D. 的 README.md
 KNOWLEDGE_MAP = {
-    "🍸 原創調酒 (Cocktails)": {
-        "raw_url": f"{MD_BASE_URL}cocktails.md",
-        "page_url": f"{HOME_URL}/cocktails.html",
-        "repo_url": "https://github.com/ciecietaipei/ciecietaipei.github.io/blob/main/ai-knowledge/cocktails.md"
-    },
-    "🍷 經典酒水 (Drinks)": {
-        "raw_url": f"{MD_BASE_URL}drinks.md",
-        "page_url": f"{HOME_URL}/drinks.html",
-        "repo_url": "https://github.com/ciecietaipei/ciecietaipei.github.io/blob/main/ai-knowledge/drinks.md"
-    },
-    "🍽️ 精緻餐點 (Foods)": {
-        "raw_url": f"{MD_BASE_URL}foods.md",
-        "page_url": f"{HOME_URL}/foods.html",
-        "repo_url": "https://github.com/ciecietaipei/ciecietaipei.github.io/blob/main/ai-knowledge/foods.md"
-    },
-    "🛋️ 空間環境 (Environment)": {
-        "raw_url": f"{MD_BASE_URL}environment.md",
-        "page_url": f"{HOME_URL}/environment.html",
-        "repo_url": "https://github.com/ciecietaipei/ciecietaipei.github.io/blob/main/ai-knowledge/environment.md"
+    "🛡️ S.H.I.E.L.D. 系統白皮書": {
+        "raw_url": "https://raw.githubusercontent.com/Deep-Learning-101/SHIELD/main/README.md",
+        "page_url": "https://deep-learning-101.github.io/SHIELD/",
+        "repo_url": "https://github.com/Deep-Learning-101/SHIELD"
     }
 }
 
 # 2. 批量抓取內容
 def fetch_all_knowledge():
     combined_knowledge = ""
-    with st.spinner("正在同步 CIE CIE 最新菜單與資訊..."):
+    with st.spinner("正在同步 S.H.I.E.L.D. 戰情資料庫..."):
         for category, info in KNOWLEDGE_MAP.items():
             try:
                 response = requests.get(info["raw_url"])
@@ -107,49 +98,51 @@ if "example_prompt" not in st.session_state:
 
 # 3. 側邊欄設計
 with st.sidebar:
-    # 放入 CIE CIE 的圓形 Logo
-    st.markdown(f'<a href="{HOME_URL}" target="_blank"><img src="{LOGO_URL}" width="100%" style="border-radius:50%; margin-bottom:20px;"></a>', unsafe_allow_html=True)
+    st.markdown(f'<a href="{HOME_URL}" target="_blank"><img src="{LOGO_URL}" width="100%" style="margin-bottom:20px;"></a>', unsafe_allow_html=True)
     st.title("⚙️ 知識庫狀態")
     for category, info in KNOWLEDGE_MAP.items():
         with st.expander(category):
             st.markdown(f"🔗 [瀏覽網頁]({info['page_url']})")
             st.markdown(f"📂 [GitHub 原始碼]({info['repo_url']})")
     st.markdown("---")
-    if st.button("🔄 手動更新菜單資訊"):
+    if st.button("🔄 手動更新情資庫"):
         st.session_state.knowledge = fetch_all_knowledge()
         st.success("資料已重新抓取！")
 
 # 4. 主介面與範例問句
-st.title("🍸 CIE CIE 線上微醺客服")
-st.caption("關於酒單、餐點、環境或訂位規範，歡迎直接詢問小助手！")
+st.title("🤖 S.H.I.E.L.D. 戰情小助手")
+st.caption("我是 Deep Learning 101 的主權 AI 戰情官，專注於解答 S.H.I.E.L.D. 的架構與防禦機制。")
 
 # 顯示專屬範例問句按鈕
-example_cols = st.columns(3)
+example_cols = st.columns(6)
 examples = [
-    "🍸 推薦一杯適合約會的花果調酒？",
-    "🍽️ 肚子有點餓，有推薦的熱食嗎？",
-    "�� 請問週末營業到幾點？低消是多少？"
+    "🧠 傳統 AI 常有幻覺，雙腦架構如何落實 AI 治理？",
+    "🏭 企業機密文檔，如何無損轉化為 AI 微調燃料？",
+    "⚔️ 揭秘「紅藍隊自主對抗」與動態語意防火牆",
+    "🕸️ 知識圖譜發威：如何秒級推演受災爆炸半徑？",
+    "🚀 捨棄傳統向量庫？解析 S.H.I.E.L.D. 雙引擎檢索",
+    "🤖 從給建議到「自動修補」：Agent 如何執行 ChatOps？"
 ]
 
 for col, ex in zip(example_cols, examples):
     if col.button(ex):
         st.session_state.example_prompt = ex
 
-# 5. 模型回覆邏輯（設定專屬酒吧人設）
+# 5. 模型回覆邏輯（設定資安 AI 架構師人設）
 def get_gemini_response(user_input):
     system_instruction = f"""
-    你現在是位於台北市大安區信義安和的微醺酒吧『CIE CIE TAIPEI』的專業線上客服小助手。
-    你的說話風格熱情、有質感且帶有一點輕鬆微醺的氛圍。
+    你現在是 Deep Learning 101 開發的『S.H.I.E.L.D. 戰情小助手』。
+    你的說話風格專業、精確，帶有資安專家與 AI 架構師的科技感，致力於推廣企業主權 AI 防禦理念。
     
-    以下是從官方網站最新同步的資訊（包含原創調酒、經典酒水、精緻餐點、環境與規範）：
+    以下是從 GitHub 同步的 S.H.I.E.L.D. 系統白皮書與技術架構資訊：
     ---
     {st.session_state.knowledge}
     ---
-    請嚴格基於上述提供的資訊來回答客人的問題。
-    如果客人問了菜單上沒有的品項，或超出了上述資訊範圍，請禮貌地告知：「目前小助手手邊沒有這項資訊，建議您直接私訊我們的 IG 或是現場詢問調酒師喔！」
+    請嚴格基於上述提供的資訊來回答使用者的問題。
+    如果使用者問了超出 S.H.I.E.L.D. 白皮書範圍的問題，請禮貌地告知：「目前的戰情資料庫尚未收錄此資訊，建議您查閱 Deep Learning 101 的其他專案或直接聯繫維護團隊。」
     """
     try:
-        # 建議使用 1.5-flash，反應速度最快且對文本理解極佳
+        # 建議使用 1.5-flash，對於技術長文的理解力較強
         model = genai.GenerativeModel(
             model_name="gemini-flash-lite-latest",
             system_instruction=system_instruction
@@ -160,12 +153,12 @@ def get_gemini_response(user_input):
     except Exception as e:
         error_msg = str(e)
         if "429" in error_msg or "quota" in error_msg.lower():
-            return "⚠️ **系統提示：**\n\n目前詢問人數較多，小助手稍微忙不過來了！請您稍等幾分鐘後再試喔！"
+            return "⚠️ **系統提示：**\n\n目前 API 請求已達上限，戰情中心通訊稍有延遲！請稍等幾分鐘後再試。"
         else:
             return f"❌ **發生預期外錯誤**\n\n訊息：{error_msg}"
 
 # 6. 對話邏輯
-prompt = st.chat_input("想喝點什麼？或是想了解什麼資訊呢？")
+prompt = st.chat_input("請輸入您對 S.H.I.E.L.D. 系統的疑問...")
 
 if st.session_state.example_prompt:
     prompt = st.session_state.example_prompt