Update app.py

app.py

@@ -132,14 +132,22 @@ st.markdown("""
 
 # Initialize session state variables
 def init_session_state():
+    """Initialize all session state variables"""
     if 'story' not in st.session_state:
         st.session_state.story = []
     if 'feedback' not in st.session_state:
         st.session_state.feedback = None
-    if 'corrections' not in st.session_state:
-        st.session_state.corrections = {}  # เก็บประวัติการแก้ไข
     if 'level' not in st.session_state:
         st.session_state.level = 'Beginner'
+    if 'unique_words' not in st.session_state:
+        st.session_state.unique_words = set()
+    if 'total_words' not in st.session_state:
+        st.session_state.total_words = 0
+    if 'should_reset' not in st.session_state:
+        st.session_state.should_reset = False
+    if 'user_input' not in st.session_state:
+        st.session_state.user_input = ""
+    # เพิ่ม points และ stats
     if 'points' not in st.session_state:
         st.session_state.points = {
             'total': 0,
@@ -155,14 +163,12 @@ def init_session_state():
             'accuracy_rate': 0.0,
             'vocabulary_used': set()
         }
-    if 'unique_words' not in st.session_state:
-        st.session_state.unique_words = set()
-    if 'total_words' not in st.session_state:
-        st.session_state.total_words = 0
-    if 'badges' not in st.session_state:
-        st.session_state.badges = []
-    if 'should_reset' not in st.session_state:
-        st.session_state.should_reset = False
+    # เพิ่ม achievements
+    if 'achievements' not in st.session_state:
+        st.session_state.achievements = []
+    # เพิ่มชื่อผู้เล่น (optional)
+    if 'player_name' not in st.session_state:
+        st.session_state.player_name = "Young Writer"
 
 init_session_state()
 
@@ -545,7 +551,7 @@ def show_achievements():
         """, unsafe_allow_html=True)
 
         # แสดงสถิติ
-        st.markdown("""### 📊 สถิติการเขียน""")
+        st.markdown("### 📊 สถิติการเขียน")
         col1, col2 = st.columns(2)
         with col1:
             st.metric("ประโยคที่เขียนทั้งหมด", st.session_state.stats['total_sentences'])
@@ -554,13 +560,13 @@ def show_achievements():
             st.metric("ความแม่นยำ", f"{st.session_state.stats['accuracy_rate']:.1f}%")
             st.metric("คำศัพท์ที่ใช้", len(st.session_state.stats['vocabulary_used']))
 
-        # แสดง Achievements
+        # แสดง Achievements ที่มี
         st.markdown("### 🏆 ความสำเร็จ")
-        for achievement_id, achievement in achievements_list.items():
-            if achievement['condition']():
-                st.success(f"{achievement['name']}: {achievement['description']}")
-            else:
-                st.info(f"🔒 {achievement['description']} (ยังไม่ปลดล็อก)")
+        if st.session_state.achievements:
+            for achievement in st.session_state.achievements:
+                st.success(f"🏆 {achievement}")
+        else:
+            st.info("ยังไม่มีความสำเร็จ - เขียนต่อไปเพื่อปลดล็อกรางวัล!")
 
 def update_achievements(text: str):
     """Update user achievements based on their writing."""
@@ -664,7 +670,7 @@ def create_story_pdf():
         elements.append(Paragraph(text, style))
 
     # เพิ่ม achievements ที่ได้รับ
-    if st.session_state.achievements:
+    if hasattr(st.session_state, 'achievements') and st.session_state.achievements:
         elements.append(Spacer(1, 20))
         elements.append(Paragraph("Achievements Earned", styles['Heading2']))
         for achievement in st.session_state.achievements: