Update app.py

app.py

@@ -1235,6 +1235,34 @@ def create_interface():
         box-shadow: 0 10px 30px rgba(0,0,0,0.3);
     }
     
+    /* 시나리오 출력 스타일 */
+    .screenplay-output {
+        font-family: 'Courier New', monospace;
+        font-size: 14px;
+        line-height: 1.6;
+        white-space: pre-wrap;
+        background: #f8f9fa;
+        padding: 20px;
+        border-radius: 8px;
+    }
+    
+    .screenplay-output h3 {
+        color: #2c3e50;
+        margin: 20px 0 10px 0;
+        font-size: 16px;
+        font-weight: bold;
+    }
+    
+    .screenplay-output strong {
+        color: #34495e;
+        font-weight: bold;
+    }
+
+    .screenplay-output em {
+        font-style: italic;
+        color: #7f8c8d;
+    }
+    
     .act-button {
         min-height: 80px;
         font-size: 1.1rem;
@@ -1258,7 +1286,7 @@ def create_interface():
         margin: 15px 0;
     }
     """
-    
+
     with gr.Blocks(theme=gr.themes.Soft(), css=css, title="AI 시나리오 작가") as interface:
         gr.HTML("""
         <div class="main-header">
@@ -1402,6 +1430,7 @@ def create_interface():
                 session_id = system.current_session_id
                 yield progress, status, formatted, planning_data, session_id
         
+        # 4. 이벤트 핸들러 수정 - act 내용 저장 개선
         def handle_act_writing(act_name, session_id, planning_data, previous_acts):
             """막별 시나리오 작성 - generator 래퍼"""
             if not session_id or not planning_data:
@@ -1410,6 +1439,7 @@ def create_interface():
             
             try:
                 system = ScreenplayGenerationSystem()
+                final_content = ""  # 최종 콘텐츠 저장
                 
                 # Generator를 순회하며 yield
                 for act_progress, status_msg in system.generate_act(
@@ -1417,46 +1447,179 @@ def create_interface():
                 ):
                     progress_html = create_act_progress_display(act_progress)
                     screenplay_display = format_screenplay_display(act_progress.content)
+                    final_content = act_progress.content  # 원본 콘텐츠 저장
                     
                     yield progress_html, status_msg, screenplay_display
+                
+                # 완료 후 원본 콘텐츠 반환
+                if final_content:
+                    yield progress_html, f"✅ {act_name} 완성 및 저장됨", screenplay_display
                     
             except Exception as e:
                 logger.error(f"Act writing error: {e}")
                 yield f"<div>❌ 오류 발생: {str(e)}</div>", f"오류: {str(e)}", ""
-        
+
         def format_screenplay_display(content):
+            """시나리오 표시 포맷팅 - 개선된 버전"""
             if not content:
                 return "*작성 중...*"
-            
-            # 시나리오 포맷팅
-            formatted = content.replace("INT.", "**INT.**").replace("EXT.", "**EXT.**")
-            # 캐릭터명 강조 (대문자로 된 줄)
-            lines = formatted.split('\n')
+    
+            lines = content.split('\n')
             result = []
+            scene_number = 0
+    
             for line in lines:
-                if line.strip() and line.strip().isupper() and len(line.strip().split()) <= 3:
-                    result.append(f"**{line.strip()}**")
+                stripped = line.strip()
+        
+        # 빈 줄 유지
+                if not stripped:
+                    result.append("")
+                    continue
+        
+        # 씬 헤더 감지 및 포맷팅 (INT. 또는 EXT.로 시작)
+                if stripped.startswith(('INT.', 'EXT.', 'INT ', 'EXT ')):
+                    scene_number += 1
+                    # 씬 구분선 추가
+                    result.append("\n" + "="*80)
+                    result.append(f"### 씬 {scene_number}")
+                    result.append(f"**{stripped}**")
+                    result.append("-"*40)
+        # 캐릭터명 감지 (대문자로만 된 짧은 줄)
+                elif stripped.isupper() and len(stripped.split()) <= 3 and not any(c.isdigit() for c in stripped):
+                    # 캐릭터명 강조
+                    result.append(f"\n**{stripped}**")
+        # 괄호로 된 지시문 (연기 지시)
+                elif stripped.startswith('(') and stripped.endswith(')'):
+                    result.append(f"*{stripped}*")
+                # 트랜지션 (CUT TO:, FADE IN: 등)
+                elif any(trans in stripped.upper() for trans in ['CUT TO:', 'FADE IN:', 'FADE OUT:', 'DISSOLVE TO:']):
+                    result.append(f"\n_{stripped}_\n")
+                # 일반 텍스트
                 else:
                     result.append(line)
-            return '\n'.join(result)
-        
-        def combine_all_acts(act1, act2a, act2b, act3):
-            """모든 막 합치기"""
+    
+    # 최종 정리
+            formatted = '\n'.join(result)
+    
+    # 연속된 빈 줄 제거
+            while '\n\n\n' in formatted:
+                formatted = formatted.replace('\n\n\n', '\n\n')
+    
+            return formatted
+
+        # 2. combine_all_acts 함수 수정 - State 값 올바르게 처리
+        def combine_all_acts(act1_state, act2a_state, act2b_state, act3_state):
+            """모든 막 합치기 - State 값 처리 개선"""
             full = "# 🎬 전체 시나리오\n\n"
             
-            if act1 and act1 != "*1막이 여기에 표시됩니다...*":
-                full += "## 제1막 - 설정\n\n" + act1 + "\n\n---\n\n"
-            if act2a and act2a != "*2막A가 여기에 표시됩니다...*":
-                full += "## 제2막A - 상승\n\n" + act2a + "\n\n---\n\n"
-            if act2b and act2b != "*2막B가 여기에 표시됩니다...*":
-                full += "## 제2막B - 복잡화\n\n" + act2b + "\n\n---\n\n"
-            if act3 and act3 != "*3막이 여기에 표시됩니다...*":
-                full += "## 제3막 - 해결\n\n" + act3 + "\n\n"
+            # State 객체인 경우 value 속성 접근
+            act1 = act1_state.value if hasattr(act1_state, 'value') else act1_state
+            act2a = act2a_state.value if hasattr(act2a_state, 'value') else act2a_state
+            act2b = act2b_state.value if hasattr(act2b_state, 'value') else act2b_state
+            act3 = act3_state.value if hasattr(act3_state, 'value') else act3_state
             
-            return full if full != "# 🎬 전체 시나리오\n\n" else "*전체 시나리오가 완성되면 여기에 표시됩니다...*"
-        
-        def save_to_file(full_screenplay):
-            """시나리오를 파일로 저장"""
+            # 실제 콘텐츠가 있는지 확인
+            has_content = False
+            
+            if act1 and act1 != "*1막이 여기에 표시됩니다...*" and act1 != "":
+                # 마크다운 제거한 원본 텍스트 추출
+                clean_act1 = act1.replace("**", "").replace("###", "").replace("="*80, "").replace("-"*40, "")
+                full += "## 제1막 - 설정\n\n" + clean_act1 + "\n\n" + "="*80 + "\n\n"
+                has_content = True
+                
+            if act2a and act2a != "*2막A가 여기에 표시됩니다...*" and act2a != "":
+                clean_act2a = act2a.replace("**", "").replace("###", "").replace("="*80, "").replace("-"*40, "")
+                full += "## 제2막A - 상승\n\n" + clean_act2a + "\n\n" + "="*80 + "\n\n"
+                has_content = True
+                
+            if act2b and act2b != "*2막B가 여기에 표시됩니다...*" and act2b != "":
+                clean_act2b = act2b.replace("**", "").replace("###", "").replace("="*80, "").replace("-"*40, "")
+                full += "## 제2막B - 복잡화\n\n" + clean_act2b + "\n\n" + "="*80 + "\n\n"
+                has_content = True
+                
+            if act3 and act3 != "*3막이 여기에 표시됩니다...*" and act3 != "":
+                clean_act3 = act3.replace("**", "").replace("###", "").replace("="*80, "").replace("-"*40, "")
+                full += "## 제3막 - 해결\n\n" + clean_act3 + "\n\n"
+                has_content = True
+            
+            if not has_content:
+                return "*전체 시나리오가 완성되면 여기에 표시됩니다...*"
+            
+            return full
+
+        # 3. save_to_file 함수 수정 - 데이터베이스에서 직접 가져오기
+        def save_to_file(full_screenplay, session_id):
+            """시나리오를 파일로 저장 - 개선된 버전"""
+            
+            # 세션 ID가 있으면 DB에서 직접 가져오기
+            if session_id:
+                try:
+                    with ScreenplayDatabase.get_db() as conn:
+                        cursor = conn.cursor()
+                        row = cursor.execute(
+                            '''SELECT user_query, title, logline, genre, screenplay_type,
+                                      act1_content, act2a_content, act2b_content, act3_content
+                               FROM screenplay_sessions WHERE session_id = ?''',
+                            (session_id,)
+                        ).fetchone()
+                        
+                        if row:
+                            # 파일 내용 구성
+                            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+                            filename = f"screenplay_{timestamp}.txt"
+                            
+                            with open(filename, 'w', encoding='utf-8') as f:
+                                # 헤더 정보
+                                f.write("="*80 + "\n")
+                                f.write("시나리오\n")
+                                f.write("="*80 + "\n\n")
+                                
+                                if row['title']:
+                                    f.write(f"제목: {row['title']}\n")
+                                if row['genre']:
+                                    f.write(f"장르: {row['genre']}\n")
+                                if row['screenplay_type']:
+                                    f.write(f"형식: {row['screenplay_type']}\n")
+                                if row['logline']:
+                                    f.write(f"로그라인: {row['logline']}\n")
+                                
+                                f.write("\n" + "="*80 + "\n\n")
+                                
+                                # 각 막 내용 추가
+                                if row['act1_content']:
+                                    f.write("제1막 - 설정\n")
+                                    f.write("="*80 + "\n\n")
+                                    f.write(row['act1_content'])
+                                    f.write("\n\n" + "="*80 + "\n\n")
+                                
+                                if row['act2a_content']:
+                                    f.write("제2막A - 상승\n")
+                                    f.write("="*80 + "\n\n")
+                                    f.write(row['act2a_content'])
+                                    f.write("\n\n" + "="*80 + "\n\n")
+                                
+                                if row['act2b_content']:
+                                    f.write("제2막B - 복잡화\n")
+                                    f.write("="*80 + "\n\n")
+                                    f.write(row['act2b_content'])
+                                    f.write("\n\n" + "="*80 + "\n\n")
+                                
+                                if row['act3_content']:
+                                    f.write("제3막 - 해결\n")
+                                    f.write("="*80 + "\n\n")
+                                    f.write(row['act3_content'])
+                                    f.write("\n\n")
+                                
+                                # 작성 정보
+                                f.write("\n" + "="*80 + "\n")
+                                f.write(f"작성일: {datetime.now().strftime('%Y년 %m월 %d일 %H:%M')}\n")
+                                f.write("="*80 + "\n")
+                            
+                            return filename
+                except Exception as e:
+                    logger.error(f"DB에서 파일 저장 중 오류: {e}")
+            
+            # DB 접근 실패시 전달받은 내용으로 저장
             if not full_screenplay or full_screenplay == "*전체 시나리오가 완성되면 여기에 표시됩니다...*":
                 return None
             
@@ -1464,8 +1627,16 @@ def create_interface():
             filename = f"screenplay_{timestamp}.txt"
             
             with open(filename, 'w', encoding='utf-8') as f:
-                # 마크다운 제거
-                clean_text = full_screenplay.replace("**", "").replace("#", "")
+                # 모든 마크다운 및 포맷팅 제거
+                clean_text = full_screenplay
+                clean_text = clean_text.replace("**", "")
+                clean_text = clean_text.replace("###", "")
+                clean_text = clean_text.replace("#", "")
+                clean_text = clean_text.replace("="*80, "-"*80)
+                clean_text = clean_text.replace("-"*40, "")
+                clean_text = clean_text.replace("_", "")
+                clean_text = clean_text.replace("*", "")
+                
                 f.write(clean_text)
             
             return filename
@@ -1562,15 +1733,16 @@ def create_interface():
         
         # 다운로드
         download_btn.click(
-            fn=save_to_file,
-            inputs=[full_output],
+            fn=lambda full, sid: save_to_file(full, sid),
+            inputs=[full_output, current_session_id],  # session_id 추가
             outputs=[download_file]
         ).then(
-            lambda x: gr.update(visible=True) if x else gr.update(visible=False),
+            lambda x: gr.update(visible=True, value=x) if x else gr.update(visible=False),
             inputs=[download_file],
             outputs=[download_file]
         )
-    
+
+
     return interface
 
 # 메인 실행