Update app.py

app.py

@@ -2423,16 +2423,20 @@ def show_completion_options():
                 help="อัตราการเขียนถูกต้อง"
             )
         
-        # Show action buttons
+        # Show action buttons with unique keys
         st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
         col1, col2 = st.columns(2)
         
         with col1:
-            if st.button("💾 บันทึกเรื่องราว", use_container_width=True):
+            if st.button("💾 บันทึกเรื่องราว", 
+                        key="save_story_completion", 
+                        use_container_width=True):
                 show_save_dialog()
         
         with col2:
-            if st.button("🔄 เริ่มเรื่องใหม่", use_container_width=True):
+            if st.button("🔄 เริ่มเรื่องใหม่", 
+                        key="new_story_completion", 
+                        use_container_width=True):
                 reset_story_with_confirmation()
 
     except Exception as e:
@@ -2447,23 +2451,28 @@ def show_save_dialog():
         col1, col2 = st.columns(2)
         
         with col1:
-            # PDF Export
-            if st.button("📑 บันทึกเป็น PDF", use_container_width=True):
+            # PDF Export with unique key
+            if st.button("📑 บันทึกเป็น PDF", 
+                        key="save_pdf_button", 
+                        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"
+                        mime="application/pdf",
+                        key="download_pdf_button"
                     )
                 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):
+            # JSON Save with unique key
+            if st.button("💾 บันทึกความก้าวหน้า", 
+                        key="save_progress_button", 
+                        use_container_width=True):
                 try:
                     story_data = {
                         'timestamp': datetime.now().isoformat(),
@@ -2481,7 +2490,8 @@ def show_save_dialog():
                         "📥 ดาวน์โหลดไฟล์บันทึก",
                         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"
+                        mime="application/json",
+                        key="download_json_button"
                     )
                 except Exception as e:
                     logging.error(f"Error saving progress: {str(e)}")
@@ -2494,9 +2504,12 @@ def reset_story_with_confirmation():
     """Reset story with confirmation dialog"""
     try:
         if st.session_state.story:
-            if st.checkbox("✅ ยืนยันการเริ่มใหม่"):
+            if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
+                          key="reset_confirmation_checkbox"):
                 st.warning("การเริ่มใหม่จะลบเรื่องราวปัจจุบันทั้งหมด")
-                if st.button("🔄 ยืนยันการเริ่มใหม่", use_container_width=True):
+                if st.button("🔄 ยืนยันการเริ่มใหม่", 
+                            key="confirm_reset_button", 
+                            use_container_width=True):
                     reset_story()
                     st.success("เริ่มต้นใหม่เรียบร้อยแล้ว!")
                     st.rerun()