Update app.py

app.py

@@ -1742,51 +1742,38 @@ def stitch_story(raw_story: List[dict], theme_id: str, level: str) -> Dict[str,
 def show_stitched_story(story_data: Dict[str, str]):
     """Display the stitched story with translation"""
     try:
-        st.markdown("""
-            <div style="
-                background-color: #f8f9fa;
-                padding: 20px;
-                border-radius: 10px;
-                margin: 20px 0;
-                border: 2px solid #4caf50;
-            ">
-                <h3 style="color: #2e7d32; text-align: center; margin-bottom: 20px;">
-                    📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์
-                </h3>
-                
-                <div style="
-                    background-color: white;
-                    padding: 15px;
-                    border-radius: 8px;
-                    margin-bottom: 15px;
-                ">
-                    <h4 style="color: #1565c0;">English Version</h4>
-                    <p style="color: #333; font-size: 1.1em; line-height: 1.6;">
-                        {story_data['polished_english']}
-                    </p>
-                </div>
-                
-                <div style="
-                    background-color: white;
-                    padding: 15px;
-                    border-radius: 8px;
-                ">
-                    <h4 style="color: #1565c0;">ฉบับภาษาไทย</h4>
-                    <p style="color: #333; font-size: 1.1em; line-height: 1.6;">
-                        {story_data['thai_translation']}
-                    </p>
-                </div>
+        st.markdown("# 📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์")
+        
+        # English Version
+        st.markdown("### 🇬🇧 English Version")
+        st.markdown(
+            f"""
+            <div style="background-color: #f8f9fa; padding: 20px; border-radius: 10px; 
+                        border-left: 4px solid #1565c0; margin-bottom: 20px;">
+                {story_data['polished_english']}
             </div>
-        """, unsafe_allow_html=True)
+            """,
+            unsafe_allow_html=True
+        )
         
-        # Save options
+        # Thai Version
+        st.markdown("### 🇹🇭 ฉบับภาษาไทย")
+        st.markdown(
+            f"""
+            <div style="background-color: #f8f9fa; padding: 20px; border-radius: 10px; 
+                        border-left: 4px solid #28a745; margin-bottom: 20px;">
+                {story_data['thai_translation']}
+            </div>
+            """,
+            unsafe_allow_html=True
+        )
+
+        # Save Options
         st.markdown("### 💾 บันทึกเรื่องราวฉบับสมบูรณ์")
         save_col1, save_col2 = st.columns(2)
         
         with save_col1:
-            if st.button("📥 บันทึกเป็น PDF", 
-                        key="save_stitched_pdf", 
-                        use_container_width=True):
+            if st.button("📥 บันทึกเป็น PDF", key="save_stitched_pdf", use_container_width=True):
                 pdf_data = create_bilingual_story_pdf(story_data)
                 st.download_button(
                     "ดาวน์โหลด PDF",
@@ -1797,9 +1784,7 @@ def show_stitched_story(story_data: Dict[str, str]):
                 )
         
         with save_col2:
-            if st.button("💾 บันทึกข้อความ", 
-                        key="save_stitched_text", 
-                        use_container_width=True):
+            if st.button("💾 บันทึกข้อความ", key="save_stitched_text", use_container_width=True):
                 text_data = json.dumps(story_data, ensure_ascii=False, indent=2)
                 st.download_button(
                     "ดาวน์โหลดข้อความ",
@@ -1811,7 +1796,7 @@ def show_stitched_story(story_data: Dict[str, str]):
                 
     except Exception as e:
         logging.error(f"Error showing stitched story: {str(e)}")
-        st.error("เกิดข้อผิดพลาดในการแสดงเรื่องราวฉบับสมบูรณ์")
+        st.error("เกิดข้อผิดพลาดในการแสดงเรื่องราว กรุณาลองใหม่อีกครั้ง")
 
 def create_bilingual_story_pdf(story_data: Dict[str, str]) -> bytes:
     """Create a PDF with both English and Thai versions of the story"""
@@ -3097,85 +3082,126 @@ def show_story_stitching_options():
 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"""
     try:
-        logging.info("Starting story generation...")
-        logging.info(f"Input params - Style: {style}, Detail: {detail_level}, Theme: {theme}, Level: {level}")
-        
-        # แปลง detail level เป็นภาษาอังกฤษ
-        detail_map = {
-            "น้อย": "minimal",
-            "ปานกลาง": "moderate",
-            "มาก": "detailed"
+        # 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
+            """
         }
-        detail_level_en = detail_map.get(detail_level, "moderate")
-        
-        # ดึงชื่อสไตล์
-        style_name = style.split(" - ")[0]
-        
-        # สร้าง prompt
-        story_summary = generate_story_summary(story)
-        logging.info(f"Story summary length: {len(story_summary)}")
-        
+
+        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]
+
         # Generate English version
-        try:
-            logging.info("Generating English version...")
-            response = client.chat.completions.create(
-                model="gpt-4",
-                messages=[
-                    {
-                        "role": "system",
-                        "content": f"You are a children's story editor. Create a {style_name} style story with {detail_level_en} detail level."
-                    },
-                    {
-                        "role": "user",
-                        "content": f"Original story:\n{story_summary}\n\nCreate a polished version:"
-                    }
-                ],
-                max_tokens=1000,
-                temperature=0.7
-            )
-            polished_english = response.choices[0].message.content.strip()
-            logging.info("English version generated successfully")
-        except Exception as e:
-            logging.error(f"Error generating English version: {str(e)}")
-            raise
+        response = client.chat.completions.create(
+            model="gpt-4",
+            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.
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Original story:\n{story_content}\n\nCreate a polished version:"
+                }
+            ],
+            max_tokens=level_settings['max_tokens'],
+            temperature=0.7
+        )
+        
+        polished_english = response.choices[0].message.content.strip()
 
         # Generate Thai translation
-        try:
-            logging.info("Generating Thai translation...")
-            translation_response = client.chat.completions.create(
-                model="gpt-4o-mini",
-                messages=[
-                    {
-                        "role": "system",
-                        "content": "You are a Thai translator. Create a natural Thai translation."
-                    },
-                    {
-                        "role": "user",
-                        "content": f"Translate this story to Thai:\n{polished_english}"
-                    }
-                ],
-                max_tokens=1000,
-                temperature=0.7
-            )
-            thai_translation = translation_response.choices[0].message.content.strip()
-            logging.info("Thai translation generated successfully")
-        except Exception as e:
-            logging.error(f"Error generating Thai translation: {str(e)}")
-            raise
+        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.
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Translate this story to Thai:\n{polished_english}"
+                }
+            ],
+            max_tokens=1000,
+            temperature=0.7
+        )
+        
+        thai_translation = translation_response.choices[0].message.content.strip()
 
-        result = {
-            'original': story_summary,
+        return {
             'polished_english': polished_english,
             'thai_translation': thai_translation,
             'style': style_name,
-            'detail_level': detail_level_en
+            'level': level
         }
-        
-        logging.info("Story generation completed successfully")
-        return result
-        
+
     except Exception as e:
-        logging.error(f"Error in story generation: {str(e)}")
+        logging.error(f"Error generating stitched story: {str(e)}")
         raise
 
 def show_stitched_result(story_data: Dict[str, str]):