Update app.py

app.py

@@ -3,6 +3,16 @@ import json
 import datetime
 from openai import OpenAI
 from typing import Dict, List, Set
+import io
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib.units import inch
+from reportlab.pdfgen import canvas
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.ttfonts import TTFont
+from datetime import datetime
 
 # ระบบ Achievements
 achievements_list = {
@@ -569,6 +579,119 @@ def update_achievements(text: str):
             st.session_state.badges.append(badge)
             st.success(f"🏆 Achievement Unlocked: {badge}!")
 
+def create_story_pdf():
+    """สร้าง PDF จากเรื่องราวที่เขียน"""
+    buffer = io.BytesIO()
+    doc = SimpleDocTemplate(
+        buffer,
+        pagesize=A4,
+        rightMargin=72,
+        leftMargin=72,
+        topMargin=72,
+        bottomMargin=72
+    )
+
+    # สร้าง styles สำหรับ PDF
+    styles = getSampleStyleSheet()
+    title_style = ParagraphStyle(
+        'CustomTitle',
+        parent=styles['Heading1'],
+        fontSize=24,
+        spaceAfter=30,
+        alignment=1  # center
+    )
+    
+    story_style = ParagraphStyle(
+        'StoryText',
+        parent=styles['Normal'],
+        fontSize=12,
+        spaceBefore=12,
+        spaceAfter=12,
+        leading=16
+    )
+    
+    stats_style = ParagraphStyle(
+        'Stats',
+        parent=styles['Normal'],
+        fontSize=10,
+        textColor=colors.gray,
+        alignment=1
+    )
+
+    # สร้างเนื้อหา PDF
+    elements = []
+    
+    # หน้าปก
+    elements.append(Paragraph("My English Story Adventure", title_style))
+    elements.append(Paragraph(
+        f"Created by: {st.session_state.get('player_name', 'Young Writer')}<br/>"
+        f"Date: {datetime.now().strftime('%B %d, %Y')}<br/>"
+        f"Level: {st.session_state.level}",
+        stats_style
+    ))
+    elements.append(Spacer(1, 30))
+
+    # เพิ่มสถิติการเขียน
+    stats_text = (
+        f"Total Points: {st.session_state.points['total']}<br/>"
+        f"Perfect Sentences: {st.session_state.points['perfect_sentences']}<br/>"
+        f"Accuracy Rate: {st.session_state.stats['accuracy_rate']:.1f}%<br/>"
+        f"Unique Words Used: {len(st.session_state.stats['vocabulary_used'])}"
+    )
+    elements.append(Paragraph(stats_text, stats_style))
+    elements.append(Spacer(1, 30))
+
+    # เพิ่มเนื้อเรื่อง
+    elements.append(Paragraph("The Story", styles['Heading2']))
+    for entry in st.session_state.story:
+        if entry['role'] == 'You':
+            text = f"👤 {entry['content']}"
+            if entry.get('is_correct'):
+                style = ParagraphStyle(
+                    'Correct',
+                    parent=story_style,
+                    textColor=colors.darkgreen
+                )
+            else:
+                style = story_style
+        else:  # AI response
+            text = f"🤖 {entry['content']}"
+            style = ParagraphStyle(
+                'AI',
+                parent=story_style,
+                textColor=colors.navy
+            )
+        elements.append(Paragraph(text, style))
+
+    # เพิ่ม achievements ที่ได้รับ
+    if st.session_state.achievements:
+        elements.append(Spacer(1, 20))
+        elements.append(Paragraph("Achievements Earned", styles['Heading2']))
+        for achievement in st.session_state.achievements:
+            elements.append(Paragraph(f"🏆 {achievement}", stats_style))
+
+    # สร้าง PDF
+    doc.build(elements)
+    pdf = buffer.getvalue()
+    buffer.close()
+    return pdf
+
+# เพิ่มฟังก์ชันโหลดความก้าวหน้า
+def load_progress(uploaded_file):
+    """โหลดความก้าวหน้าจากไฟล์ JSON"""
+    try:
+        data = json.loads(uploaded_file.getvalue())
+        st.session_state.level = data['level']
+        st.session_state.story = data['story']
+        st.session_state.achievements = data['achievements']
+        st.session_state.points = data['points']
+        st.session_state.stats = data['stats']
+        st.session_state.stats['vocabulary_used'] = set(data['stats']['vocabulary_used'])
+        st.success("โหลดความก้าวหน้าเรียบร้อย!")
+        st.rerun()
+    except Exception as e:
+        st.error(f"เกิดข้อผิดพลาดในการโหลดไฟล์: {str(e)}")
+
 def reset_story():
     """Reset the story and related state variables."""
     st.session_state.story = []
@@ -589,6 +712,16 @@ show_parent_guide()  # Add parent guide right after welcome section
 
 # Sidebar for settings
 with st.sidebar:
+    st.markdown("### 📂 โห���ดความก้าวหน้า")
+    uploaded_file = st.file_uploader(
+        "เลือกไฟล์ .json",
+        type=['json'],
+        help="เลือกไฟล์ความก้าวหน้าที่บันทึกไว้"
+    )
+    if uploaded_file:
+        if st.button("โหลดความก้าวหน้า"):
+            load_progress(uploaded_file)
+            
     st.markdown("""
         <div class="thai">
         🎯 เลือกระดับการเรียนรู้
@@ -615,6 +748,7 @@ with st.sidebar:
     if st.button("เริ่มเรื่องใหม่ | Start New Story"):
         st.session_state.should_reset = True
         st.rerun()
+        
 
 # Main content area
 col1, col2 = st.columns([3, 1])
@@ -763,18 +897,37 @@ with col2:
     
     # Save Story Button
     if st.session_state.story:
-        if st.button("บันทึกเรื่องราว | Save Story"):
+        st.markdown("### 💾 บันทึกเรื่องราว")
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            # บันทึกเป็น PDF
+            pdf = create_story_pdf()
+            st.download_button(
+                label="📑 ดาวน์โหลดเป็น PDF",
+                data=pdf,
+                file_name=f"my_story_{datetime.now().strftime('%Y%m%d')}.pdf",
+                mime="application/pdf"
+            )
+        
+        with col2:
+            # บันทึกข้อมูลดิบ (JSON) สำหรับโหลดกลับมาเล่นต่อ
             story_data = {
                 'level': st.session_state.level,
-                'date': datetime.datetime.now().isoformat(),
+                'date': datetime.now().isoformat(),
                 'story': st.session_state.story,
-                'achievements': st.session_state.badges,
-                'total_words': st.session_state.total_words,
-                'unique_words': list(st.session_state.unique_words)
+                'achievements': st.session_state.achievements,
+                'points': st.session_state.points,
+                'stats': {
+                    'total_sentences': st.session_state.stats['total_sentences'],
+                    'correct_first_try': st.session_state.stats['correct_first_try'],
+                    'accuracy_rate': st.session_state.stats['accuracy_rate'],
+                    'vocabulary_used': list(st.session_state.stats['vocabulary_used'])
+                }
             }
             st.download_button(
-                label="ดาวน์โหลดเรื่องราว | Download Story",
+                label="💾 บันทึกความก้าวหน้า",
                 data=json.dumps(story_data, indent=2),
-                file_name='joystory.json',
-                mime='application/json'
+                file_name=f"story_progress_{datetime.now().strftime('%Y%m%d')}.json",
+                mime="application/json"
             )