Upload persistent_user_manager.py

persistent_user_manager.py

@@ -63,9 +63,8 @@ class PersistentUserManager:
             logger.debug("Cookie管理システム既に初期化済み - スキップ")
             return
         
-        # セッション固有のCookie初期化チェック（重複防止）
-        session_id = id(st.session_state)
-        browser_session_key = f"hf_cookie_init_{session_id}"
+        # 固定キーでCookie初期化チェック（永続化対応）
+        browser_session_key = "hf_persistent_cookie_initialized"
         
         # より厳密な重複チェック
         if (not force_init and 
@@ -74,34 +73,32 @@ class PersistentUserManager:
             hasattr(self.cookies, 'ready') and
             self.cookies.ready()):
             self._cookie_initialized = True
-            logger.debug("Cookie管理システム既に準備完了 - スキップ")
+            logger.debug("永続Cookie管理システム既に準備完了 - スキップ")
             return
         
         # 初期化処理中フラグを設定（重複初期化防止）
-        init_in_progress_key = f"hf_cookie_initializing_{session_id}"
+        init_in_progress_key = "hf_persistent_cookie_initializing"
         if st.session_state.get(init_in_progress_key, False):
-            logger.warning("Cookie初期化処理中 - 重複初期化を防止")
+            logger.warning("永続Cookie初期化処理中 - 重複初期化を防止")
             return
         
         st.session_state[init_in_progress_key] = True
         
         try:
-            logger.info("HF Spaces永続Cookie管理システム初期化開始...")
+            logger.info("HF Spaces永続Cookie管理システム初期化開始（固定prefix使用）...")
             
             # セキュアなパスワードを生成（環境変数から取得、なければ生成）
             cookie_password = os.getenv("MARI_COOKIE_PASSWORD", "mari_chat_secure_key_2024")
             
-            # セッション固有のユニークなprefixを生成（重複回避）
-            session_id = id(st.session_state)
-            import time
-            timestamp = int(time.time() * 1000)  # ミリ秒タイムスタンプ
-            unique_prefix = f"hf_{session_id}_{timestamp}_"
+            # 固定のprefixを使用（永続Cookie読み込みのため）
+            # リロード後も同じCookieを読み込めるよう、固定のprefixを使用
+            fixed_prefix = "hf_persistent_mari_"
             
-            logger.debug(f"Cookie初期化: prefix={unique_prefix}, session_id={session_id}, timestamp={timestamp}")
+            logger.debug(f"Cookie初期化: prefix={fixed_prefix} (固定prefix使用)")
             
             # EncryptedCookieManagerを初期化（永続Cookie対応）
             cookies = EncryptedCookieManager(
-                prefix=unique_prefix,  # タイムスタンプ付きの完全にユニークなprefix
+                prefix=fixed_prefix,  # 固定prefixで永続Cookie読み込み対応
                 password=cookie_password
             )
             
@@ -125,35 +122,39 @@ class PersistentUserManager:
             self.cookies = cookies
             self._cookie_initialized = True
             
-            # セッション固有のフラグを設定
+            # 永続化フラグを設定
             st.session_state[browser_session_key] = True
             
             # 初期化処理中フラグをクリア
-            init_in_progress_key = f"hf_cookie_initializing_{session_id}"
             if init_in_progress_key in st.session_state:
                 del st.session_state[init_in_progress_key]
             
             # 既存Cookieの確認
             try:
                 existing_cookies = dict(self.cookies)
-                logger.info(f"HF Spaces永続Cookie管理システム初期化完了 - prefix: {unique_prefix}, 既存Cookie数: {len(existing_cookies)}")
+                logger.info(f"HF Spaces永続Cookie管理システム初期化完了 - prefix: {fixed_prefix}, 既存Cookie数: {len(existing_cookies)}")
+                
+                # 既存のユーザーIDがあるかチェック
+                if self.cookie_name in existing_cookies:
+                    existing_user_id = existing_cookies[self.cookie_name]
+                    logger.info(f"既存の永続Cookie発見: ユーザーID {existing_user_id[:8]}...")
+                else:
+                    logger.info("新規ユーザー: 永続Cookieなし")
+                    
             except Exception as cookie_check_e:
                 logger.debug(f"Cookie確認エラー（通常動作に影響なし）: {cookie_check_e}")
-                logger.info(f"HF Spaces永続Cookie管理システム初期化完了 - prefix: {unique_prefix}")
+                logger.info(f"HF Spaces永続Cookie管理システム初期化完了 - prefix: {fixed_prefix}")
             
         except Exception as e:
-            logger.error(f"Cookie管理初期化エラー: {e}")
+            logger.error(f"永続Cookie管理初期化エラー: {e}")
             
             # 初期化処理中フラグをクリア
-            session_id = id(st.session_state)
-            init_in_progress_key = f"hf_cookie_initializing_{session_id}"
             if init_in_progress_key in st.session_state:
                 del st.session_state[init_in_progress_key]
             
             # フォールバック: セッション状態のみ使用
             self.cookies = None
             self._cookie_initialized = True
-            browser_session_key = f"hf_cookie_init_{session_id}"
             st.session_state[browser_session_key] = True
     
     def get_or_create_user_id(self, force_reset: bool = False) -> str: