Update app.py

app.py

@@ -399,53 +399,18 @@ class ScreenplayDatabase:
         return session_id
 
 # In ScreenplayDatabase.save_stage method, add page count tracking
-@staticmethod
-def save_stage(session_id: str, stage_number: int, stage_name: str,
-               role: str, content: str, status: str = 'complete'):
-    page_count = 0
-    if role == "screenwriter" and content:
-        # Use the accurate page calculation
-        system = ScreenplayGenerationSystem()
-        page_count = system.calculate_screenplay_pages(content)
-    
-    with ScreenplayDatabase.get_db() as conn:
-        cursor = conn.cursor()
-        cursor.execute('''
-            INSERT INTO screenplay_stages 
-            (session_id, stage_number, stage_name, role, content, page_count, status)
-            VALUES (?, ?, ?, ?, ?, ?, ?)
-            ON CONFLICT(session_id, stage_number)
-            DO UPDATE SET content=?, page_count=?, status=?, updated_at=datetime('now')
-        ''', (session_id, stage_number, stage_name, role, content, page_count, status,
-              content, page_count, status))
-        
-        # Update total pages in session
-        if role == "screenwriter" and "Expanded" in stage_name:
-            cursor.execute('''
-                UPDATE screenplay_sessions 
-                SET total_pages = (
-                    SELECT SUM(page_count) 
-                    FROM screenplay_stages 
-                    WHERE session_id = ? 
-                    AND role = 'screenwriter' 
-                    AND stage_name LIKE '%Expanded%'
-                ),
-                updated_at = datetime('now')
-                WHERE session_id = ?
-            ''', (session_id, session_id))
-        
-        conn.commit()# In ScreenplayDatabase.save_stage method, add page count tracking
-
-        
     @staticmethod
     def save_stage(session_id: str, stage_number: int, stage_name: str,
                    role: str, content: str, status: str = 'complete'):
         page_count = 0
+
+# save_stage 함수 내부
         if role == "screenwriter" and content:
             # Use the accurate page calculation
-            system = ScreenplayGenerationSystem()
+            system = ScreenplayGenerationSystem()  # ❌ 매번 새 인스턴스 생성
             page_count = system.calculate_screenplay_pages(content)
         
+        
         with ScreenplayDatabase.get_db() as conn:
             cursor = conn.cursor()
             cursor.execute('''
@@ -472,8 +437,9 @@ def save_stage(session_id: str, stage_number: int, stage_name: str,
                     WHERE session_id = ?
                 ''', (session_id, session_id))
             
-            conn.commit()
+            conn.commit()# In ScreenplayDatabase.save_stage method, add page count tracking
     
+            
 
 
     @staticmethod
@@ -1342,7 +1308,6 @@ Provide specific revision suggestions with improved examples."""
         
         return lang_prompts.get(language, lang_prompts["English"])
 
-    # ⭐ 여기에 새로운 함수 추가 (기존 create_script_doctor_prompt 바로 아래)
     def create_script_doctor_expansion_prompt(self, act_content: str, act: str,
                                              screenplay_type: str, genre: str, 
                                              language: str) -> str:
@@ -1355,13 +1320,89 @@ Provide specific revision suggestions with improved examples."""
         expansion_needed = target_pages - current_pages
         
         lang_prompts = {
-            "Korean": f"""당신은 스크립트 닥터입니다...""",
-            "English": f"""You are a script doctor..."""
+            "Korean": f"""당신은 스크립트 닥터입니다. {act}를 검토하고 확장 지시사항을 제공하세요.
+
+**현재 분량:** {current_pages:.1f}페이지 ({current_lines}줄)
+**목표 분량:** {target_pages}페이지
+**필요 추가분:** {expansion_needed:.1f}페이지 (약 {int(expansion_needed * 55)}줄)
+
+**작성된 내용:**
+{act_content}
+
+**확장 분석 및 지시사항:**
+
+1. **대화 확장 기회**
+   각 대화 시퀀스를 검토하고 다음을 추가하세요:
+   - 빠르게 진행된 대화에 서브텍스트 추가
+   - 감정적 비트와 반응 추가
+   - 갈등을 더 깊이 탐구할 대화
+   - 캐릭터의 내면을 드러낼 대사
+
+2. **액션/묘사 확장**
+   - 각 씬 오프닝에 분위기 묘사 추가
+   - 캐릭터 등장 시 상세한 비주얼 묘사
+   - 중요한 순간의 슬로우모션 효과
+   - 전환 장면과 브릿지 추가
+
+3. **새로운 비트 삽입**
+   - 긴장 완급을 위한 호흡 장면
+   - 캐릭터의 일상적 행동 추가
+   - 관계를 보여주는 작은 순간들
+   - 복선이 될 수 있는 디테일
+
+4. **{genre} 장르 강화**
+   다음 장르 요소를 더 추가하세요:
+   {chr(10).join([f"   - {element}" for element in GENRE_TEMPLATES[genre]['key_elements']])}
+
+5. **구체적 확장 위치**
+   각 씬별로 정확히 어디에 무엇을 추가해야 하는지 명시하세요:
+   [씬 번호] - [확장 내용] - [예상 추가 줄 수]
+
+**목표: 원본의 2배 분량으로 확장하되, 자연스럽고 필요한 내용만 추가**""",
+
+        "English": f"""You are a script doctor. Review {act} and provide expansion instructions.
+
+**Current Length:** {current_pages:.1f} pages ({current_lines} lines)
+**Target Length:** {target_pages} pages
+**Additional Needed:** {expansion_needed:.1f} pages (approx {int(expansion_needed * 55)} lines)
+
+**Written Content:**
+{act_content}
+
+**Expansion Analysis & Instructions:**
+
+1. **Dialogue Expansion Opportunities**
+   Review each dialogue sequence and add:
+   - Subtext to rushed conversations
+   - Emotional beats and reactions
+   - Deeper conflict exploration
+   - Lines revealing character interior
+
+2. **Action/Description Expansion**
+   - Atmosphere description at scene openings
+   - Detailed visual when characters appear
+   - Slow-motion effect for key moments
+   - Transition scenes and bridges
+
+3. **New Beat Insertions**
+   - Breathing scenes for pacing
+   - Character routine actions
+   - Small relationship moments
+   - Foreshadowing details
+
+4. **{genre} Genre Enhancement**
+   Add more of these genre elements:
+   {chr(10).join([f"   - {element}" for element in GENRE_TEMPLATES[genre]['key_elements']])}
+
+5. **Specific Expansion Locations**
+   Specify exactly where and what to add per scene:
+   [Scene Number] - [Expansion Content] - [Estimated Additional Lines]
+
+**Goal: Expand to double the original length with natural, necessary content**"""
         }
-        
+    
         return lang_prompts.get(language, lang_prompts["English"])
 
-# ⭐ create_script_doctor_expansion_prompt 함수 바로 아래에 추가
     def create_expansion_writer_prompt(self, act: str, original_content: str,
                                       expansion_notes: str, screenplay_type: str,
                                       language: str) -> str:
@@ -1371,10 +1412,75 @@ Provide specific revision suggestions with improved examples."""
         target_lines = target_pages * 55
         
         lang_prompts = {
-            "Korean": f"""원본 {act}를 확장 지시사항에 따라...""",
-            "English": f"""Rewrite {act} following expansion instructions..."""
+            "Korean": f"""원본 {act}를 확장 지시사항에 따라 다시 작성하세요.
+
+**목표 분량:** {target_pages}페이지 ({target_lines}줄)
+
+**원본 초고:**
+{original_content}
+
+**확장 지시사항:**
+{expansion_notes}
+
+**확장 작성 원칙:**
+
+1. **구조 유지**
+   - 원본의 씬 순서와 핵심 사건 유지
+   - 스토리 진행의 논리성 보존
+   - 캐릭터 아크의 일관성
+
+2. **자연스러운 확장**
+   - 추가 내용이 억지스럽지 않게
+   - 기존 내용과 유기적으로 연결
+   - 페이싱을 해치지 않는 선에서
+
+3. **품질 향상**
+   - 더 깊은 감정 표현
+   - 더 선명한 시각적 이미지
+   - 더 강력한 드라마틱 임팩트
+
+4. **포맷 준수**
+   - 표준 시나리오 포맷 엄격히 준수
+   - 적절한 여백과 간격
+   - 읽기 쉬운 구성
+
+**반드시 {target_lines}줄 이상의 완전한 확장 버전을 작성하세요.**""",
+
+        "English": f"""Rewrite {act} following expansion instructions.
+
+**Target Length:** {target_pages} pages ({target_lines} lines)
+
+**Original Draft:**
+{original_content}
+
+**Expansion Instructions:**
+{expansion_notes}
+
+**Expansion Principles:**
+
+1. **Structure Maintenance**
+   - Keep original scene order and key events
+   - Preserve story progression logic
+   - Character arc consistency
+
+2. **Natural Expansion**
+   - Additional content not forced
+   - Organically connected to existing
+   - Without harming pacing
+
+3. **Quality Enhancement**
+   - Deeper emotional expression
+   - Clearer visual imagery
+   - Stronger dramatic impact
+
+4. **Format Compliance**
+   - Strict standard screenplay format
+   - Proper margins and spacing
+   - Reader-friendly composition
+
+**MUST write complete expanded version of {target_lines}+ lines.**"""
         }
-        
+    
         return lang_prompts.get(language, lang_prompts["English"])
     
     def create_final_reviewer_prompt(self, complete_screenplay: str,