Update src/streamlit_app.py

src/streamlit_app.py

@@ -17,7 +17,7 @@ st.set_page_config(
 )
 
 # 設定 Gemini API 金鑰
-GEMINI_API_KEY = st.sidebar.text_input("Gemini API Key", "AIzaSyC3fbJVYQbINa8ztuOS5dFE4ud0I9jhy3o", type="password")
+GEMINI_API_KEY = st.sidebar.text_input("Gemini API Key", "AIzaSyBxrwdxgs6JemK25piF_RFnxJ9SqKuqhEE", type="password")
 
 if GEMINI_API_KEY:
     # 設定 API 金鑰
@@ -36,22 +36,6 @@ if 'csv_log' not in st.session_state:
     # 添加標題行
     st.session_state.csv_log.append(["Timestamp", "User", "Gemini"])
 
-# 函數: 生成並下載 CSV
-def get_csv_download_link(csv_data, filename="chat_log.csv"):
-    # 創建 CSV 字符串
-    csv_string = io.StringIO()
-    writer = csv.writer(csv_string)
-    for row in csv_data:
-        writer.writerow(row)
-    
-    # 對 CSV 字符串進行 base64 編碼
-    csv_string = csv_string.getvalue()
-    b64 = base64.b64encode(csv_string.encode()).decode()
-    
-    # 創建下載連結
-    href = f'<a href="data:file/csv;base64,{b64}" download="{filename}">下載 CSV 檔案</a>'
-    return href
-
 # 函數: 記錄對話到 CSV
 def log_to_csv(user_message, gemini_response):
     timestamp = datetime.now().isoformat()
@@ -131,9 +115,23 @@ with st.sidebar:
     st.title("台南開放資料 Gemini 分析")
     st.markdown("---")
     
-    # 下載 CSV 按鈕
+    # 下載 CSV 按鈕 - 使用原生 download_button 代替 HTML 連結
     st.markdown("### 下載對話記錄")
-    st.markdown(get_csv_download_link(st.session_state.csv_log), unsafe_allow_html=True)
+    
+    # 準備 CSV 資料
+    csv_string = io.StringIO()
+    writer = csv.writer(csv_string)
+    for row in st.session_state.csv_log:
+        writer.writerow(row)
+    csv_bytes = csv_string.getvalue().encode('utf-8-sig')
+    
+    # 使用原生下載按鈕
+    st.download_button(
+        label="下載對話記錄 CSV",
+        data=csv_bytes,
+        file_name="chat_log.csv",
+        mime="text/csv",
+    )
     
     st.markdown("---")
     st.markdown("### 關於")