Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,95 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import google.generativeai as genai
+import requests
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+st.set_page_config(page_title="AI 新知小助手", page_icon="📚", layout="wide")
+
+if "GEMINI_API_KEY" not in st.secrets:
+    st.error("請在 Secrets 中設定 GEMINI_API_KEY")
+    st.stop()
+
+genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
+
+# 1. 定義你要鎖定的 GitHub 檔案清單
+# 請把 YOUR_ACCOUNT/YOUR_REPO 換成你真實的 GitHub 資訊
+BASE_URL = "https://raw.githubusercontent.com/Deep-Learning-101/deep-learning-101.github.io/main/"
+TARGET_FILES = {
+    "大型語言模型 (LLM)": f"{BASE_URL}Large-Language-Model.md",
+    "自然語言處理 (NLP)": f"{BASE_URL}Natural-Language-Processing.md",
+    "語音處理 (Speech)": f"{BASE_URL}Speech-Processing.md",
+    "電腦視覺 (CV)": f"{BASE_URL}Computer-Vision.md"
+}
+
+# 2. 批量抓取並格式化內容
+def fetch_all_knowledge():
+    combined_knowledge = ""
+    with st.spinner("正在從 GitHub 同步 4 大領域最新資訊..."):
+        for category, url in TARGET_FILES.items():
+            try:
+                response = requests.get(url)
+                response.raise_for_status()
+                # 在每個檔案內容前加上明確的標題，幫助 AI 分類記憶
+                combined_knowledge += f"\n\n## 【領域：{category}】\n"
+                combined_knowledge += response.text
+            except Exception as e:
+                st.warning(f"無法同步 {category} 的資料：{e}")
+    return combined_knowledge
+
+# 初始化 Session State
+if "knowledge" not in st.session_state:
+    st.session_state.knowledge = fetch_all_knowledge()
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# 側邊欄設計
+with st.sidebar:
+    st.title("⚙️ 知識庫狀態")
+    st.write("目前收錄以下領域：")
+    for category in TARGET_FILES.keys():
+        st.markdown(f"- {category}")
+        
+    st.markdown("---")
+    if st.button("🔄 手動更新知識庫"):
+        st.session_state.knowledge = fetch_all_knowledge()
+        st.success("所有領域資料已重新抓取！")
+
+# 主介面
+st.title("📚 AI 演算法與論文社群助手")
+st.caption("知識庫涵蓋 LLM、NLP、Speech、CV。歡迎直接提問！")
+
+def get_gemini_response(user_input):
+    system_instruction = f"""
+    你是一位專業的 AI 技術分析專家。
+    以下是我從 GitHub 同步的四個領域 (LLM, NLP, Speech, CV) 的最新技術簡介與重點：
+    ---
+    {st.session_state.knowledge}
+    ---
+    請嚴格基於上述提供的資訊來回答使用者的問題。
+    如果使用者詢問了上述資訊中沒有涵蓋的細節，請明確告知：「目前這份懶人包中尚未收錄關於此問題的詳細資訊」。
+    """
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-pro",
+        system_instruction=system_instruction
+    )
+    
+    chat = model.start_chat(history=[])
+    response = chat.send_message(user_input)
+    return response.text
+
+# 顯示對話紀錄
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# 接收輸入
+if prompt := st.chat_input("例如：請幫我比較一下 CV 跟 NLP 目前常用的技術？"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        response_text = get_gemini_response(prompt)
+        st.markdown(response_text)
+    
+    st.session_state.messages.append({"role": "assistant", "content": response_text})