Update appback.py3

appback.py3

@@ -1602,6 +1602,382 @@ def show_story_progress():
             </div>
         """, unsafe_allow_html=True)
 
+def generate_story_summary(story_entries: List[dict]) -> str:
+    """Generate a concise summary of the story for the stitching process"""
+    try:
+        # Extract main story content
+        story_text = []
+        for entry in story_entries:
+            if entry.get('content'):
+                text = entry['content'].strip()
+                if text.endswith('.'):
+                    text = text[:-1]  # Remove trailing period for better concatenation
+                story_text.append(text)
+        
+        story_summary = '. '.join(story_text) + '.'
+        return story_summary
+        
+    except Exception as e:
+        logging.error(f"Error generating story summary: {str(e)}")
+        return "Error creating story summary."
+
+def stitch_story(raw_story: List[dict], theme_id: str, level: str) -> Dict[str, str]:
+    """Create a polished, coherent version of the story with Thai translation"""
+    try:
+        # Get theme details for context
+        theme = story_themes.get(theme_id, {})
+        theme_context = f"Theme: {theme.get('name_en', 'General')} - {theme.get('description_en', '')}"
+        
+        # Get story summary
+        story_summary = generate_story_summary(raw_story)
+        
+        # Create level-appropriate instructions
+        level_context = {
+            'Beginner': {
+                'instructions': """
+                    Create a simple, clear narrative using:
+                    - Basic vocabulary
+                    - Short, straightforward sentences
+                    - Present tense
+                    - Clear transitions
+                    Maintain the original story's key points but make it flow smoothly.
+                """,
+                'max_tokens': 500,
+                'temperature': 0.7
+            },
+            'Intermediate': {
+                'instructions': """
+                    Create an engaging narrative using:
+                    - Grade-appropriate vocabulary
+                    - Mix of simple and compound sentences
+                    - Present and past tense
+                    - Natural transitions
+                    Preserve the original story's elements while enhancing the flow.
+                """,
+                'max_tokens': 700,
+                'temperature': 0.7
+            },
+            'Advanced': {
+                'instructions': """
+                    Create a sophisticated narrative using:
+                    - Rich vocabulary
+                    - Varied sentence structures
+                    - Multiple tenses
+                    - Elegant transitions
+                    Maintain the story's essence while adding literary flourish.
+                """,
+                'max_tokens': 1000,
+                'temperature': 0.8
+            }
+        }
+
+        level_settings = level_context[level]
+        
+        # Generate polished English version
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    You are a professional children's story editor.
+                    {level_settings['instructions']}
+                    Context: {theme_context}
+                    
+                    Task: Create a polished, coherent version of this story while:
+                    1. Maintaining the original plot points
+                    2. Improving flow and transitions
+                    3. Adding appropriate descriptive elements
+                    4. Making the narrative more engaging
+                    5. Keeping the language level appropriate for {level} students
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Original story:\n{story_summary}\n\nCreate a polished version:"
+                }
+            ],
+            max_tokens=level_settings['max_tokens'],
+            temperature=level_settings['temperature']
+        )
+        
+        polished_english = response.choices[0].message.content.strip()
+        
+        # Generate Thai translation
+        translation_response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": """
+                    You are a professional Thai-English translator specializing in children's literature.
+                    Create a natural, flowing Thai translation that:
+                    1. Captures the story's meaning and emotion
+                    2. Uses appropriate Thai language for the target age group
+                    3. Maintains cultural relevance
+                    4. Reads 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()
+        
+        return {
+            'original': story_summary,
+            'polished_english': polished_english,
+            'thai_translation': thai_translation
+        }
+        
+    except Exception as e:
+        logging.error(f"Error in story stitching: {str(e)}")
+        raise
+
+def show_stitched_story(story_data: Dict[str, str]):
+    """Display the stitched story with translation"""
+    try:
+        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
+        )
+        
+        # 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):
+                pdf_data = create_bilingual_story_pdf(story_data)
+                st.download_button(
+                    "ดาวน์โหลด PDF",
+                    data=pdf_data,
+                    file_name=f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf",
+                    mime="application/pdf",
+                    key="download_stitched_pdf"
+                )
+        
+        with save_col2:
+            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(
+                    "ดาวน์โหลดข้อความ",
+                    data=text_data,
+                    file_name=f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.json",
+                    mime="application/json",
+                    key="download_stitched_text"
+                )
+                
+    except Exception as e:
+        logging.error(f"Error showing stitched story: {str(e)}")
+        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"""
+    try:
+        buffer = io.BytesIO()
+        doc = SimpleDocTemplate(
+            buffer,
+            pagesize=A4,
+            rightMargin=72,
+            leftMargin=72,
+            topMargin=72,
+            bottomMargin=72
+        )
+        
+        # Create styles
+        styles = getSampleStyleSheet()
+        title_style = ParagraphStyle(
+            'CustomTitle',
+            parent=styles['Title'],
+            fontSize=24,
+            spaceAfter=30,
+            alignment=1  # Center alignment
+        )
+        heading_style = ParagraphStyle(
+            'CustomHeading',
+            parent=styles['Heading1'],
+            fontSize=18,
+            spaceAfter=12,
+            textColor=colors.blue
+        )
+        body_style = ParagraphStyle(
+            'CustomBody',
+            parent=styles['Normal'],
+            fontSize=12,
+            spaceBefore=6,
+            spaceAfter=6,
+            leading=16
+        )
+        
+        # Create story elements
+        elements = []
+        
+        # Add title
+        elements.append(Paragraph("My Story - JoyStory", title_style))
+        elements.append(Spacer(1, 20))
+        
+        # Add English version
+        elements.append(Paragraph("English Version", heading_style))
+        elements.append(Paragraph(story_data['polished_english'], body_style))
+        elements.append(Spacer(1, 20))
+        
+        # Add Thai version
+        elements.append(Paragraph("ฉบับภาษาไทย", heading_style))
+        elements.append(Paragraph(story_data['thai_translation'], body_style))
+        
+        # Build PDF
+        doc.build(elements)
+        pdf = buffer.getvalue()
+        buffer.close()
+        
+        return pdf
+        
+    except Exception as e:
+        logging.error(f"Error creating bilingual PDF: {str(e)}")
+        raise
+
+def show_story_stats():
+    """Display comprehensive story statistics"""
+    try:
+        st.markdown("""
+            <div style="
+                background-color: #f3e5f5;
+                padding: 20px;
+                border-radius: 10px;
+                margin: 20px 0;
+            ">
+                <h3 style="color: #6a1b9a; margin-bottom: 15px; text-align: center;">
+                    📊 สถิติการเขียนเรื่องราว
+                </h3>
+            """, unsafe_allow_html=True)
+
+        # Basic statistics in 2 columns
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            st.metric(
+                "จำนวนประโยคทั้งหมด",
+                len(st.session_state.story),
+                help="จำนวนประโยคในเรื่องทั้งหมด"
+            )
+            st.metric(
+                "คำศัพท์ที่ใช้",
+                len(st.session_state.stats['vocabulary_used']),
+                help="จำนวนคำศัพท์ที่ไม่ซ้ำกัน"
+            )
+            st.metric(
+                "ประโยคที่ถูกต้องแล้ว",
+                st.session_state.stats['correct_first_try'],
+                help="จำนวนประโยคที่เขียนถูกต้องตั้งแต่ครั้งแรก"
+            )
+        
+        with col2:
+            st.metric(
+                "คะแนนรวม",
+                st.session_state.points['total'],
+                help="คะแนนรวมที่ได้รับ"
+            )
+            st.metric(
+                "ความแม่นยำ",
+                f"{st.session_state.stats['accuracy_rate']:.1f}%",
+                help="อัตราการเขียนถูกต้อง"
+            )
+            st.metric(
+                "Streak สูงสุด",
+                st.session_state.points['max_streak'],
+                help="จำนวนประโยคถูกต้องติดต่อกันมากที่สุด"
+            )
+
+        # Show achievements
+        if st.session_state.achievements:
+            st.markdown("""
+                <h4 style="color: #6a1b9a; margin: 20px 0 10px 0;">
+                    🏆 ความสำเร็จที่ได้รับ
+                </h4>
+            """, unsafe_allow_html=True)
+            
+            for achievement in st.session_state.achievements:
+                st.success(achievement)
+        
+        # Show writing progress details
+        st.markdown("""
+            <h4 style="color: #6a1b9a; margin: 20px 0 10px 0;">
+                📝 รายละเอียดการเขียน
+            </h4>
+        """, unsafe_allow_html=True)
+        
+        details_col1, details_col2 = st.columns(2)
+        
+        with details_col1:
+            # Story composition metrics
+            st.markdown("##### 📖 องค์ประกอบเรื่อง")
+            avg_sentence_length = st.session_state.stats.get('average_sentence_length', 0)
+            st.markdown(f"""
+                - ความยาวประโยคเฉลี่ย: {avg_sentence_length:.1f} คำ
+                - จำนวนคำทั้งหมด: {st.session_state.stats.get('total_words', 0)} คำ
+                - การแก้ไข: {st.session_state.stats.get('corrections_made', 0)} ครั้ง
+            """)
+        
+        with details_col2:
+            # Time and progress metrics
+            st.markdown("##### ⏱️ เวลาและความก้าวหน้า")
+            session_duration = st.session_state.stats.get('session_duration', 0)
+            duration_minutes = session_duration / 60
+            st.markdown(f"""
+                - เวลาที่ใช้: {duration_minutes:.1f} นาที
+                - ความก้าวหน้า: {min(len(st.session_state.story) * 5, 100)}%
+                - สถานะ: {"จบเรื่องแล้ว" if st.session_state.story_completed else "กำลังเขียน"}
+            """)
+        
+        # Show vocabulary usage
+        if st.session_state.stats['vocabulary_used']:
+            with st.expander("📚 คำศัพท์ที่ใช้"):
+                vocab_list = sorted(list(st.session_state.stats['vocabulary_used']))
+                st.markdown(f"""
+                    <div style="
+                        background-color: white;
+                        padding: 10px;
+                        border-radius: 5px;
+                        max-height: 200px;
+                        overflow-y: auto;
+                    ">
+                        {', '.join(vocab_list)}
+                    </div>
+                """, unsafe_allow_html=True)
+        
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+    except Exception as e:
+        logging.error(f"Error showing story stats: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงสถิติ")
+
 def show_story_ending_options():
     """Display story ending options and guidance"""
     if len(st.session_state.story) >= 10:  # Show options when story is long enough
@@ -2275,8 +2651,20 @@ def show_sidebar():
         else:
             st.sidebar.info("ยังไม่มีเรื่องราวที่จะรีเซ็ต")
 
+def clear_input_state():
+    """Clear all input-related session states"""
+    if 'text_input' in st.session_state:
+        st.session_state.text_input = ""
+    if 'story_input_area' in st.session_state:
+        st.session_state.story_input_area = ""
+    if 'last_submission' in st.session_state:
+        del st.session_state.last_submission
+    st.session_state.clear_input = True
+    # Increment clear count to force new input field
+    st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+
 def show_story_input():
-    """Display story input section"""
+    """Display story input section with improved clear functionality"""
     st.markdown("""
         <div class="thai-eng">
             <div class="thai">✏️ ถึงตาคุณแล้ว</div>
@@ -2284,59 +2672,83 @@ def show_story_input():
         </div>
     """, unsafe_allow_html=True)
     
-    # Initialize clear_input flag if not exists
+    # Initialize states if not exists
     if 'clear_input' not in st.session_state:
         st.session_state.clear_input = False
-    
-    # Default value for text input
-    default_value = "" if st.session_state.clear_input else st.session_state.get('text_input', "")
-    
-    # If clear_input flag is True, reset it
-    if st.session_state.clear_input:
-        st.session_state.clear_input = False
+    if 'text_input' not in st.session_state:
+        st.session_state.text_input = ""
+    if 'last_submission' not in st.session_state:
+        st.session_state.last_submission = None
 
     # Show remaining sentences if in ending mode
     if st.session_state.get('ending_mode'):
         remaining = st.session_state.sentences_to_end
         st.info(f"🎭 โหมดจบเรื่อง - เหลืออีก {remaining} ประโยค")
     
-    # Input area
+    # Input area with key based on clear state
     text_input = st.text_area(
         "เขียนต่อจากเรื่องราว | Continue the story:",
-        value=st.session_state.get('text_input', ""),
+        value=st.session_state.text_input,
         height=100,
-        key="story_input_area",
+        key=f"story_input_area_{st.session_state.get('clear_count', 0)}",
         help="พิมพ์ประโยคภาษาอังกฤษเพื่อต่อเรื่อง",
         label_visibility="collapsed"
     )
     
-    # Save current input to session state
+    # Keep track of current input
     st.session_state.text_input = text_input
     
-    # Submit button with character count
-    col1, col2 = st.columns([3, 1])
+    # Submit and Clear buttons
+    col1, col2, col3 = st.columns([3, 1, 1])
+    
     with col1:
-        if st.button("📝 ส่งคำตอบ | Submit", use_container_width=True):
-            if not text_input.strip():
+        submit_button = st.button(
+            "📝 ส่งคำตอบ | Submit",
+            use_container_width=True,
+            disabled=not text_input.strip()  # Disable if empty
+        )
+        
+        if submit_button:
+            if text_input.strip():
+                try:
+                    # Save current submission
+                    st.session_state.last_submission = text_input.strip()
+                    with st.spinner("กำลังวิเคราะห์ประโยค..."):
+                        handle_story_submission(text_input.strip())
+                except Exception as e:
+                    logging.error(f"Error submitting story: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
+            else:
                 st.warning("กรุณาเขียนข้อความก่อนส่ง")
-                return
-            try:
-                with st.spinner("กำลังวิเคราะห์ประโยค..."):
-                    handle_story_submission(text_input.strip())
-            except Exception as e:
-                logging.error(f"Error submitting story: {str(e)}")
-                st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้���")
     
     with col2:
+        # Clear button with unique key
+        if st.button(
+            "🗑️ ล้างข้อความ | Clear",
+            key=f"clear_button_{st.session_state.get('clear_count', 0)}",
+            use_container_width=True,
+            disabled=not text_input.strip()  # Disable if already empty
+        ):
+            # Increment clear count to force new key generation
+            st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+            clear_input_state()
+            st.rerun()
+    
+    with col3:
+        # Character count
         char_count = len(text_input)
         st.markdown(f"""
-            <div style="text-align: right; color: {'red' if char_count > 200 else '#666'};">
+            <div style="text-align: right; color: {'red' if char_count > 200 else '#666'}; padding: 8px;">
                 {char_count}/200 ตัวอักษร
             </div>
         """, unsafe_allow_html=True)
 
+    # Warning for long text
+    if char_count > 200:
+        st.warning("⚠️ ข้อความยาวเกิน 200 ตัวอักษร")
+
 def handle_story_submission(text: str):
-    """Handle story submission and processing"""
+    """Handle story submission with improved state management"""
     if not st.session_state.story:
         st.error("กรุณาเลือกธีมเรื่องราวก่อนเริ่มเขียน")
         return
@@ -2344,6 +2756,8 @@ def handle_story_submission(text: str):
     # Check if in ending mode
     if st.session_state.get('ending_mode'):
         handle_ending_mode(text)
+        # Clear input after handling ending mode
+        clear_input_state()
         return
 
     # Regular processing
@@ -2410,14 +2824,23 @@ def handle_story_submission(text: str):
         # Update session stats
         update_session_stats()
         
-        # Set flag to clear input on next rerun
-        st.session_state.clear_input = True
+        # Clear input state completely before rerun
+        clear_input_state()
+        
+        # Make sure we don't have any lingering text
+        if 'story_input_area' in st.session_state:
+            st.session_state.story_input_area = ""
+        
+        # Increment clear count to force new input field
+        st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
         
         # Rerun to update UI
         st.rerun()
         
     except Exception as e:
         logging.error(f"Error in story submission: {str(e)}")
+        # Clear input even on error to prevent stuck state
+        clear_input_state()
         raise
 
 
@@ -2546,8 +2969,20 @@ def show_completion_options():
     try:
         st.balloons()
         
-        # Show completion message
-        st.markdown("""
+        # Show completion message with proper ending type display
+        ending_type_display = {
+            "Happy Ending": "จบแบบมีความสุข",
+            "Mysterious Ending": "จบแบบทิ้งท้ายให้คิดต่อ",
+            "Lesson Learned": "จบแบบได้ข้อคิด",
+            "Surprise Ending": "จบแบบพลิกความคาดหมาย"
+        }
+        
+        current_ending = ending_type_display.get(
+            st.session_state.ending_type,
+            st.session_state.ending_type
+        )
+        
+        st.markdown(f"""
             <div style="
                 background-color: #e8f5e9;
                 padding: 20px;
@@ -2560,50 +2995,58 @@ def show_completion_options():
                     🎉 ยินดีด้วย! คุณเขียนเรื่องราวจบสมบูรณ์แล้ว
                 </h2>
                 <p style="color: #1b5e20;">
-                    เรื่องราวของคุณจบลงด้วย: {st.session_state.ending_type}
+                    เรื่องราวของคุณจบลงด้วย: {current_ending}
                 </p>
             </div>
         """, unsafe_allow_html=True)
         
         # Show statistics
-        metrics_col1, metrics_col2 = st.columns(2)
-        with metrics_col1:
-            st.metric(
-                "จำนวนประโยคทั้งหมด",
-                len(st.session_state.story),
-                help="จำนวนประโยคในเรื่องทั้งหมด"
-            )
-            st.metric(
-                "คำศัพท์ที่ใช้",
-                len(st.session_state.stats['vocabulary_used']),
-                help="จำนวนคำศัพท์ที่ไม่ซ้ำกัน"
-            )
+        show_story_stats()
         
-        with metrics_col2:
-            st.metric(
-                "คะแนนรวม",
-                st.session_state.points['total'],
-                help="คะแนนรวมที่ได้รับ"
-            )
-            st.metric(
-                "ความแม่นยำ",
-                f"{st.session_state.stats['accuracy_rate']:.1f}%",
-                help="อัตราการเขียนถูกต้อง"
-            )
+        # Offer story stitching option
+        st.markdown("### ✨ ต้องการให้ AI ช่วยเรียบเรียงเรื่องราวให้สมบูรณ์ขึ้นไหม?")
         
-        # Save options
-        st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
-        save_completed_story()
-        
-        # New story option
-        if st.button("🔄 เริ่มเรื่องใหม่", 
-                    key="new_story_button", 
+        if st.button("🎨 เรียบเรียงเรื่องราว", 
+                    key="stitch_story_button", 
                     use_container_width=True):
-            if st.checkbox("✅ ยืนยันการเริ่มใหม่", key="confirm_reset"):
-                st.warning("การเริ่มใหม่จะลบเรื่องราวปัจจุบันทั้งหมด")
-                if st.button("🔄 ยืนยันการเริ่มใหม่", 
-                            key="final_confirm_reset", 
-                            use_container_width=True):
+            with st.spinner("กำลังเรียบเรียงเรื่องราว..."):
+                try:
+                    stitched_story = generate_stitched_story(
+                        story=st.session_state.story,
+                        style="Classic Fairytale",  # Default style
+                        detail_level="ปานกลาง",    # Default detail level
+                        theme=st.session_state.current_theme,
+                        level=st.session_state.level
+                    )
+                    st.session_state.stitched_story = stitched_story
+                    show_stitched_story(stitched_story)
+                except Exception as e:
+                    logging.error(f"Error in story stitching: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการเรียบเรียงเรื่องราว กรุณาลองใหม่อีกครั้ง")
+        
+        # Original save options
+        st.markdown("### 💾 บันทึกเรื่องราวต้นฉบับ")
+        save_col1, save_col2 = st.columns(2)
+        
+        with save_col1:
+            if st.button("📑 บันทึกเป็น PDF", 
+                        key="save_original_pdf", 
+                        use_container_width=True):
+                pdf_data = create_story_pdf()
+                st.download_button(
+                    "ดาวน์โหลด PDF",
+                    data=pdf_data,
+                    file_name=f"original_story_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf",
+                    mime="application/pdf",
+                    key="download_original_pdf"
+                )
+        
+        with save_col2:
+            if st.button("🔄 เริ่มเรื่องใหม่", 
+                        key="new_story_button", 
+                        use_container_width=True):
+                if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
+                             key="confirm_new_story"):
                     reset_story()
                     st.rerun()
 
@@ -2611,6 +3054,211 @@ def show_completion_options():
         logging.error(f"Error showing completion options: {str(e)}")
         st.error("เกิดข้อผิดพลาดในการแสดงตัวเลือกหลังจบเรื่อง")
 
+def show_story_stitching_options():
+    """Display story stitching options and handle the flow"""
+    try:
+        st.markdown("""
+            <div style="
+                background-color: #e3f2fd;
+                padding: 20px;
+                border-radius: 10px;
+                margin: 20px 0;
+            ">
+                <h3 style="color: #1565c0; margin-bottom: 15px;">
+                    ✨ เรียบเรียงเรื่องราวให้สมบูรณ์
+                </h3>
+                <p style="color: #333;">
+                    AI จะช่วย:
+                    - ปรับแต่งการเชื่อมประโยคให้ลื่นไหล
+                    - เพิ่มรายละเอียดที่น่าสนใจ
+                    - แปลเป็นภาษาไทย
+                    - รักษาเนื้อเรื่องและความคิดสร้างสรรค์ของคุณไว้
+                </p>
+            </div>
+        """, unsafe_allow_html=True)
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            style_option = st.selectbox(
+                "เลือกรูปแบบการเล่าเรื่อง:",
+                options=[
+                    "Classic Fairytale - นิทานแบบคลาสสิก",
+                    "Modern Adventure - การผจญภัยสมัยใหม่",
+                    "Poetic Style - แบบกวีนิพนธ์",
+                    "Simple and Clear - เรียบง่ายและชัดเจน"
+                ],
+                key="story_style_selector"
+            )
+        
+        with col2:
+            detail_level = st.select_slider(
+                "ระดับรายละเอียด:",
+                options=["น้อย", "ปานกลาง", "มาก"],
+                value="ปานกลาง",
+                key="detail_level_selector"
+            )
+        
+        if st.button("🎨 เริ่มเรียบเรียงเรื่องราว", 
+                    key="start_stitching", 
+                    use_container_width=True):
+            with st.spinner("กำลังเรียบเรียงเรื่องราว..."):
+                try:
+                    stitched_story = generate_stitched_story(
+                        story=st.session_state.story,
+                        style=style_option,
+                        detail_level=detail_level,
+                        theme=st.session_state.current_theme,
+                        level=st.session_state.level
+                    )
+                    
+                    # Save stitched story to session state
+                    st.session_state.stitched_story = stitched_story
+                    
+                    # Show the result
+                    show_stitched_result(stitched_story)
+                    
+                except Exception as e:
+                    logging.error(f"Error in story stitching: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการเรียบเรียงเรื่องราว กรุณาลองใหม่อีกครั้ง")
+    
+    except Exception as e:
+        logging.error(f"Error showing stitching options: {str(e)}")
+        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 with minimal enhancements"""
+    try:
+        # แยกประโยคต่างๆ ตาม 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')), '')
+        }
+
+        # สร้าง prompt ที่เน้นการรักษาประโยคเดิมและเชื่อมต่อแบบกระชับ
+        prompt_template = f"""
+        Rewrite this story into a flowing narrative.
+
+        CRITICAL RULES:
+        1. Keep original sentences almost exactly as they are
+        2. Add only minimal connecting words or phrases
+        3. Do not add new scenes or events
+        4. Do not add lengthy descriptions
+        5. Focus on making the story flow naturally
+        6. Keep any additions very brief and simple
+
+        Examples of good enhancement:
+        Original: "The project started. We made robots."
+        Good: "The project started with great excitement. We made robots, our first real creation."
+        BAD: "In the sunny classroom, with students gathered around shiny tables, the project started. We made incredible robots with flashing lights and complex circuits."
+
+        Original Story Parts:
+        - Beginning: {story_parts['starter']}
+        - Main Story: {' '.join(sum(zip(story_parts['user_sentences'], story_parts['ai_responses']), ()))}
+        - Ending: {story_parts['ending']}
+        
+        Create a coherent story that stays very close to the original while making it flow smoothly.
+        Remember: Less is more - add only what's necessary for flow.
+        """
+
+        # Generate English version with minimal enhancements
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": """You are a story editor who specializes in minimal enhancement.
+                    Your goal is to make the story flow while keeping it as close as possible to the original.
+                    Remember: Only add what's absolutely necessary for coherence."""
+                },
+                {
+                    "role": "user",
+                    "content": prompt_template
+                }
+            ],
+            temperature=0.5  # Lower temperature for more consistent, conservative output
+        )
+        
+        polished_english = response.choices[0].message.content.strip()
+
+        # Generate Thai translation with same minimal approach
+        translation_prompt = f"""
+        Translate this story to Thai. 
+
+        IMPORTANT:
+        1. Keep the translation concise and close to the English version
+        2. Do not add extra details or explanations
+        3. Maintain the same level of simplicity
+        4. Focus on natural Thai flow while keeping original content
+        
+        Story to translate:
+        {polished_english}
+        """
+
+        translation_response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": """You are a Thai translator who specializes in concise, accurate translations.
+                    Keep the same level of detail as the English version without adding extra elements."""
+                },
+                {
+                    "role": "user",
+                    "content": translation_prompt
+                }
+            ],
+            temperature=0.5
+        )
+        
+        thai_translation = translation_response.choices[0].message.content.strip()
+
+        return {
+            'polished_english': polished_english,
+            'thai_translation': thai_translation,
+            'style': style,
+            'level': level
+        }
+
+    except Exception as e:
+        logging.error(f"Error generating stitched story: {str(e)}")
+        raise
+
+def show_stitched_result(story_data: Dict[str, str]):
+    """Display the stitched story result with enhanced formatting"""
+    try:
+        # Check if story data exists
+        if not story_data or not isinstance(story_data, dict):
+            st.error("ไม่พบข้อมูลเรื่องราว กรุณาลองใหม่อีกครั้ง")
+            return
+
+        # Header
+        st.markdown("### 📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์")
+        
+        # English Version
+        st.subheader("🇬🇧 English Version")
+        st.markdown(
+            f'<div style="background-color: white; padding: 15px; border-radius: 8px; margin-bottom: 15px;">'
+            f'{story_data.get("polished_english", "No content available")}'
+            f'</div>',
+            unsafe_allow_html=True
+        )
+        
+        # Thai Version
+        st.subheader("🇹🇭 ฉบับภาษาไทย")
+        st.markdown(
+            f'<div style="background-color: white; padding: 15px; border-radius: 8px;">'
+            f'{story_data.get("thai_translation", "ไม่พบเนื้อหา")}'
+            f'</div>',
+            unsafe_allow_html=True
+        )
+
+    except Exception as e:
+        logging.error(f"Error showing stitched result: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงผลเรื่องราว กรุณาลองใหม่อีกครั้ง")
+
 def show_save_dialog():
     """Display save options dialog"""
     try: