Update app.py

app.py

@@ -104,9 +104,15 @@ 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()
 
+# เพิ่มฟังก์ชันสำหรับจัดการ input
+def clear_input():
+    st.session_state.user_input = ""
+
 def show_welcome_section():
     st.markdown("""
         <div class="welcome-header">
@@ -457,15 +463,32 @@ with col1:
         </div>
     """, unsafe_allow_html=True)
     
-    user_input = st.text_area("เขียนต่อจากเรื่องราว | Continue the story:", height=100)
+    user_input = st.text_area(
+        "เขียนต่อจากเรื่องราว | Continue the story:",
+        height=100,
+        key="user_input",  # ใช้ key เดียวกับ session state
+        value=st.session_state.user_input  # ผูก value กับ session state
+    )
     
     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.")