Update src/streamlit_app.py

src/streamlit_app.py

@@ -13,32 +13,23 @@ st.set_page_config(
     layout="wide"
 )
 
-# 確保 .streamlit 目錄存在並建立 secrets.toml
-def ensure_secrets_file():
-    # 建立 .streamlit 目錄
-    os.makedirs('.streamlit', exist_ok=True)
-    
-    # 建立 secrets.toml 檔案，如果還不存在
-    secrets_path = os.path.join('.streamlit', 'secrets.toml')
-    if not os.path.exists(secrets_path):
-        with open(secrets_path, 'w') as f:
-            f.write('GOOGLE_API_KEY = "AIzaSyBxrwdxgs6JemK25piF_RFnxJ9SqKuqhEE"\n')
-        st.success("已建立 .streamlit/secrets.toml 檔案")
-
-# 在應用啟動時確保 secrets 檔案存在
-ensure_secrets_file()
+# 使用環境變數或直接設定 API 金鑰
+# 在生產環境中，應該使用更安全的方法來存儲 API 金鑰
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "AIzaSyBxrwdxgs6JemK25piF_RFnxJ9SqKuqhEE")
 
 # 設定 Gemini API 金鑰
 @st.cache_resource
 def setup_genai():
     try:
         import google.generativeai as genai
-        api_key = st.secrets.get("GOOGLE_API_KEY", "AIzaSyBxrwdxgs6JemK25piF_RFnxJ9SqKuqhEE")
-        genai.configure(api_key=api_key)
+        genai.configure(api_key=GOOGLE_API_KEY)
         return genai.GenerativeModel("gemini-1.5-flash")
     except ImportError:
         st.error("無法載入 google.generativeai 模組。請確保您已安裝此套件: pip install google-generativeai")
         return None
+    except Exception as e:
+        st.error(f"設定 Gemini API 時發生錯誤: {str(e)}")
+        return None
 
 model = setup_genai()