Update app.py

app.py

@@ -2463,68 +2463,67 @@ def show_main_interface():
 def save_completed_story():
     """Save completed story with all available formats"""
     try:
-        st.markdown("### 💾 บันทึกเรื่องราวที่เสร็จสมบูรณ์")
+        st.markdown("### 💾 บันทึกเรื่องราว", unsafe_allow_html=True)
         
-        # PDF Option
-        if st.button("📑 บันทึกเป็น PDF", 
-                    key="save_completed_pdf", 
-                    use_container_width=True):
+        save_col1, save_col2 = st.columns(2)
+        
+        with save_col1:
+            # Create and offer PDF download
             try:
-                pdf = create_story_pdf()
+                pdf_data = create_story_pdf()
                 st.download_button(
-                    "📥 ดาวน์โหลด PDF",
-                    data=pdf,
-                    file_name=f"completed_story_{datetime.now().strftime('%Y%m%d')}.pdf",
+                    label="📥 ดาวน์โหลด PDF",
+                    data=pdf_data,
+                    file_name=f"joystory_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf",
                     mime="application/pdf",
-                    key="download_completed_pdf"
+                    key="download_story_pdf",
+                    use_container_width=True
                 )
-                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):
+                logging.error(f"PDF creation failed: {str(e)}")
+                st.error("ไม่สามารถสร้างไฟล์ PDF ได้")
+        
+        with save_col2:
+            # Create and offer JSON download
             try:
                 story_data = {
                     'timestamp': datetime.now().isoformat(),
                     'level': st.session_state.level,
+                    'theme': st.session_state.current_theme,
                     '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
+                    'achievements': st.session_state.achievements,
+                    'ending_type': st.session_state.ending_type
                 }
                 
+                json_str = json.dumps(story_data, ensure_ascii=False, indent=2)
+                
                 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",
+                    label="📥 ดาวน์โหลดเรื่องราว",
+                    data=json_str,
+                    file_name=f"joystory_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
                     mime="application/json",
-                    key="download_completed_json"
+                    key="download_story_json",
+                    use_container_width=True
                 )
-                st.success("สร้างไฟล์บันทึกเรียบร้อยแล้ว!")
             except Exception as e:
-                logging.error(f"Error saving progress: {str(e)}")
-                st.error("เกิดข้อผิดพลาดในการบันทึกความก้าวหน้า")
+                logging.error(f"JSON creation failed: {str(e)}")
+                st.error("ไม่สามารถสร้างไฟล์บันทึกได้")
 
     except Exception as e:
-        logging.error(f"Error saving completed story: {str(e)}")
+        logging.error(f"Error in save_completed_story: {str(e)}")
         st.error("เกิดข้อผิดพลาดในการบันทึกเรื่องราว")
-
+        
 def show_completion_options():
     """Display options and summary when story is completed"""
     try:
-        # Show completion celebration
         st.balloons()
         
-        # Create completion message
+        # Show completion message
         st.markdown("""
             <div style="
                 background-color: #e8f5e9;
@@ -2543,9 +2542,7 @@ def show_completion_options():
             </div>
         """, unsafe_allow_html=True)
         
-        # Show story statistics
-        st.markdown("### 📊 สรุปเรื่องราว")
-        
+        # Show statistics
         metrics_col1, metrics_col2 = st.columns(2)
         with metrics_col1:
             st.metric(
@@ -2571,22 +2568,21 @@ def show_completion_options():
                 help="อัตราการเขียนถูกต้อง"
             )
         
-        # Show action buttons (without nested columns)
+        # Save options
         st.markdown("### 🎯 ต้องการทำอะไรต่อ?")
+        save_completed_story()
         
-        if st.button("💾 บันทึกเรื่องราว", 
-                    key="save_final_story", 
-                    use_container_width=True):
-            save_completed_story()
-        
+        # New story option
         if st.button("🔄 เริ่มเรื่องใหม่", 
-                    key="start_new_story", 
+                    key="new_story_button", 
                     use_container_width=True):
-            if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
-                          key="confirm_new_story"):
-                reset_story()
-                st.success("เริ่มต้นใหม่เรียบร้อยแล้ว!")
-                st.rerun()
+            if st.checkbox("✅ ยืนยันการเริ่มใหม่", key="confirm_reset"):
+                st.warning("การเริ่มใหม่จะลบเรื่องราวปัจจุบันทั้งหมด")
+                if st.button("🔄 ยืนยันการเริ่มใหม่", 
+                            key="final_confirm_reset", 
+                            use_container_width=True):
+                    reset_story()
+                    st.rerun()
 
     except Exception as e:
         logging.error(f"Error showing completion options: {str(e)}")
@@ -2667,45 +2663,100 @@ def reset_story_with_confirmation():
         st.error("เกิดข้อผิดพลาดในการรีเซ็ตเรื่อง")
 
 def create_story_pdf():
-    """Create PDF of the story"""
+    """Create PDF from story content"""
     try:
         buffer = io.BytesIO()
-        doc = SimpleDocTemplate(buffer, pagesize=A4)
-        story_elements = []
+        doc = SimpleDocTemplate(
+            buffer, 
+            pagesize=A4,
+            rightMargin=72,
+            leftMargin=72,
+            topMargin=72,
+            bottomMargin=72
+        )
         
-        # Add title
+        # Prepare styles
         styles = getSampleStyleSheet()
         title_style = ParagraphStyle(
             'CustomTitle',
             parent=styles['Title'],
             fontSize=24,
-            spaceAfter=30
+            spaceAfter=30,
+            alignment=1  # Center alignment
         )
-        story_elements.append(Paragraph("My Story", title_style))
-        
-        # Add content
-        content_style = ParagraphStyle(
+        normal_style = ParagraphStyle(
             'CustomBody',
             parent=styles['Normal'],
             fontSize=12,
-            leading=16,
-            spaceAfter=12
+            spaceBefore=6,
+            spaceAfter=6
+        )
+        
+        # Create story elements list
+        elements = []
+        
+        # Add title
+        title_text = "My Story - JoyStory"
+        elements.append(Paragraph(title_text, title_style))
+        elements.append(Spacer(1, 12))
+        
+        # Add metadata
+        metadata_style = ParagraphStyle(
+            'Metadata',
+            parent=styles['Normal'],
+            fontSize=10,
+            textColor=colors.gray,
+            alignment=1
         )
+        metadata_text = f"Level: {st.session_state.level}<br/>Date: {datetime.now().strftime('%Y-%m-%d')}"
+        elements.append(Paragraph(metadata_text, metadata_style))
+        elements.append(Spacer(1, 20))
         
+        # Add story content
         for entry in st.session_state.story:
+            # Skip system messages or empty entries
+            if not entry.get('content'):
+                continue
+                
+            # Format based on role
             if entry['role'] == 'AI':
+                text = f"🤖 AI: {entry['content']}"
                 color = colors.blue
             else:
+                text = f"👤 You: {entry['content']}"
                 color = colors.black
             
-            p = Paragraph(
-                f"<font color={color}>{entry['content']}</font>",
-                content_style
+            # Create paragraph style with color
+            style = ParagraphStyle(
+                'ColoredText',
+                parent=normal_style,
+                textColor=color
             )
-            story_elements.append(p)
+            
+            # Add paragraph
+            elements.append(Paragraph(text, style))
+            elements.append(Spacer(1, 6))
+        
+        # Add statistics
+        elements.append(Spacer(1, 20))
+        stats_style = ParagraphStyle(
+            'Stats',
+            parent=styles['Normal'],
+            fontSize=10,
+            textColor=colors.gray
+        )
+        
+        stats_text = f"""
+        Story Statistics:<br/>
+        Total Sentences: {len(st.session_state.story)}<br/>
+        Unique Words: {len(st.session_state.stats['vocabulary_used'])}<br/>
+        Accuracy Rate: {st.session_state.stats['accuracy_rate']:.1f}%<br/>
+        Total Points: {st.session_state.points['total']}
+        """
+        elements.append(Paragraph(stats_text, stats_style))
         
         # Build PDF
-        doc.build(story_elements)
+        doc.build(elements)
         pdf = buffer.getvalue()
         buffer.close()