sirochild commited on
Commit
3e44aa4
·
verified ·
1 Parent(s): ac8c211

Upload main_app.py

Browse files
Files changed (1) hide show
  1. main_app.py +8 -6
main_app.py CHANGED
@@ -28,15 +28,10 @@ st.set_page_config(
28
  USER_DATA_DIR = "/mnt/data/mari_users"
29
  os.makedirs(USER_DATA_DIR, exist_ok=True)
30
 
31
- # --- 1. HF トークン取得 ---
32
- # secrets.toml に HF_TOKEN="your_token_here" として保存しておく
33
- hf_token = st.secrets.get("HF_TOKEN")
34
-
35
-
36
  # --- 2. ユーザー ID 判定(毎回再認証・再発行) ---
37
 
38
  def get_current_user_id():
39
- hf_token = st.secrets.get("HF_TOKEN")
40
  # HFトークンが存在する場合は毎回whoamiで認証・ID取得
41
  if hf_token:
42
  try:
@@ -1351,6 +1346,13 @@ def render_chat_tab_content(managers):
1351
 
1352
 
1353
  with st.expander("💾 データ保存"):
 
 
 
 
 
 
 
1354
  # 保存データの存在確認(永続ストレージ優先)
1355
  # persistent_user_manager 完全廃止
1356
  user_id_manager = managers["user_id_manager"] # フォールバック用
 
28
  USER_DATA_DIR = "/mnt/data/mari_users"
29
  os.makedirs(USER_DATA_DIR, exist_ok=True)
30
 
 
 
 
 
 
31
  # --- 2. ユーザー ID 判定(毎回再認証・再発行) ---
32
 
33
  def get_current_user_id():
34
+ hf_token = st.session_state.get("hf_token", None)
35
  # HFトークンが存在する場合は毎回whoamiで認証・ID取得
36
  if hf_token:
37
  try:
 
1346
 
1347
 
1348
  with st.expander("💾 データ保存"):
1349
+ # HFトークン入力欄
1350
+ st.markdown("---")
1351
+ st.markdown("**HFトークン入力欄(これを入れないとデータの復元が出来ません)**")
1352
+ hf_token_input = st.text_input("HuggingFaceアクセストークン", value=st.session_state.get("hf_token", ""), type="password", help="HuggingFaceのアクセストークンを入力してください")
1353
+ if hf_token_input:
1354
+ st.session_state.hf_token = hf_token_input
1355
+ st.success("HFトークンを保存しました。ページを再読み込みすると反映されます。")
1356
  # 保存データの存在確認(永続ストレージ優先)
1357
  # persistent_user_manager 完全廃止
1358
  user_id_manager = managers["user_id_manager"] # フォールバック用