Update cache_db.py

cache_db.py

@@ -576,7 +576,7 @@ class ContentVectorCache:
             "failed_count": len(self._index_status["failed"])
         }
 class UserProfileCache:
-    """사용자 프로필 저장/로드 클래스 (이메일 기반)"""
+    """사용자 프로필 저장/로드 클래스 (이메일 기반, 키워드 포함)"""
     def __init__(self):
         self.profiles_file = PROFILE_DIR / "profiles.json"
         self.profiles = {}
@@ -601,30 +601,54 @@ class UserProfileCache:
     def _normalize_email(self, email: str) -> str:
         """이메일 정규화 (소문자 변환, 공백 제거)"""
         return email.strip().lower()
-    def save_profile(self, email: str, profile_data: dict) -> Tuple[bool, str]:
-        """프로필 저장"""
+    def save_profile(self, email: str, profile_data: dict, keywords: List[str] = None) -> Tuple[bool, str]:
+        """프로필 저장 (키워드 포함)"""
         if not email or '@' not in email:
             return False, "올바른 이메일 주소를 입력해주세요."
         email_key = self._normalize_email(email)
+        
+        # 키워드 정리 (최대 5개, 빈 문자열 제거 후 저장)
+        clean_keywords = []
+        if keywords:
+            clean_keywords = [kw.strip() for kw in keywords if kw and kw.strip()][:5]
+        
         self.profiles[email_key] = {
             "email": email_key,
             "profile": profile_data,
+            "keywords": clean_keywords,  # ⭐ 키워드 추가
             "updated_at": datetime.now(KST).isoformat(),
             "created_at": self.profiles.get(email_key, {}).get("created_at", datetime.now(KST).isoformat())
         }
         self._save_profiles()
-        logger.info(f"Profile saved for: {email_key}")
-        return True, f"✅ 프로필이 저장되었습니다. ({email_key})"
-    def load_profile(self, email: str) -> Tuple[Optional[dict], str]:
-        """프로필 로드"""
+        kw_str = ", ".join(clean_keywords) if clean_keywords else "없음"
+        logger.info(f"Profile saved for: {email_key} (keywords: {kw_str})")
+        return True, f"✅ 프로필이 저장되었습니다. ({email_key}, 키워드 {len(clean_keywords)}개)"
+    def load_profile(self, email: str) -> Tuple[Optional[dict], str, List[str]]:
+        """프로필 로드 (키워드 포함) - 3개 값 반환"""
         if not email or '@' not in email:
-            return None, "올바른 이메일 주소를 입력해주세요."
+            return None, "올바른 이메일 주소를 입력해주세요.", []
         email_key = self._normalize_email(email)
         if email_key in self.profiles:
             data = self.profiles[email_key]
+            profile = data.get("profile", {})
+            keywords = data.get("keywords", [])
+            # 키워드가 5개 미만이면 빈 문자열로 패딩
+            while len(keywords) < 5:
+                keywords.append("")
             logger.info(f"Profile loaded for: {email_key}")
-            return data.get("profile", {}), f"✅ 프로필을 불러왔습니다. (마지막 수정: {data.get('updated_at', '알 수 없음')[:10]})"
-        return None, f"ℹ️ '{email_key}'에 저장된 프로필이 없습니다. 새로 입력해주세요."
+            return profile, f"✅ 프로필을 불러왔습니다. (마지막 수정: {data.get('updated_at', '알 수 없음')[:10]})", keywords[:5]
+        return None, f"ℹ️ '{email_key}'에 저장된 프로필이 없습니다. 새로 입력해주세요.", []
+    def get_profile_keywords(self, email: str) -> List[str]:
+        """키워드만 조회"""
+        if not email or '@' not in email:
+            return []
+        email_key = self._normalize_email(email)
+        if email_key not in self.profiles:
+            return []
+        keywords = self.profiles[email_key].get("keywords", [])
+        while len(keywords) < 5:
+            keywords.append("")
+        return keywords[:5]
     def delete_profile(self, email: str) -> Tuple[bool, str]:
         """프로필 삭제"""
         email_key = self._normalize_email(email)
@@ -636,6 +660,9 @@ class UserProfileCache:
     def get_profile_count(self) -> int:
         """저장된 프로필 수"""
         return len(self.profiles)
+    def get_all_profiles(self) -> dict:
+        """모든 프로필 반환 (관리자용)"""
+        return self.profiles
 _cache_instance = None
 _content_cache_instance = None
 _profile_cache_instance = None
@@ -934,5 +961,4 @@ def initialize_cache_system():
 if __name__ == "__main__":
     print("Testing cache system...")
     status = initialize_cache_system()
-    print(f"Status: {json.dumps(status, ensure_ascii=False, indent=2)}")
-    
+    print(f"Status: {json.dumps(status, ensure_ascii=False, indent=2)}")