Update app.py

app.py

@@ -952,7 +952,15 @@ def generate_dynamic_story_starter(theme_id: str, level: str) -> Dict[str, str]:
         'th': starter_th
     }
 
-
+def convert_sets_to_lists(obj):
+    """แปลง set เป็น list สำหรับการ serialize เป็น JSON"""
+    if isinstance(obj, set):
+        return list(obj)
+    elif isinstance(obj, dict):
+        return {key: convert_sets_to_lists(value) for key, value in obj.items()}
+    elif isinstance(obj, list):
+        return [convert_sets_to_lists(item) for item in obj]
+    return obj
             
 def get_theme_story_starter(theme_id: str, level: str) -> Dict[str, str]:
     """Get a dynamically generated story starter for the selected theme and level."""
@@ -1702,8 +1710,12 @@ def load_progress(uploaded_file):
         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.session_state.stats = {
+            'total_sentences': data['stats']['total_sentences'],
+            'correct_first_try': data['stats']['correct_first_try'],
+            'accuracy_rate': data['stats']['accuracy_rate'],
+            'vocabulary_used': set(data['stats']['vocabulary_used'])  # แปลง list กลับเป็น set
+        }
         st.success("โหลดความก้าวหน้าเรียบร้อย!")
         st.rerun()
     except Exception as e:
@@ -1952,22 +1964,32 @@ with col2:
             )
         
         with col2:
-            # บันทึกความก้าวหน้า
+            # บันทึกความก้าวหน้า (แก้ไขส่วนนี้)
             story_data = {
                 'level': st.session_state.level,
                 'date': datetime.now().isoformat(),
                 'story': st.session_state.story,
                 'achievements': st.session_state.achievements,
                 'points': st.session_state.points,
-                'stats': st.session_state.stats
+                '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'])  # แปลง set เป็น list
+                }
             }
+            
+            # แปลงข้อมูลทั้งหมดที่อาจมี set
+            serializable_data = convert_sets_to_lists(story_data)
+            
             st.download_button(
                 label="💾 Save",
-                data=json.dumps(story_data, indent=2),
+                data=json.dumps(serializable_data, indent=2),
                 file_name=f"story_progress_{datetime.now().strftime('%Y%m%d')}.json",
                 mime="application/json"
             )
 
+
 # Add new CSS for improved layout
 st.markdown("""
     <style>