Update app.py

app.py

@@ -485,6 +485,11 @@ class ScreenplayGenerationSystem:
             "Authorization": f"Bearer {self.api_key}"
         }
 
+    def get_session_data(self, session_id: str) -> Optional[Dict]:
+        """세션 데이터 조회 - 인스턴스 메서드"""
+        return ScreenplayDatabase.get_session_data(session_id)
+
+    
     def call_llm_streaming(self, messages: List[Dict[str, str]], max_tokens: int = 8000) -> Generator[str, None, None]:
         """LLM 호출 - 개선된 버전"""
         try:
@@ -743,23 +748,29 @@ class ScreenplayGenerationSystem:
                     previous_acts: Dict) -> Generator[Tuple[ActProgress, str], None, None]:
         """막별 시나리오 생성 - 개선된 버전"""
         
+        # act_progress를 먼저 초기화
+        act_progress = ActProgress(
+            act_name=act_name,
+            current_stage=0,
+            total_stages=0,
+            current_expert="",
+            status="initializing"
+        )
+        
         try:
             self.current_session_id = session_id
             self.planning_data = planning_data
             
             # 세션 정보 복원
-            session_data = ScreenplayDatabase.get_session_data(session_id)
+            session_data = self.get_session_data(session_id)  # 메서드 호출 수정
             if session_data:
                 self.original_query = session_data.get('user_query', '')
             
             stages = ACT_WRITING_STAGES.get(act_name, [])
-            act_progress = ActProgress(
-                act_name=act_name,
-                current_stage=0,
-                total_stages=len(stages),
-                current_expert="",
-                status="in_progress"
-            )
+            
+            # act_progress 업데이트
+            act_progress.total_stages = len(stages)
+            act_progress.status = "in_progress"
             
             act_content = ""
             
@@ -853,44 +864,81 @@ class ScreenplayGenerationSystem:
             logger.error(f"Act generation error: {e}\n{traceback.format_exc()}")
             act_progress.status = "error"
             yield act_progress, f"❌ 오류: {str(e)}"
-    
+
     def _clean_output(self, text: str) -> str:
         """출력 텍스트 정리"""
-        # 영어 단어 제거 (시나리오 포맷 제외)
-        format_terms = ["INT.", "EXT.", "CUT TO:", "FADE IN:", "FADE OUT:", "DISSOLVE TO:"]
+        # 시나리오 포맷 용어 확장
+        format_terms = [
+            "INT.", "EXT.", "CUT TO:", "FADE IN:", "FADE OUT:", 
+            "DISSOLVE TO:", "CONT'D", "O.S.", "V.O.", "POV",
+            "MONTAGE", "FLASHBACK", "DREAM SEQUENCE", "CLOSE UP",
+            "WIDE SHOT", "ESTABLISHING SHOT"
+        ]
+        
+        # 한국어 시나리오에서 사용되는 약어들
+        korean_screenplay_abbreviations = [
+            "CG", "SFX", "BGM", "OST", "NG", "OK", "VIP", "CEO", "AI",
+            "DJ", "MC", "PD", "VJ", "IT", "PC", "TV", "DVD", "USB"
+        ]
         
         lines = text.split('\n')
         cleaned_lines = []
         
         for line in lines:
             # 포맷 용어는 유지
-            if any(term in line for term in format_terms):
+            if any(term in line.upper() for term in format_terms):
                 cleaned_lines.append(line)
             else:
-                # 영어 감지 및 제거
-                if not self._contains_english(line):
+                # 약어가 포함된 경우 허용
+                contains_abbr = any(abbr in line for abbr in korean_screenplay_abbreviations)
+                
+                # 영어 감지 (약어 제외)
+                if contains_abbr or not self._contains_english(line):
                     # 불완전한 부분 제거
-                    line = line.replace("......", "").replace(".....", "").replace("...", "")
+                    line = line.replace("......", "").replace(".....", "").replace("...", ".")
                     cleaned_lines.append(line)
                 else:
-                    # 영어가 포함된 줄은 스킵하거나 경고
-                    logger.warning(f"Skipping English line: {line[:50]}")
+                    # 실제 영어가 포함된 경우만 경고
+                    if len(line.strip()) > 10:  # 짧은 줄은 무시
+                        logger.debug(f"Skipping potential English line: {line[:50]}")
         
         return '\n'.join(cleaned_lines)
-    
+
+
+
     def _contains_english(self, text: str) -> bool:
-        """영어 포함 여부 확인 (포맷 용어 제외)"""
-        format_terms = ["INT", "EXT", "CUT", "FADE", "DISSOLVE"]
+        """영어 포함 여부 확인 (포맷 용어 및 허용 단어 제외)"""
+        # 시나리오에서 허용되는 용어들
+        allowed_terms = [
+            "INT", "EXT", "CUT", "FADE", "DISSOLVE", "CONT'D",
+            "O.S.", "V.O.", "POV", "CG", "SFX", "BGM", "OST",
+            "vs", "ex", "D", "Main", "Conflict", "wang", "PO", "V"
+        ]
         
         # 포맷 용어 제거 후 확인
         test_text = text
-        for term in format_terms:
+        for term in allowed_terms:
             test_text = test_text.replace(term, "")
         
-        # 알파벳 연속 3개 이상이면 영어로 판단
+        # 알파벳 연속 4개 이상이면 영어로 판단 (3개에서 4개로 변경)
         import re
-        english_pattern = re.compile(r'[a-zA-Z]{3,}')
-        return bool(english_pattern.search(test_text))
+        english_pattern = re.compile(r'[a-zA-Z]{4,}')
+        matches = english_pattern.findall(test_text)
+        
+        # 발견된 영어 단어가 있으면 로깅하고 확인
+        if matches:
+            # 추가 필터링: 한국어 속의 약어나 이니셜은 허용
+            real_english = []
+            for match in matches:
+                if len(match) > 5:  # 5자 이상의 영어 단어만 실제 영어로 판단
+                    real_english.append(match)
+            
+            if real_english:
+                logger.debug(f"Real English words found: {real_english}")
+                return True
+        
+        return False
+
     
     def _validate_and_clean_screenplay(self, content: str) -> str:
         """시나리오 검증 및 정리"""
@@ -1107,6 +1155,23 @@ class ScreenplayGenerationSystem:
                 summary += f"[{act}]\n{content[:300]}...\n"
         return summary[:1000]
 
+
+
+    @staticmethod
+    def get_session_data(session_id: str) -> Optional[Dict]:
+        """세션 데이터 조회"""
+        with ScreenplayDatabase.get_db() as conn:
+            cursor = conn.cursor()
+            row = cursor.execute(
+                'SELECT * FROM screenplay_sessions WHERE session_id = ?',
+                (session_id,)
+            ).fetchone()
+            if row:
+                # Row 객체를 딕셔너리로 변환
+                columns = [description[0] for description in cursor.description]
+                return dict(zip(columns, row))
+        return None
+
 # UI 함수들
 def create_act_progress_display(act_progress: ActProgress) -> str:
     """막별 진행 상황 표시"""