Optimize the generation logic

reelforge/prompts/title_generation.py

@@ -0,0 +1,39 @@
+"""
+Title generation prompt
+
+For generating video title from content.
+"""
+
+
+TITLE_GENERATION_PROMPT = """Please generate a short, attractive title (within 10 characters) for the following content.
+
+Content:
+{content}
+
+Requirements:
+1. Brief and concise, within 10 characters
+2. Accurately summarize the core content
+3. Attractive, suitable as a video title
+4. Output only the title text, no other content
+
+Title:"""
+
+
+def build_title_generation_prompt(content: str, max_length: int = 500) -> str:
+    """
+    Build title generation prompt
+    
+    Args:
+        content: Content to generate title from
+        max_length: Maximum content length to use (default 500 chars)
+    
+    Returns:
+        Formatted prompt
+    """
+    # Take first max_length chars to avoid overly long prompts
+    content_preview = content[:max_length]
+    
+    return TITLE_GENERATION_PROMPT.format(
+        content=content_preview
+    )
+