Upload 3 files

main_app.py

@@ -19,6 +19,20 @@ if sys.platform.startswith('win'):
 # .envファイルから環境変数を読み込み
 load_dotenv()
 
+# PyTorchのクラス登録問題を回避するための設定
+try:
+    import torch
+    # スレッド数を制限してクラス登録問題を回避
+    torch.set_num_threads(int(os.getenv("TORCH_NUM_THREADS", "1")))
+    # 警告を抑制
+    import warnings
+    warnings.filterwarnings("ignore", category=UserWarning, module="torch")
+    logger.info("PyTorch初期化設定完了")
+except ImportError:
+    logger.info("PyTorchが利用できません - ルールベース処理を使用")
+except Exception as e:
+    logger.warning(f"PyTorch初期化設定エラー: {e}")
+
 # ロガー設定
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -504,32 +518,29 @@ def initialize_session_state(managers, force_reset_override=False):
     user_id_manager = managers["user_id_manager"]  # フォールバック用
     session_api_client = managers["session_api_client"]
     
-    # セッションIDを取得または生成（重複防止・統一化）
+    # Cookie認証ベースのユーザーID取得（起動時uuid4()生成を排除）
     # 重複ID生成防止: 既にuser_idがあり、force_resetでない場合は既存IDを使用
     if 'user_id' in st.session_state and not force_reset:
         session_id = st.session_state.user_id
         logger.debug(f"既存ユーザーID使用（rerun対応）: {session_id[:8]}...")
     else:
-        # 新規作成またはリセット時のみ
-        session_id = None
-        
+        # Cookie認証ベースでユーザーIDを取得（新規作成またはリセット時のみ）
         try:
-            # 永続ストレージからCookieベースでユーザーIDを取得
-            session_id = persistent_user_manager.get_or_create_user_id()
-            logger.info(f"永続ストレージからユーザーIDを取得: {session_id[:8]}...")
+            # Cookie認証を最優先で使用
+            session_id = persistent_user_manager.get_or_create_user_id(force_reset=force_reset)
+            logger.info(f"Cookie認証ベースユーザーID取得: {session_id[:8]}...")
         except Exception as e:
-            logger.warning(f"永続ストレージ取得エラー、フォールバック使用: {e}")
+            logger.error(f"Cookie認証失敗: {e}")
             
+            # フォールバック: 従来のローカルファイル方式（uuid4()生成を避ける）
             try:
-                # フォールバック: 従来のローカルファイル方式
                 session_id = user_id_manager.get_or_create_user_id()
-                logger.info(f"フォールバック: ローカルファイルからユーザーIDを取得: {session_id[:8]}...")
+                logger.warning(f"フォールバック: ローカルファイル方式使用: {session_id[:8]}...")
             except Exception as e2:
-                logger.error(f"フォールバックも失敗: {e2}")
-                # 最終フォールバック: 一時的なIDを生成
-                import uuid
-                session_id = str(uuid.uuid4())
-                logger.warning(f"最終フォールバック: 一時的なIDを生成: {session_id[:8]}...")
+                logger.error(f"全ての認証方式が失敗: {e2}")
+                # 最終フォールバック: セッション固有の一時的なIDを生成（起動時uuid4()を避ける）
+                session_id = f"temp_{id(st.session_state)}"
+                logger.warning(f"最終フォールバック: セッション固有一時ID: {session_id}")
     
     # ユーザーIDとしてセッションIDを使用
     session_changed = ('user_id' not in st.session_state or 

persistent_user_manager.py

@@ -57,19 +57,21 @@ class PersistentUserManager:
             logger.warning(f"フォールバック: ローカルディレクトリを使用 {self.user_data_dir}")
     
     def _ensure_cookie_manager(self, force_init: bool = False):
-        """Cookie管理システムの初期化（初回またはフルリセット時のみ）"""
+        """Cookie管理システムの初期化（ブラウザごとに一度だけ）"""
         # 既に初期化済みで、強制初期化でない場合はスキップ
         if self._cookie_initialized and not force_init:
+            logger.debug("Cookie管理システム既に初期化済み")
             return
         
-        # アプリ全体でのCookie初期化状態をチェック（グローバルフラグ）
-        if not force_init and st.session_state.get('cookie_manager_initialized', False):
-            logger.debug("Cookie管理システム既に初期化済み - スキップ")
+        # ブラウザセッション単位でのCookie初期化チェック
+        browser_session_key = f"cookie_initialized_{id(st.session_state)}"
+        if not force_init and st.session_state.get(browser_session_key, False):
+            logger.debug("このブラウザセッションでCookie管理システム既に初期化済み")
             self._cookie_initialized = True
             return
         
         try:
-            logger.info("Cookie管理システム初期化開始...")
+            logger.info("Cookie管理システム初期化開始（ブラウザセッション単位）...")
             
             # セキュアなパスワードを生成（環境変数から取得、なければ生成）
             cookie_password = os.getenv("MARI_COOKIE_PASSWORD", "mari_chat_secure_key_2024")
@@ -81,26 +83,27 @@ class PersistentUserManager:
             
             # Cookieが準備できるまで待機
             if not cookies.ready():
+                logger.warning("Cookie準備中 - 待機")
                 st.stop()
             
             self.cookies = cookies
             self._cookie_initialized = True
             
-            # グローバルフラグを設定（アプリ全体で一度だけ初期化）
-            st.session_state.cookie_manager_initialized = True
+            # ブラウザセッション単位でフラグを設定
+            st.session_state[browser_session_key] = True
             
-            logger.info("Cookie管理システム初期化完了")
+            logger.info("Cookie管理システム初期化完了（ブラウザセッション単位）")
             
         except Exception as e:
             logger.error(f"Cookie管理初期化エラー: {e}")
             # フォールバック: セッション状態のみ使用
             self.cookies = None
             self._cookie_initialized = True
-            st.session_state.cookie_manager_initialized = True
+            st.session_state[browser_session_key] = True
     
     def get_or_create_user_id(self, force_reset: bool = False) -> str:
         """
-        ユーザーIDを取得または新規作成（Cookie連携・重複防止）
+        Cookie認証ベースのユーザーID取得（ブラウザごとに一つのCookie）
         
         Args:
             force_reset: フルリセット時のみTrue
@@ -109,69 +112,52 @@ class PersistentUserManager:
             ユーザーID
         """
         try:
-            # 最優先: 既にStreamlitセッション状態にuser_idがある場合はそれを使用（フルリセット以外）
+            # フルリセット時以外で、既にセッション状態にuser_idがある場合はそれを使用
             if not force_reset and 'user_id' in st.session_state:
                 existing_id = st.session_state.user_id
                 if existing_id and self._is_valid_uuid(existing_id):
-                    logger.debug(f"Streamlitセッション状態からユーザーID使用: {existing_id[:8]}...")
+                    logger.debug(f"セッション状態からユーザーID使用: {existing_id[:8]}...")
                     return existing_id
             
-            # 重複チェック防止: 既にセッション状態にある場合はそれを使用（フルリセット以外）
-            if not force_reset and hasattr(st.session_state, 'persistent_user_id_checked'):
-                existing_id = st.session_state.get('persistent_user_id')
-                if existing_id and self._is_valid_uuid(existing_id):
-                    logger.debug(f"既にチェック済みのユーザーID使用: {existing_id[:8]}...")
-                    return existing_id
+            # Cookie管理システムを初期化（ブラウザごとに一度だけ）
+            self._ensure_cookie_manager(force_init=force_reset)
             
-            # Cookie初期化は初回またはフルリセット時のみ
-            if not self._cookie_initialized or force_reset:
-                # 1. Cookieから��ーザーIDを取得
-                user_id = self._get_user_id_from_cookie()
-                
-                if user_id and self._is_valid_user_id(user_id):
-                    # 有効なユーザーIDが存在
-                    self._update_user_access_time(user_id)
-                    st.session_state.persistent_user_id = user_id
-                    st.session_state.persistent_user_id_checked = True
-                    logger.info(f"CookieからユーザーID取得: {user_id[:8]}...")
-                    return user_id
-                
-                # 2. セッション状態から取得を試行（Cookieが無効な場合のみ）
-                if not force_reset:
-                    user_id = st.session_state.get('persistent_user_id')
-                    if user_id and self._is_valid_user_id(user_id):
-                        # セッション状態にあるIDを使用してCookieを更新
-                        self._set_user_id_cookie(user_id)
-                        self._update_user_access_time(user_id)
-                        st.session_state.persistent_user_id_checked = True
-                        logger.info(f"セッション状態からユーザーID復元: {user_id[:8]}...")
-                        return user_id
-                
-                # 3. 新規ユーザーIDを作成
-                user_id = self._create_new_user()
-                st.session_state.persistent_user_id_checked = True
-                logger.info(f"新規ユーザーID作成: {user_id[:8]}...")
+            # 1. CookieからユーザーIDを取得（最優先）
+            user_id = self._get_user_id_from_cookie()
+            
+            if user_id and self._is_valid_user_id(user_id):
+                # 有効なCookieベースのユーザーIDが存在
+                self._update_user_access_time(user_id)
+                st.session_state.user_id = user_id  # セッション状態に保存
+                logger.info(f"Cookie認証成功: {user_id[:8]}...")
                 return user_id
-            else:
-                # Cookie初期化済みの場合は、セッション状態から取得
-                existing_id = st.session_state.get('persistent_user_id')
-                if existing_id and self._is_valid_uuid(existing_id):
-                    logger.debug(f"Cookie初期化済み - セッション状態からユーザーID使用: {existing_id[:8]}...")
-                    return existing_id
-                
-                # フォールバック: 新規作成
-                user_id = self._create_new_user()
-                st.session_state.persistent_user_id_checked = True
-                logger.info(f"フォールバック新規ユーザーID作成: {user_id[:8]}...")
+            
+            # 2. Cookieが無効または存在しない場合は新規作成
+            if force_reset or not user_id:
+                user_id = self._create_new_user_with_cookie()
+                logger.info(f"新規Cookie認証ユーザー作成: {user_id[:8]}...")
+                return user_id
+            
+            # 3. 最終フォールバック（Cookieが無効だが何らかのIDがある場合）
+            if user_id and self._is_valid_uuid(user_id):
+                # UUIDとして有効だが、ユーザーファイルが存在しない場合は新規作成
+                user_id = self._create_new_user_with_cookie()
+                logger.info(f"フォールバック新規ユーザー作成: {user_id[:8]}...")
                 return user_id
             
+            # 4. 完全フォールバック（Cookie無効時）
+            logger.warning("Cookie認証失敗 - 一時的なセッションIDを使用")
+            temp_id = str(uuid.uuid4())
+            st.session_state.user_id = temp_id
+            return temp_id
+            
         except Exception as e:
-            logger.error(f"ユーザーID取得エラー: {e}")
-            # フォールバック: 一時的なIDを生成
-            fallback_id = str(uuid.uuid4())
-            st.session_state.persistent_user_id = fallback_id
-            st.session_state.persistent_user_id_checked = True
-            return fallback_id
+            logger.error(f"Cookie認証エラー: {e}")
+            # 完全フォールバック: 一時的なIDを生成（Cookieに保存しない）
+            temp_id = str(uuid.uuid4())
+            st.session_state.user_id = temp_id
+            logger.warning(f"一時的なセッションID使用: {temp_id[:8]}...")
+            return temp_id
     
     def _get_user_id_from_cookie(self) -> Optional[str]:
         """CookieからユーザーIDを取得"""
@@ -239,8 +225,8 @@ class PersistentUserManager:
             logger.warning(f"ユーザーID検証エラー: {e}")
             return False
     
-    def _create_new_user(self) -> str:
-        """新規ユーザーを作成"""
+    def _create_new_user_with_cookie(self) -> str:
+        """Cookie認証ベースの新���ユーザーを作成"""
         try:
             user_id = str(uuid.uuid4())
             
@@ -250,6 +236,7 @@ class PersistentUserManager:
                 "created_at": datetime.now().isoformat(),
                 "last_access": datetime.now().isoformat(),
                 "version": "1.0",
+                "browser_fingerprint": self._generate_browser_fingerprint(),
                 "game_data": {
                     "affection": 30,
                     "messages": [{"role": "assistant", "content": "何の用？遊びに来たの？", "is_initial": True}],
@@ -267,19 +254,43 @@ class PersistentUserManager:
             with open(user_file, 'w', encoding='utf-8') as f:
                 json.dump(user_data, f, ensure_ascii=False, indent=2)
             
-            # Cookieとセッション状態に保存
+            # Cookieに保存（ブラウザごとに一つ）
             self._set_user_id_cookie(user_id)
-            st.session_state.persistent_user_id = user_id
             
-            logger.info(f"新規ユーザー作成完了: {user_id[:8]}...")
+            # セッション状態に保存
+            st.session_state.user_id = user_id
+            
+            logger.info(f"Cookie認証ベース新規ユーザー作成完了: {user_id[:8]}...")
             return user_id
             
         except Exception as e:
-            logger.error(f"新規ユーザー作成エラー: {e}")
-            # フォールバック
-            fallback_id = str(uuid.uuid4())
-            st.session_state.persistent_user_id = fallback_id
-            return fallback_id
+            logger.error(f"Cookie認証ベース新規ユーザー作成エラー: {e}")
+            # フォールバック: 一時的なIDを生成（Cookieに保存しない）
+            temp_id = str(uuid.uuid4())
+            st.session_state.user_id = temp_id
+            logger.warning(f"フォールバック一時ID: {temp_id[:8]}...")
+            return temp_id
+    
+    def _create_new_user(self) -> str:
+        """従来の新規ユーザー作成（後方互換性のため残す）"""
+        return self._create_new_user_with_cookie()
+    
+    def _generate_browser_fingerprint(self) -> str:
+        """ブラウザフィンガープリントを生成（簡易版）"""
+        try:
+            # Streamlitのセッション情報を使用してフィンガープリントを生成
+            import hashlib
+            
+            # セッション固有の情報を組み合わせ
+            session_info = f"{id(st.session_state)}_{datetime.now().strftime('%Y%m%d')}"
+            fingerprint = hashlib.md5(session_info.encode()).hexdigest()[:16]
+            
+            logger.debug(f"ブラウザフィンガープリント生成: {fingerprint}")
+            return fingerprint
+            
+        except Exception as e:
+            logger.warning(f"ブラウザフィンガープリント生成エラー: {e}")
+            return "unknown_browser"
     
     def _update_user_access_time(self, user_id: str):
         """ユーザーの最終アクセス時刻を更新"""

requirements.txt

@@ -6,8 +6,9 @@ groq>=0.4.0
 requests>=2.31.0
 unidic-lite
 typing-extensions>=4.0.0
-transformers>=4.20.0
-torch>=1.12.0
+# PyTorchとTransformers（クラス登録問題対応版）
+torch>=1.12.0,<2.0.0
+transformers>=4.20.0,<5.0.0
 sentencepiece>=0.1.95
 fugashi>=1.1.0
 ipadic>=1.0.0