Update app.py

app.py

@@ -3080,114 +3080,92 @@ def show_story_stitching_options():
         st.error("เกิดข้อผิดพลาดในการแสดงตัวเลือกการเรียบเรียง")
 
 def generate_stitched_story(story: List[dict], style: str, detail_level: str, theme: str, level: str) -> Dict[str, str]:
-    """Generate a polished version of the story based on selected style and detail level"""
+    """Generate a polished version of the story with balanced additional details"""
     try:
-        # Create a comprehensive prompt for the final story
-        story_content = " ".join([entry['content'] for entry in story])
-        
-        style_prompts = {
-            "Classic Fairytale": """
-                Create a classic fairytale style story that:
-                - Uses traditional storytelling elements
-                - Has a clear beginning, middle, and end
-                - Incorporates descriptive language
-                - Maintains a magical, enchanting tone
-            """,
-            "Modern Adventure": """
-                Create a modern adventure story that:
-                - Uses contemporary settings and situations
-                - Has exciting plot developments
-                - Incorporates relatable elements
-                - Maintains an engaging, active pace
-            """,
-            "Simple and Clear": """
-                Create a straightforward story that:
-                - Uses clear, simple language
-                - Has a logical flow of events
-                - Focuses on key story elements
-                - Maintains easy readability
-            """
+        # แยกประโยคต่างๆ ตาม role
+        story_parts = {
+            'starter': next((entry['content'] for entry in story if entry.get('is_starter')), ''),
+            'user_sentences': [entry['content'] for entry in story if entry['role'] == 'You'],
+            'ai_responses': [entry['content'] for entry in story if entry['role'] == 'AI' and not entry.get('is_starter')],
+            'ending': next((entry['content'] for entry in story if entry.get('is_final')), '')
         }
 
-        level_context = {
-            'Beginner': {
-                'instructions': """
-                Additional Guidelines:
-                - Use simple vocabulary
-                - Keep sentences short and clear
-                - Use basic tenses
-                - Focus on common words and phrases
-                """,
-                'max_tokens': 500
-            },
-            'Intermediate': {
-                'instructions': """
-                Additional Guidelines:
-                - Use grade-appropriate vocabulary
-                - Mix simple and compound sentences
-                - Use common tenses
-                - Add some descriptive elements
-                """,
-                'max_tokens': 700
-            },
-            'Advanced': {
-                'instructions': """
-                Additional Guidelines:
-                - Use rich vocabulary
-                - Vary sentence structures
-                - Use multiple tenses
-                - Add literary elements and descriptions
-                """,
-                'max_tokens': 1000
-            }
-        }
-
-        # Get style name without suffix
-        style_name = style.split(" - ")[0] if " - " in style else style
-        level_settings = level_context[level]
+        # สร้าง prompt ที่เน้นการเพิ่มรายละเอียดอย่างสมดุล
+        prompt_template = f"""
+        Rewrite this story into a flowing narrative. 
+
+        IMPORTANT RULES:
+        1. Keep all original sentences and events intact
+        2. Add appropriate setting details and context
+        3. Include simple descriptive elements that enhance the story
+        4. Add natural transitions between events
+        5. Keep additional details balanced and relevant
+        6. Maintain the exact sequence of events
+        
+        Enhancement Guidelines:
+        - Add setting details (where things happen)
+        - Include character reactions and emotions
+        - Add simple descriptive words for actions
+        - Create smooth transitions between scenes
+        - Keep the magical/whimsical tone of the story
+        
+        Original Story Structure:
+        - Beginning: {story_parts['starter']}
+        - Main Story: {' '.join(sum(zip(story_parts['user_sentences'], story_parts['ai_responses']), ()))}
+        - Ending: {story_parts['ending']}
+        
+        Style: {style}
+        Level: {level}
+        
+        Create an enhanced narrative that preserves all original events while adding appropriate context and descriptions.
+        """
 
-        # Generate English version
+        # Generate English version with balanced details
         response = client.chat.completions.create(
-            model="gpt-4",
+            model="gpt-4o-mini",
             messages=[
                 {
                     "role": "system",
-                    "content": f"""
-                    You are a children's story editor.
-                    {style_prompts.get(style_name, style_prompts['Simple and Clear'])}
-                    {level_settings['instructions']}
-                    Create a polished, cohesive story that preserves the original plot while improving flow and readability.
-                    """
+                    "content": """You are a story editor who enhances stories while preserving their original content.
+                    Focus on adding just enough detail to make the story more engaging without overshadowing the original events."""
                 },
                 {
                     "role": "user",
-                    "content": f"Original story:\n{story_content}\n\nCreate a polished version:"
+                    "content": prompt_template
                 }
             ],
-            max_tokens=level_settings['max_tokens'],
             temperature=0.7
         )
         
         polished_english = response.choices[0].message.content.strip()
 
-        # Generate Thai translation
+        # Generate Thai translation with similar enhancement approach
+        translation_prompt = f"""
+        Translate this enhanced story to Thai. 
+        
+        IMPORTANT:
+        1. Maintain the same level of detail as the English version
+        2. Keep all story elements and added descriptions
+        3. Translate naturally while preserving the story's tone
+        4. Ensure the magical elements come through clearly
+        
+        Story to translate:
+        {polished_english}
+        """
+
         translation_response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {
                     "role": "system",
-                    "content": """
-                    You are a Thai translator specializing in children's literature.
-                    Create a natural, engaging Thai translation that maintains the story's essence and style.
-                    Ensure the translation is appropriate for young readers and flows naturally in Thai.
-                    """
+                    "content": """You are a Thai translator who specializes in children's stories.
+                    Maintain the enhanced details while creating a natural Thai narrative."""
                 },
                 {
                     "role": "user",
-                    "content": f"Translate this story to Thai:\n{polished_english}"
+                    "content": translation_prompt
                 }
             ],
-            max_tokens=1000,
             temperature=0.7
         )
         
@@ -3196,7 +3174,7 @@ def generate_stitched_story(story: List[dict], style: str, detail_level: str, th
         return {
             'polished_english': polished_english,
             'thai_translation': thai_translation,
-            'style': style_name,
+            'style': style,
             'level': level
         }