Update app.py

app.py

@@ -2872,31 +2872,32 @@ if __name__ == "__main__":
             return value.strip() if value else None
         return None
 
+import re
 from typing import List
 
     def _parse_character_profile(self, content: str, role: str) -> CharacterProfile:
-        """Parse character profile from content and return CharacterProfile dataclass"""
+        """Parse character profile from content and return a CharacterProfile object."""
         # --- 1. 로그 ---
         logger.debug(f"Parsing character profile for role: {role}")
         logger.debug(f"Content preview: {content[:200]}...")
-    
+
         # --- 2. 이름 추출 ---
-        name = f"Character_{role}"          # fallback
+        name = f"Character_{role}"  # fallback
         name_patterns = [
-            r'(?:이름|Name)[:\s]*([^\n,(]+)',      # 예: "이름: 홍길동"
-            r'^\s*[-*•]\s*([^\n,(]+)',             # 예: "- 홍길동"
-            r'^([^\n,(]+)'                         # 문단 첫 단어
+            r'(?:이름|Name)[:\s]*([^\n,(]+)',     # 예: "이름: 홍길동"
+            r'^\s*[-*•]\s*([^\n,(]+)',            # 예: "- 홍길동"
+            r'^([^\n,(]+)'                        # 문단 첫 단어
         ]
         for pat in name_patterns:
             m = re.search(pat, content, re.IGNORECASE | re.MULTILINE)
             if m and m.group(1).strip():
                 extracted = m.group(1).strip()
-                # 마크다운 기호·불필요 문자 제거
+                # 마크다운·특수 기호 제거
                 extracted = re.sub(r'[\*:\s]+', '', extracted)
                 if len(extracted) > 1:
                     name = extracted
                     break
-    
+
         # --- 3. 필드 추출 헬퍼 ---
         def extract_clean_field(patterns) -> str:
             patterns = [patterns] if isinstance(patterns, str) else patterns
@@ -2910,8 +2911,8 @@ from typing import List
                     val = re.sub(r'\s+', ' ', val)
                     return val
             return ""
-            
-        # Extract all fields with safer extraction
+
+        # --- 4. Personality(여러 줄) 추출은 별도 메서드 사용 ---
         profile = CharacterProfile(
             name=name,
             role=role,
@@ -2953,37 +2954,144 @@ from typing import List
                 r"변화"
             ])
         )
-        
+
         logger.debug(f"Parsed character: {profile.name}")
         return profile
 
     def _extract_personality_traits(self, content: str) -> List[str]:
-        """Extract personality traits from content"""
-        traits = []
-        # Look for personality section with multiple pattern options
+        """Extract personality traits from content."""
+        traits: List[str] = []
         personality_patterns = [
             r"(?:Personality|성격 특성|성격)[:\s]*([^\n]+(?:\n(?![\w가-힣]+:)[^\n]+)*)",
             r"성격[:\s]*(?:\n?[-•*]\s*[^\n]+)+"
         ]
-        
+
         for pattern in personality_patterns:
             match = re.search(pattern, content, re.IGNORECASE | re.DOTALL)
             if match and match.group(1):
-                personality_section = match.group(1)
-                # Extract individual traits (usually listed)
-                trait_lines = personality_section.split('\n')
-                for line in trait_lines:
+                section = match.group(1)
+                for line in section.split('\n'):
                     line = line.strip()
                     if line and not line.endswith(':'):
-                        # Remove list markers
-                        trait = re.sub(r'^\s*[-•*]\s*', '', line)
-                        trait = re.sub(r'^\d+\.\s*', '', trait)  # Remove numbered lists
-                        if trait and len(trait) > 2:  # Skip very short entries
+                        trait = re.sub(r'^\s*[-•*]\s*', '', line)   # 글머리표 제거
+                        trait = re.sub(r'^\d+\.\s*', '', trait)     # 번호 제거
+                        if len(trait) > 2:
                             traits.append(trait)
-                if traits:  # If we found traits, stop looking
+                if traits:
                     break
+
+        return traits[:5]  # 최대 5개
         
-        return traits[:5]  # Limit to 5 traits
+import re
+from typing import List
+
+    def _parse_character_profile(self, content: str, role: str) -> CharacterProfile:
+        """Parse character profile from content and return a CharacterProfile object."""
+        # --- 1. 로그 ---
+        logger.debug(f"Parsing character profile for role: {role}")
+        logger.debug(f"Content preview: {content[:200]}...")
+
+        # --- 2. 이름 추출 ---
+        name = f"Character_{role}"  # fallback
+        name_patterns = [
+            r'(?:이름|Name)[:\s]*([^\n,(]+)',     # 예: "이름: 홍길동"
+            r'^\s*[-*•]\s*([^\n,(]+)',            # 예: "- 홍길동"
+            r'^([^\n,(]+)'                        # 문단 첫 단어
+        ]
+        for pat in name_patterns:
+            m = re.search(pat, content, re.IGNORECASE | re.MULTILINE)
+            if m and m.group(1).strip():
+                extracted = m.group(1).strip()
+                # 마크다운·특수 기호 제거
+                extracted = re.sub(r'[\*:\s]+', '', extracted)
+                if len(extracted) > 1:
+                    name = extracted
+                    break
+
+        # --- 3. 필드 추출 헬퍼 ---
+        def extract_clean_field(patterns) -> str:
+            patterns = [patterns] if isinstance(patterns, str) else patterns
+            for p in patterns:
+                m = re.search(rf'{p}[:\s]*([^\n*]+?)(?=\n|$)', content,
+                              re.IGNORECASE | re.DOTALL)
+                if m and m.group(1).strip():
+                    val = m.group(1).strip()
+                    val = re.sub(r'^[-*•:\s]+', '', val)
+                    val = re.sub(r'\*+', '', val)
+                    val = re.sub(r'\s+', ' ', val)
+                    return val
+            return ""
+
+        # --- 4. Personality(여러 줄) 추출은 별도 메서드 사용 ---
+        profile = CharacterProfile(
+            name=name,
+            role=role,
+            archetype=extract_clean_field([
+                r"캐릭터 아크타입",
+                r"Character Archetype",
+                r"Archetype",
+                r"아크타입"
+            ]),
+            want=extract_clean_field([
+                r"WANT\s*\(외적 목표\)",
+                r"WANT",
+                r"외적 목표",
+                r"External Goal"
+            ]),
+            need=extract_clean_field([
+                r"NEED\s*\(내적 필요\)",
+                r"NEED",
+                r"내적 필요",
+                r"Internal Need"
+            ]),
+            backstory=extract_clean_field([
+                r"백스토리",
+                r"Backstory",
+                r"핵심 상처",
+                r"Core Wound"
+            ]),
+            personality=self._extract_personality_traits(content),
+            speech_pattern=extract_clean_field([
+                r"말투.*?패턴",
+                r"Speech Pattern",
+                r"언어 패턴",
+                r"말투"
+            ]),
+            character_arc=extract_clean_field([
+                r"캐릭터 아크",
+                r"Character Arc",
+                r"Arc",
+                r"변화"
+            ])
+        )
+
+        logger.debug(f"Parsed character: {profile.name}")
+        return profile
+
+    def _extract_personality_traits(self, content: str) -> List[str]:
+        """Extract personality traits from content."""
+        traits: List[str] = []
+        personality_patterns = [
+            r"(?:Personality|성격 특성|성격)[:\s]*([^\n]+(?:\n(?![\w가-힣]+:)[^\n]+)*)",
+            r"성격[:\s]*(?:\n?[-•*]\s*[^\n]+)+"
+        ]
+
+        for pattern in personality_patterns:
+            match = re.search(pattern, content, re.IGNORECASE | re.DOTALL)
+            if match and match.group(1):
+                section = match.group(1)
+                for line in section.split('\n'):
+                    line = line.strip()
+                    if line and not line.endswith(':'):
+                        trait = re.sub(r'^\s*[-•*]\s*', '', line)   # 글머리표 제거
+                        trait = re.sub(r'^\d+\.\s*', '', trait)     # 번호 제거
+                        if len(trait) > 2:
+                            traits.append(trait)
+                if traits:
+                    break
+
+        return traits[:5]  # 최대 5개
+
 
     def _process_character_content(self, content: str):
         """Process character designer output with better error handling"""