Update app.py

app.py

@@ -2370,6 +2370,137 @@ def show_main_interface():
         # Save options
         show_save_options()
 
+def show_completion_options():
+    """Display options and summary when story is completed"""
+    try:
+        # Show completion celebration
+        st.balloons()
+        
+        # Create completion message
+        st.markdown("""
+            <div style="
+                background-color: #e8f5e9;
+                padding: 20px;
+                border-radius: 10px;
+                text-align: center;
+                margin: 20px 0;
+                border: 2px solid #4caf50;
+            ">
+                <h2 style="color: #2e7d32; margin-bottom: 15px;">
+                    🎉 ยินดีด้วย! คุณเขียนเรื่องราวจบสมบูรณ์แล้ว
+                </h2>
+                <p style="color: #1b5e20;">
+                    เรื่องราวของคุณจบลงด้วย: {st.session_state.ending_type}
+                </p>
+            </div>
+        """, unsafe_allow_html=True)
+        
+        # Show story statistics
+        st.markdown("### 📊 สรุปเรื่องราว")
+        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="จำนวนคำศัพท์ที่ไม่ซ้ำกัน"
+            )
+        
+        with col2:
+            st.metric(
+                "คะแนนรวม",
+                st.session_state.points['total'],
+                help="คะแนนรวมที่ได้รับ"
+            )
+            st.metric(
+                "ความแม่นยำ",
+                f"{st.session_state.stats['accuracy_rate']:.1f}%",
+                help="อัตราการเขียนถูกต้อง"
+            )
+        
+        # Show action buttons
+        st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            if st.button("💾 บันทึกเรื่องราว", use_container_width=True):
+                show_save_dialog()
+        
+        with col2:
+            if st.button("🔄 เริ่มเรื่องใหม่", use_container_width=True):
+                reset_story_with_confirmation()
+
+    except Exception as e:
+        logging.error(f"Error showing completion options: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงตัวเลือกหลังจบเรื่อง")
+
+def show_save_dialog():
+    """Display save options dialog"""
+    try:
+        st.markdown("### 💾 บันทึกเรื่องราว")
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            # PDF Export
+            if st.button("📑 บันทึกเป็น PDF", use_container_width=True):
+                try:
+                    pdf = create_story_pdf()
+                    st.download_button(
+                        "📥 ดาวน์โหลด PDF",
+                        data=pdf,
+                        file_name=f"story_{datetime.now().strftime('%Y%m%d')}.pdf",
+                        mime="application/pdf"
+                    )
+                except Exception as e:
+                    logging.error(f"Error creating PDF: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการสร้างไฟล์ PDF")
+        
+        with col2:
+            # JSON Save
+            if st.button("💾 บันทึกความก้าวหน้า", use_container_width=True):
+                try:
+                    story_data = {
+                        'timestamp': datetime.now().isoformat(),
+                        'level': st.session_state.level,
+                        'story': st.session_state.story,
+                        'achievements': st.session_state.achievements,
+                        'stats': {
+                            key: list(value) if isinstance(value, set) else value
+                            for key, value in st.session_state.stats.items()
+                        },
+                        'points': st.session_state.points
+                    }
+                    
+                    st.download_button(
+                        "📥 ดาวน์โหลดไฟล์บันทึก",
+                        data=json.dumps(story_data, ensure_ascii=False, indent=2),
+                        file_name=f"story_progress_{datetime.now().strftime('%Y%m%d')}.json",
+                        mime="application/json"
+                    )
+                except Exception as e:
+                    logging.error(f"Error saving progress: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการบันทึกความก้าวหน้า")
+
+def reset_story_with_confirmation():
+    """Reset story with confirmation dialog"""
+    try:
+        if st.session_state.story:
+            if st.checkbox("✅ ยืนยันการเริ่มใหม่"):
+                st.warning("การเริ่มใหม่จะลบเรื่องราวปัจจุบันทั้งหมด")
+                if st.button("🔄 ยืนยันการเริ่มใหม่", use_container_width=True):
+                    reset_story()
+                    st.success("เริ่มต้นใหม่เรียบร้อยแล้ว!")
+                    st.rerun()
+    except Exception as e:
+        logging.error(f"Error in reset confirmation: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการรีเซ็ตเรื่อง")
+
 # === 6. MAIN APPLICATION LOGIC ===
 def main():
     try: