Upload 2 files

persistent_user_manager.py

@@ -102,22 +102,32 @@ class PersistentUserManager:
                 password=cookie_password
             )
             
-            # Cookieが準備できるまで待機（最大3回試行）
-            max_attempts = 3
+            # Cookieが準備できるまで待機（より長い待機時間で既存Cookie読み込み対応）
+            max_attempts = 10  # 試行回数を増加
+            total_wait_time = 0
+            max_wait_time = 3.0  # 最大3秒待機
+            
             for attempt in range(max_attempts):
                 if cookies.ready():
+                    logger.info(f"Cookie準備完了 ({attempt + 1}回目の試行で成功)")
                     break
-                logger.debug(f"Cookie準備待機中... ({attempt + 1}/{max_attempts})")
-                if attempt < max_attempts - 1:
-                    import time
-                    time.sleep(0.1)
+                
+                wait_time = 0.2 + (attempt * 0.1)  # 段階的に待機時間を延長
+                total_wait_time += wait_time
+                
+                if total_wait_time > max_wait_time:
+                    logger.warning(f"Cookie準備タイムアウト ({total_wait_time:.1f}秒経過)")
+                    break
+                
+                logger.debug(f"Cookie準備待機中... ({attempt + 1}/{max_attempts}, {wait_time:.1f}秒待機)")
+                import time
+                time.sleep(wait_time)
             
+            # Cookie準備ができていない場合でも、一度試行してみる
             if not cookies.ready():
-                logger.warning("Cookie準備タイムアウト - フォールバックモード")
-                self.cookies = None
-                self._cookie_initialized = True
-                st.session_state[browser_session_key] = True
-                return
+                logger.warning("Cookie準備未完了だが、既存Cookie読み込みを試行")
+                # フォールバックモードにせず、Cookieオブジェクトは保持
+                # 既存のCookieがある場合は読み込める可能性がある
             
             self.cookies = cookies
             self._cookie_initialized = True
@@ -237,7 +247,7 @@ class PersistentUserManager:
             return temp_id
     
     def _get_user_id_from_cookie(self) -> Optional[str]:
-        """CookieからユーザーIDを取得（永続Cookie対応）"""
+        """CookieからユーザーIDを取得（永続Cookie対応・準備未完了でも試行）"""
         try:
             # Cookie管理の初期化（初回のみ）
             self._ensure_cookie_manager()
@@ -246,35 +256,43 @@ class PersistentUserManager:
                 logger.debug("Cookie管理システム無効 - None返却")
                 return None
             
-            # Cookieが準備完了するまで待機
-            if not self.cookies.ready():
-                logger.debug("Cookie準備中 - 待機")
-                return None
-            
-            # 全てのCookieを確認（デバッグ用）
-            all_cookies = dict(self.cookies)
-            logger.debug(f"利用可能なCookie: {list(all_cookies.keys())}")
+            # Cookie準備状態をチェックするが、準備未完了でも読み込みを試行
+            is_ready = self.cookies.ready()
+            if not is_ready:
+                logger.debug("Cookie準備未完了だが、既存Cookie読み込みを試行")
+            else:
+                logger.debug("Cookie準備完了 - 通常読み込み")
             
-            user_id = self.cookies.get(self.cookie_name)
-            if user_id:
-                logger.debug(f"Cookie値取得: {user_id[:8] if len(user_id) >= 8 else user_id}...")
+            # 既存Cookieの読み込みを試行（準備状態に関係なく）
+            try:
+                # 全てのCookieを確認（デバッグ用）
+                all_cookies = dict(self.cookies)
+                logger.debug(f"利用可能なCookie: {list(all_cookies.keys())} (準備状態: {is_ready})")
                 
-                if self._is_valid_uuid(user_id):
-                    logger.info(f"HF Spaces永続CookieからユーザーID取得成功: {user_id[:8]}... (キー: {self.cookie_name})")
-                    return user_id
-                else:
-                    logger.warning(f"Cookie内のユーザーIDが無効な形式: {user_id}")
-                    return None
-            
-            logger.debug("Cookie内に有効なユーザーIDなし")
-            return None
+                user_id = self.cookies.get(self.cookie_name)
+                if user_id:
+                    logger.debug(f"Cookie値取得: {user_id[:8] if len(user_id) >= 8 else user_id}...")
+                    
+                    if self._is_valid_uuid(user_id):
+                        logger.info(f"HF Spaces永続CookieからユーザーID取得成功: {user_id[:8]}... (キー: {self.cookie_name}, 準備状態: {is_ready})")
+                        return user_id
+                    else:
+                        logger.warning(f"Cookie内のユーザーIDが無効な形式: {user_id}")
+                        return None
+                
+                logger.debug(f"Cookie内に有効なユーザーIDなし (準備状態: {is_ready})")
+                return None
+                
+            except Exception as cookie_read_e:
+                logger.warning(f"Cookie読み込み試行エラー: {cookie_read_e}")
+                return None
             
         except Exception as e:
             logger.warning(f"Cookie取得エラー: {e}")
             return None
     
     def _set_user_id_cookie(self, user_id: str):
-        """ユーザーIDを永続的なクッキーに設定"""
+        """ユーザーIDを永続的なクッキーに設定（準備未完了でも試行）"""
         try:
             # Cookie管理システムが初期化されていない場合はスキップ
             if not self._cookie_initialized:
@@ -285,16 +303,22 @@ class PersistentUserManager:
                 logger.debug("Cookie管理システム無効 - Cookie設定スキップ")
                 return
             
+            # Cookie準備状態をチェックするが、準備未完了でも設定を試行
+            is_ready = self.cookies.ready()
+            if not is_ready:
+                logger.debug("Cookie準備未完了だが、Cookie設定を試行")
+            
             # 永続的なCookieを設定
             self.cookies[self.cookie_name] = user_id
             
             # Cookieを保存（streamlit-cookies-managerは自動的に永続化される）
             self.cookies.save()
             
-            logger.info(f"HF Spaces永続クッキーに保存: {user_id[:8]}... (キー: {self.cookie_name})")
+            logger.info(f"HF Spaces永続クッキーに保存: {user_id[:8]}... (キー: {self.cookie_name}, 準備状態: {is_ready})")
             
         except Exception as e:
             logger.warning(f"永続Cookie設定エラー: {e}")
+            # エラーが発生してもアプリケーションは継続
     
     def _is_valid_uuid(self, uuid_string: str) -> bool:
         """UUIDの形式チェック"""