Update app.py

app.py

@@ -2460,6 +2460,64 @@ def show_main_interface():
         # Save options
         show_save_options()
 
+def save_completed_story():
+    """Save completed story with all available formats"""
+    try:
+        st.markdown("### 💾 บันทึกเรื่องราวที่เสร็จสมบูรณ์")
+        
+        # PDF Option
+        if st.button("📑 บันทึกเป็น PDF", 
+                    key="save_completed_pdf", 
+                    use_container_width=True):
+            try:
+                pdf = create_story_pdf()
+                st.download_button(
+                    "📥 ดาวน์โหลด PDF",
+                    data=pdf,
+                    file_name=f"completed_story_{datetime.now().strftime('%Y%m%d')}.pdf",
+                    mime="application/pdf",
+                    key="download_completed_pdf"
+                )
+                st.success("สร้าง PDF เรียบร้อยแล้ว!")
+            except Exception as e:
+                logging.error(f"Error creating PDF: {str(e)}")
+                st.error("เกิดข้อผิดพลาดในการสร้างไฟล์ PDF")
+
+        # JSON Progress Save
+        if st.button("💾 บันทึกความก้าวหน้า", 
+                    key="save_completed_progress", 
+                    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,
+                    'ending_type': st.session_state.ending_type,
+                    'is_completed': True
+                }
+                
+                st.download_button(
+                    "📥 ดาวน์โหลดไฟล์บันทึก",
+                    data=json.dumps(story_data, ensure_ascii=False, indent=2),
+                    file_name=f"completed_story_{datetime.now().strftime('%Y%m%d')}.json",
+                    mime="application/json",
+                    key="download_completed_json"
+                )
+                st.success("สร้างไฟล์บันทึกเรียบร้อยแล้ว!")
+            except Exception as e:
+                logging.error(f"Error saving progress: {str(e)}")
+                st.error("เกิดข้อผิดพลาดในการบันทึกความก้าวหน้า")
+
+    except Exception as e:
+        logging.error(f"Error saving completed story: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการบันทึกเรื่องราว")
+
 def show_completion_options():
     """Display options and summary when story is completed"""
     try:
@@ -2487,9 +2545,9 @@ def show_completion_options():
         
         # Show story statistics
         st.markdown("### 📊 สรุปเรื่องราว")
-        col1, col2 = st.columns(2)
         
-        with col1:
+        metrics_col1, metrics_col2 = st.columns(2)
+        with metrics_col1:
             st.metric(
                 "จำนวนประโยคทั้งหมด",
                 len(st.session_state.story),
@@ -2501,7 +2559,7 @@ def show_completion_options():
                 help="จำนวนคำศัพท์ที่ไม่ซ้ำกัน"
             )
         
-        with col2:
+        with metrics_col2:
             st.metric(
                 "คะแนนรวม",
                 st.session_state.points['total'],
@@ -2513,21 +2571,22 @@ def show_completion_options():
                 help="อัตราการเขียนถูกต้อง"
             )
         
-        # Show action buttons with unique keys
+        # Show action buttons (without nested columns)
         st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
-        col1, col2 = st.columns(2)
         
-        with col1:
-            if st.button("💾 บันทึกเรื่องราว", 
-                        key="save_story_completion", 
-                        use_container_width=True):
-                show_save_dialog()
+        if st.button("💾 บันทึกเรื่องราว", 
+                    key="save_final_story", 
+                    use_container_width=True):
+            save_completed_story()
         
-        with col2:
-            if st.button("🔄 เริ่มเรื่องใหม่", 
-                        key="new_story_completion", 
-                        use_container_width=True):
-                reset_story_with_confirmation()
+        if st.button("🔄 เริ่มเรื่องใหม่", 
+                    key="start_new_story", 
+                    use_container_width=True):
+            if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
+                          key="confirm_new_story"):
+                reset_story()
+                st.success("เริ่มต้นใหม่เรียบร้อยแล้ว!")
+                st.rerun()
 
     except Exception as e:
         logging.error(f"Error showing completion options: {str(e)}")