Update app.py

app.py

@@ -1602,6 +1602,397 @@ 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-4",
+            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-4",
+            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("""
+            <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>
+            </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
@@ -2566,50 +2957,365 @@ def show_completion_options():
         """, 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="คะแนนรวมที่ได้รับ"
+        # Offer story stitching option
+        st.markdown("### ✨ ต้องการให้ AI ช่วยเรียบเรียงเรื่องราวให้สมบูรณ์ขึ้นไหม?")
+        
+        if st.button("🎨 เรียบเรียงเรื่องราว", 
+                    key="stitch_story_button", 
+                    use_container_width=True):
+            with st.spinner("กำลังเรียบเรียงเรื่องราว..."):
+                try:
+                    stitched_story = stitch_story(
+                        st.session_state.story,
+                        st.session_state.current_theme,
+                        st.session_state.level
+                    )
+                    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()
+
+    except Exception as e:
+        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"
             )
-            st.metric(
-                "ความแม่นยำ",
-                f"{st.session_state.stats['accuracy_rate']:.1f}%",
-                help="อัตราการเขียนถูกต้อง"
+        
+        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 based on selected style and detail level"""
+    try:
+        # Convert detail level to English
+        detail_map = {
+            "น้อย": "minimal",
+            "ปานกลาง": "moderate",
+            "มาก": "detailed"
+        }
+        detail_level_en = detail_map.get(detail_level, "moderate")
+        
+        # Extract style name
+        style_name = style.split(" - ")[0]
+        
+        # Create style-specific instructions
+        style_instructions = {
+            "Classic Fairytale": """
+                Create a classic fairytale style with:
+                - Traditional opening and closing phrases
+                - Magical and whimsical elements
+                - Clear moral or lesson
+                - Flowing, storybook language
+            """,
+            "Modern Adventure": """
+                Create a contemporary adventure style with:
+                - Dynamic pacing
+                - Modern settings and references
+                - Engaging action sequences
+                - Relatable characters
+            """,
+            "Poetic Style": """
+                Create a poetic narrative style with:
+                - Rich imagery and metaphors
+                - Rhythmic language
+                - Evocative descriptions
+                - Artistic expression
+            """,
+            "Simple and Clear": """
+                Create a clear, straightforward style with:
+                - Direct language
+                - Clear sequence of events
+                - Focused storytelling
+                - Easy-to-follow structure
+            """
+        }
+        
+        # Create detail level instructions
+        detail_instructions = {
+            "minimal": "Keep descriptions concise and focus on key events.",
+            "moderate": "Balance description and action with moderate detail.",
+            "detailed": "Include rich descriptions and elaborate on scenes and characters."
+        }
+        
+        # Generate polished version
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    You are a professional children's story editor and translator.
+                    
+                    Style Guidelines:
+                    {style_instructions[style_name]}
+                    
+                    Detail Level:
+                    {detail_instructions[detail_level_en]}
+                    
+                    Educational Level: {level}
+                    
+                    Create a polished version that:
+                    1. Maintains the original story's core elements
+                    2. Adapts to the requested style
+                    3. Provides appropriate detail level
+                    4. Ensures age-appropriate language
+                    5. Creates a cohesive narrative flow
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Original story:\n{generate_story_summary(story)}\n\nCreate a polished version:"
+                }
+            ],
+            max_tokens=1500,
+            temperature=0.7
+        )
+        
+        polished_english = response.choices[0].message.content.strip()
+        
+        # Generate Thai translation with matching style
+        translation_response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    You are a professional Thai translator specializing in children's literature.
+                    
+                    Create a Thai translation that:
+                    1. Matches the {style_name} style
+                    2. Maintains the same detail level ({detail_level_en})
+                    3. Uses natural, flowing Thai language
+                    4. Preserves the story's tone and feeling
+                    5. Adapts cultural elements appropriately
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Translate this story to Thai:\n{polished_english}"
+                }
+            ],
+            max_tokens=1500,
+            temperature=0.7
+        )
+        
+        thai_translation = translation_response.choices[0].message.content.strip()
+        
+        return {
+            'original': generate_story_summary(story),
+            'polished_english': polished_english,
+            'thai_translation': thai_translation,
+            'style': style_name,
+            'detail_level': detail_level_en
+        }
+        
+    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:
+        st.markdown("""
+            <div style="
+                background-color: #f8f9fa;
+                padding: 20px;
+                border-radius: 10px;
+                margin: 20px 0;
+            ">
+                <h2 style="color: #1565c0; text-align: center; margin-bottom: 20px;">
+                    ✨ เรื่องราวฉบับสมบูรณ์ | Complete Story
+                </h2>
+                
+                <div style="
+                    background-color: white;
+                    padding: 20px;
+                    border-radius: 8px;
+                    margin-bottom: 20px;
+                    border: 1px solid #e0e0e0;
+                ">
+                    <h3 style="color: #2e7d32; margin-bottom: 10px;">
+                        English Version
+                    </h3>
+                    <div style="
+                        color: #333;
+                        font-size: 1.1em;
+                        line-height: 1.6;
+                        white-space: pre-line;
+                    ">
+                        {story_data['polished_english']}
+                    </div>
+                </div>
+                
+                <div style="
+                    background-color: white;
+                    padding: 20px;
+                    border-radius: 8px;
+                    border: 1px solid #e0e0e0;
+                ">
+                    <h3 style="color: #2e7d32; margin-bottom: 10px;">
+                        ฉบับภาษาไทย
+                    </h3>
+                    <div style="
+                        color: #333;
+                        font-size: 1.1em;
+                        line-height: 1.6;
+                        white-space: pre-line;
+                    ">
+                        {story_data['thai_translation']}
+                    </div>
+                </div>
+            </div>
+        """, unsafe_allow_html=True)
+        
         # Save options
-        st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
-        save_completed_story()
+        st.markdown("### 💾 บันทึกเรื่องราว")
+        
+        save_col1, save_col2 = st.columns(2)
+        
+        with save_col1:
+            if st.button("📥 บันทึกเป็น PDF", 
+                        key="save_final_pdf", 
+                        use_container_width=True):
+                try:
+                    pdf_data = create_bilingual_story_pdf(story_data)
+                    file_name = f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf"
+                    st.download_button(
+                        "ดาวน์โหลด PDF",
+                        data=pdf_data,
+                        file_name=file_name,
+                        mime="application/pdf",
+                        key="download_final_pdf"
+                    )
+                except Exception as e:
+                    logging.error(f"Error saving PDF: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการสร้างไฟล์ PDF")
+        
+        with save_col2:
+            if st.button("📝 บันทึกข้อความ", 
+                        key="save_final_text", 
+                        use_container_width=True):
+                try:
+                    text_data = json.dumps(story_data, ensure_ascii=False, indent=2)
+                    file_name = f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.json"
+                    st.download_button(
+                        "ดาวน์โหลดข้อความ",
+                        data=text_data,
+                        file_name=file_name,
+                        mime="application/json",
+                        key="download_final_text"
+                    )
+                except Exception as e:
+                    logging.error(f"Error saving text: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการบันทึกข้อความ")
         
         # New story option
         if st.button("🔄 เริ่มเรื่องใหม่", 
-                    key="new_story_button", 
+                    key="new_story_after_stitch", 
                     use_container_width=True):
-            if st.checkbox("✅ ยืนยันการเริ่มใหม่", key="confirm_reset"):
-                st.warning("การเริ่มใหม่จะลบเรื่องราวปัจจุบันทั้งหมด")
-                if st.button("🔄 ยืนยันการเริ่มใหม่", 
-                            key="final_confirm_reset", 
-                            use_container_width=True):
-                    reset_story()
-                    st.rerun()
-
+            if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
+                         key="confirm_new_after_stitch"):
+                reset_story()
+                st.rerun()
+                
     except Exception as e:
-        logging.error(f"Error showing completion options: {str(e)}")
-        st.error("เกิดข้อผิดพลาดในการแสดงตัวเลือกหลังจบเรื่อง")
+        logging.error(f"Error showing stitched result: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงผลลัพธ์")
 
 def show_save_dialog():
     """Display save options dialog"""