Update app.py

app.py

@@ -104,8 +104,6 @@ def init_session_state():
         st.session_state.badges = []
     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 = ""
 
 init_session_state()
 
@@ -113,6 +111,25 @@ init_session_state()
 def clear_input():
     st.session_state.user_input = ""
 
+# Callback function for submit button
+def submit_story():
+    if st.session_state.text_input.strip():
+        # Add user's input to story
+        st.session_state.story.append({"role": "You", "content": st.session_state.text_input})
+        
+        # Update achievements
+        update_achievements(st.session_state.text_input)
+        
+        # Generate feedback
+        st.session_state.feedback = provide_feedback(st.session_state.text_input, st.session_state.level)
+        
+        # Generate AI continuation
+        ai_response = generate_story_continuation(st.session_state.text_input, st.session_state.level)
+        st.session_state.story.append({"role": "AI", "content": ai_response})
+        
+        # Clear the input by setting it to empty
+        st.session_state.text_input = ""
+
 def show_welcome_section():
     st.markdown("""
         <div class="welcome-header">
@@ -463,35 +480,18 @@ with col1:
         </div>
     """, unsafe_allow_html=True)
     
-    user_input = st.text_area(
+    # Text input with callback
+    st.text_area(
         "เขียนต่อจากเรื่องราว | Continue the story:",
         height=100,
-        key="user_input",  # ใช้ key เดียวกับ session state
-        value=st.session_state.user_input  # ผูก value กับ session state
+        key="text_input"
     )
     
-    if st.button("ส่งคำตอบ | Submit"):
-        if user_input.strip():
-            # Add user's input to story
-            st.session_state.story.append({"role": "You", "content": user_input})
-            
-            # Update achievements
-            update_achievements(user_input)
-            
-            # Generate feedback
-            st.session_state.feedback = provide_feedback(user_input, st.session_state.level)
-            
-            # Generate AI continuation
-            ai_response = generate_story_continuation(user_input, st.session_state.level)
-            st.session_state.story.append({"role": "AI", "content": ai_response})
-            
-            # Clear the input
-            clear_input()
-            
-            # Rerun the app
-            st.rerun()
-        else:
-            st.warning("กรุณาเขียนเนื้อเรื่องก่อนกดส่ง | Please write something to continue the story.")
+    # Submit button with callback
+    st.button(
+        "ส่งคำตอบ | Submit",
+        on_click=submit_story
+    )
 
 with col2:
     # Feedback Display