Update app.py

app.py

@@ -505,6 +505,11 @@ def init_session_state():
             if key not in st.session_state:
                 st.session_state[key] = value
 
+    if 'clear_input' not in st.session_state:
+        st.session_state.clear_input = False
+    if 'text_input' not in st.session_state:
+        st.session_state.text_input = ""
+
 def init_theme_state():
     """Initialize theme-specific state variables"""
     if 'current_theme' not in st.session_state:
@@ -1727,15 +1732,30 @@ def show_story_input():
         </div>
     """, unsafe_allow_html=True)
     
+    # Initialize clear_input flag if not exists
+    if 'clear_input' not in st.session_state:
+        st.session_state.clear_input = False
+    
+    # Default value for text input
+    default_value = "" if st.session_state.clear_input else st.session_state.get('text_input', "")
+    
+    # If clear_input flag is True, reset it
+    if st.session_state.clear_input:
+        st.session_state.clear_input = False
+    
     # Input area
     text_input = st.text_area(
         "เขียนต่อจากเรื่องราว | Continue the story:",
+        value=default_value,
         height=100,
         key="story_input_area",
         help="พิมพ์ประโยคภาษาอังกฤษเพื่อต่อเรื่อง",
         label_visibility="collapsed"
     )
     
+    # Save current input to session state
+    st.session_state.text_input = text_input
+    
     # Submit button with character count
     col1, col2 = st.columns([3, 1])
     with col1:
@@ -1801,8 +1821,11 @@ def handle_story_submission(text: str):
         # Update session stats
         update_session_stats()
         
-        # Clear input
-        st.session_state.story_input_area = ""
+        # Set flag to clear input on next rerun
+        st.session_state.clear_input = True
+        
+        # Rerun to update UI
+        st.rerun()
         
     except Exception as e:
         logging.error(f"Error in story submission: {str(e)}")