Update app.py

app.py

@@ -398,18 +398,20 @@ class ScreenplayDatabase:
             conn.commit()
         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
-
-# save_stage 함수 내부
+    
+        # save_stage 함수 내부
         if role == "screenwriter" and content:
-            # Use the accurate page calculation
-            system = ScreenplayGenerationSystem()  # ❌ 매번 새 인스턴스 생성
-            page_count = system.calculate_screenplay_pages(content)
-        
+            # 페이지 계산을 여기서 더 정확하게 수행
+            if "Expanded Final Version" in stage_name:
+                # 확장 최종 버전은 정확한 계산 사용
+                page_count = ScreenplayGenerationSystem.calculate_screenplay_pages(content)
+            else:
+                # 초안은 간단한 계산 사용
+                page_count = len(content.split('\n')) / 55
         
         with ScreenplayDatabase.get_db() as conn:
             cursor = conn.cursor()
@@ -437,16 +439,14 @@ class ScreenplayDatabase:
                     WHERE session_id = ?
                 ''', (session_id, session_id))
             
-            conn.commit()# In ScreenplayDatabase.save_stage method, add page count tracking
+            conn.commit()
+    
     
-            
-
-
     @staticmethod
     def save_screenplay_bible(session_id: str, bible: ScreenplayBible):
         """Save screenplay bible"""
         with ScreenplayDatabase.get_db() as conn:
-            bible_json = json.dumps(asdict(bible))
+        bible_json = json.dumps(asdict(bible))
             conn.cursor().execute(
                 'UPDATE screenplay_sessions SET screenplay_bible = ? WHERE session_id = ?',
                 (bible_json, session_id)
@@ -1697,12 +1697,12 @@ Provide specific solutions for each issue."""
         # For now, returning a placeholder
         return f"Scenes for {act} from the breakdown"
 
-
-    # ⭐ 여기에 새로운 함수 추가 (_extract_act_scenes 같은 헬퍼 메서드들 근처)
-    def calculate_screenplay_pages(self, content: str) -> float:
+    @staticmethod
+    def calculate_screenplay_pages(content: str) -> float:
         """Calculate screenplay pages more accurately"""
         if not content:
             return 0.0
+ 
             
         lines = content.split('\n')
         
@@ -2161,12 +2161,18 @@ You provide feedback that's critical yet encouraging."""
         
         return ""
     
-
-
     def _get_previous_acts(self, stages: List[Dict], current_idx: int) -> str:
-        """Get previous acts content"""
+        """Get previous acts content - Fixed for 3-stage structure"""
         previous = []
-        act_indices = {5: [], 7: [5], 9: [5, 7], 11: [5, 7, 9]}
+        
+        # 3단계 구조에 맞는 인덱스 매핑
+        # 각 막의 최종 확장 버전 인덱스만 사용
+        act_indices = {
+            5: [],      # Act 1 draft - no previous acts
+            8: [7],     # Act 2A draft - use Act 1 final
+            11: [7, 10], # Act 2B draft - use Act 1 & 2A finals
+            14: [7, 10, 13] # Act 3 draft - use Act 1, 2A & 2B finals
+        }
         
         if current_idx in act_indices:
             for idx in act_indices[current_idx]:
@@ -2175,7 +2181,6 @@ You provide feedback that's critical yet encouraging."""
         
         return "\n\n---\n\n".join(previous) if previous else ""
 
-
     def _extract_field(self, content: str, field_pattern: str) -> Optional[str]:
             """Extract field value from content with improved parsing"""
             pattern = rf'{field_pattern}[:\s]*([^\n]+?)(?=\n[A-Z가-힣]|$)'