Update app.py

app.py

@@ -12,7 +12,7 @@ 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
+from datetime import datetime 
 from typing import Dict, List, Optional, Tuple
 import random
 
@@ -428,13 +428,15 @@ def init_session_state():
             'streak': 0,
             'max_streak': 0
         }
+    if 'corrections' not in st.session_state:
+        st.session_state.corrections = {}
     if 'stats' not in st.session_state:
         st.session_state.stats = {
             'total_sentences': 0,
             'correct_first_try': 0,
             'accuracy_rate': 0.0,
-            'vocabulary_used': set()
-        }
+            'vocabulary_used': set(),
+            'corrections_made': 0
     if 'achievements' not in st.session_state:
         st.session_state.achievements = []
 
@@ -1326,6 +1328,7 @@ def apply_correction(story_index: int, corrected_text: str):
     """Apply correction to a specific story entry."""
     if 0 <= story_index < len(st.session_state.story):
         original_text = st.session_state.story[story_index]['content']
+        
         # เก็บประวัติการแก้ไข
         if 'corrections' not in st.session_state:
             st.session_state.corrections = {}
@@ -1333,14 +1336,21 @@ def apply_correction(story_index: int, corrected_text: str):
         st.session_state.corrections[story_index] = {
             'original': original_text,
             'corrected': corrected_text,
-            'timestamp': datetime.datetime.now().isoformat()
+            'timestamp': datetime.now().isoformat()  # แก้จาก datetime.datetime.now()
         }
+        
         # แก้ไขประโยคในเรื่อง
         st.session_state.story[story_index]['content'] = corrected_text
         st.session_state.story[story_index]['is_corrected'] = True
         
         # แสดงข้อความยืนยันการแก้ไข
         st.success("✅ แก้ไขประโยคเรียบร้อยแล้ว!")
+        
+        # อัพเดทสถิติ (ถ้ามี)
+        if 'stats' in st.session_state:
+            # อาจเพิ่มการนับจำนวนครั้งที่แก้ไข
+            st.session_state.stats['corrections_made'] = \
+                st.session_state.stats.get('corrections_made', 0) + 1
 
 def get_vocabulary_suggestions() -> List[str]:
     """Get contextual vocabulary suggestions with Thai translations."""