Update app.py

app.py

@@ -1342,6 +1342,41 @@ 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:
+        """Script doctor prompt focused on expansion and enhancement"""
+        
+        # Calculate current pages
+        current_lines = len(act_content.split('\n'))
+        current_pages = current_lines / 55.0
+        target_pages = int(SCREENPLAY_LENGTHS[screenplay_type]['pages'] * 0.25)
+        expansion_needed = target_pages - current_pages
+        
+        lang_prompts = {
+            "Korean": f"""당신은 스크립트 닥터입니다...""",
+            "English": f"""You are a script doctor..."""
+        }
+        
+        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:
+        """Writer prompt for expanded version"""
+        
+        target_pages = int(SCREENPLAY_LENGTHS[screenplay_type]['pages'] * 0.25)
+        target_lines = target_pages * 55
+        
+        lang_prompts = {
+            "Korean": f"""원본 {act}를 확장 지시사항에 따라...""",
+            "English": f"""Rewrite {act} following expansion instructions..."""
+        }
+        
+        return lang_prompts.get(language, lang_prompts["English"])
+    
     def create_final_reviewer_prompt(self, complete_screenplay: str, 
                                    screenplay_type: str, genre: str, language: str) -> str:
         """Final comprehensive review"""
@@ -1556,6 +1591,54 @@ 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:
+        """Calculate screenplay pages more accurately"""
+        if not content:
+            return 0.0
+            
+        lines = content.split('\n')
+        
+        # Initialize counters
+        total_line_count = 0
+        
+        for line in lines:
+            line = line.strip()
+            
+            if not line:
+                # Empty lines count in screenplay format
+                total_line_count += 1
+                continue
+            
+            # Scene heading (INT./EXT.)
+            if line.startswith(('INT.', 'EXT.')):
+                total_line_count += 3  # Scene headings take more space
+            
+            # Character name (all caps, short)
+            elif line.isupper() and len(line.split()) <= 3 and not any(c.isdigit() for c in line):
+                total_line_count += 2  # Character names have spacing
+            
+            # Parenthetical
+            elif line.startswith('(') and line.endswith(')'):
+                total_line_count += 1
+            
+            # Dialogue (indented or after character name)
+            elif len(line) < 60 and any(prev.isupper() and len(prev.split()) <= 3 
+                                        for prev in lines[max(0, lines.index(line)-2):lines.index(line)]):
+                total_line_count += 1
+            
+            # Action lines
+            else:
+                # Long action lines wrap
+                wrapped_lines = len(line) / 60  # Approximate 60 characters per line
+                total_line_count += max(1, int(wrapped_lines))
+        
+        # Standard screenplay format: ~55 lines per page
+        pages = total_line_count / 55.0
+        
+        return pages
+    
     # --- LLM call functions ---
     def call_llm_sync(self, messages: List[Dict[str, str]], role: str, language: str) -> str:
         full_content = ""