Update app.py

app.py

@@ -572,15 +572,26 @@ def init_theme_state():
         st.session_state.theme_story_starter = None
 
 def reset_story():
-    """Reset story and related state variables"""
+    """Reset story and related state variables completely"""
     try:
-        # Reset story-related states
-        st.session_state.story = []
-        st.session_state.feedback = None
-        st.session_state.theme_story_starter = None
+        # Reset all story-related states
+        new_states = {
+            'story': [],
+            'feedback': None,
+            'theme_story_starter': None,
+            'text_input': "",
+            'current_theme': None,
+            'ending_mode': False,
+            'sentences_to_end': 0,
+            'ending_type': None,
+            'story_completed': False,
+            'stitched_story': None,
+            'clear_input': True,
+            'last_submission': None
+        }
         
         # Reset points
-        st.session_state.points = {
+        new_points = {
             'total': 0,
             'perfect_sentences': 0,
             'corrections_made': 0,
@@ -589,7 +600,7 @@ def reset_story():
         }
         
         # Reset stats
-        st.session_state.stats = {
+        new_stats = {
             'total_sentences': 0,
             'correct_first_try': 0,
             'accuracy_rate': 0.0,
@@ -600,12 +611,27 @@ def reset_story():
             'session_duration': 0
         }
         
-        # Reset progress
+        # Reset all session states
+        for key, value in new_states.items():
+            st.session_state[key] = value
+        st.session_state.points = new_points
+        st.session_state.stats = new_stats
+        
+        # Reset achievements and progress
+        st.session_state.achievements = []
         st.session_state.current_milestone = 0
         st.session_state.next_milestone = 5
         
-        # Reset flag
-        st.session_state.should_reset = False
+        # Reset UI elements
+        if 'story_input_area' in st.session_state:
+            del st.session_state.story_input_area
+        
+        # Force new input field creation
+        st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+        
+        # Reset theme selection
+        st.session_state.theme_button_counter = 0
+        st.session_state.theme_selection_id = datetime.now().strftime('%Y%m%d%H%M%S')
         
         # Log reset
         logging.info("Story state reset successfully")
@@ -1419,6 +1445,10 @@ def handle_theme_selection(theme: dict):
     """Handle theme selection and initialization"""
     try:
         with st.spinner("กำลังเตรียมเรื่องราว..."):
+            # Reset story completely first
+            reset_story()
+            
+            # Then set new theme
             st.session_state.current_theme = theme['id']
             starter = generate_dynamic_story_starter(
                 theme['id'],
@@ -1500,6 +1530,12 @@ def show_story():
     story_display = st.container()
     
     with story_display:
+        # Clear previous story if resetting
+        if st.session_state.get('should_reset'):
+            reset_story()
+            st.rerun()
+            return
+            
         if not st.session_state.story:
             st.info("เลือกธีมเรื่องราวที่ต้องการเพื่อเริ่มต้นการผจญภัย!")
             return
@@ -3521,6 +3557,12 @@ def main():
         init_session_state()
         init_theme_state()
 
+        # Check if reset is needed
+        if st.session_state.get('should_reset'):
+            reset_story()
+            st.rerun()
+            return
+
         # Initialize audio system
         initialize_audio()