Update app.py

app.py

@@ -2651,8 +2651,20 @@ def show_sidebar():
         else:
             st.sidebar.info("ยังไม่มีเรื่องราวที่จะรีเซ็ต")
 
+def clear_input_state():
+    """Clear all input-related session states"""
+    if 'text_input' in st.session_state:
+        st.session_state.text_input = ""
+    if 'story_input_area' in st.session_state:
+        st.session_state.story_input_area = ""
+    if 'last_submission' in st.session_state:
+        del st.session_state.last_submission
+    st.session_state.clear_input = True
+    # Increment clear count to force new input field
+    st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+
 def show_story_input():
-    """Display story input section with clear button"""
+    """Display story input section with improved clear functionality"""
     st.markdown("""
         <div class="thai-eng">
             <div class="thai">✏️ ถึงตาคุณแล้ว</div>
@@ -2660,55 +2672,66 @@ def show_story_input():
         </div>
     """, unsafe_allow_html=True)
     
-    # Initialize clear_input flag if not exists
+    # Initialize states 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
+    if 'text_input' not in st.session_state:
+        st.session_state.text_input = ""
+    if 'last_submission' not in st.session_state:
+        st.session_state.last_submission = None
 
     # Show remaining sentences if in ending mode
     if st.session_state.get('ending_mode'):
         remaining = st.session_state.sentences_to_end
         st.info(f"🎭 โหมดจบเรื่อง - เหลืออีก {remaining} ประโยค")
     
-    # Input area
+    # Input area with key based on clear state
     text_input = st.text_area(
         "เขียนต่อจากเรื่องราว | Continue the story:",
-        value=st.session_state.get('text_input', ""),
+        value=st.session_state.text_input,
         height=100,
-        key="story_input_area",
+        key=f"story_input_area_{st.session_state.get('clear_count', 0)}",
         help="พิมพ์ประโยคภาษาอังกฤษเพื่อต่อเรื่อง",
         label_visibility="collapsed"
     )
     
-    # Save current input to session state
+    # Keep track of current input
     st.session_state.text_input = text_input
     
-    # Submit and Clear buttons with character count
+    # Submit and Clear buttons
     col1, col2, col3 = st.columns([3, 1, 1])
     
     with col1:
-        if st.button("📝 ส่งคำตอบ | Submit", use_container_width=True):
-            if not text_input.strip():
+        submit_button = st.button(
+            "📝 ส่งคำตอบ | Submit",
+            use_container_width=True,
+            disabled=not text_input.strip()  # Disable if empty
+        )
+        
+        if submit_button:
+            if text_input.strip():
+                try:
+                    # Save current submission
+                    st.session_state.last_submission = text_input.strip()
+                    with st.spinner("กำลังวิเคราะห์ประโยค..."):
+                        handle_story_submission(text_input.strip())
+                except Exception as e:
+                    logging.error(f"Error submitting story: {str(e)}")
+                    st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
+            else:
                 st.warning("กรุณาเขียนข้อความก่อนส่ง")
-                return
-            try:
-                with st.spinner("กำลังวิเคราะห์ประโยค..."):
-                    handle_story_submission(text_input.strip())
-            except Exception as e:
-                logging.error(f"Error submitting story: {str(e)}")
-                st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
     
     with col2:
-        # Add Clear button with space padding
-        if st.button("🗑️ ล้างข้อความ | Clear", use_container_width=True):
-            st.session_state.text_input = ""
-            st.session_state.clear_input = True
+        # Clear button with unique key
+        if st.button(
+            "🗑️ ล้างข้อความ | Clear",
+            key=f"clear_button_{st.session_state.get('clear_count', 0)}",
+            use_container_width=True,
+            disabled=not text_input.strip()  # Disable if already empty
+        ):
+            # Increment clear count to force new key generation
+            st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+            clear_input_state()
             st.rerun()
     
     with col3:
@@ -2720,8 +2743,12 @@ def show_story_input():
             </div>
         """, unsafe_allow_html=True)
 
+    # Warning for long text
+    if char_count > 200:
+        st.warning("⚠️ ข้อความยาวเกิน 200 ตัวอักษร")
+
 def handle_story_submission(text: str):
-    """Handle story submission and processing"""
+    """Handle story submission with improved state management"""
     if not st.session_state.story:
         st.error("กรุณาเลือกธีมเรื่องราวก่อนเริ่มเขียน")
         return
@@ -2729,6 +2756,8 @@ def handle_story_submission(text: str):
     # Check if in ending mode
     if st.session_state.get('ending_mode'):
         handle_ending_mode(text)
+        # Clear input after handling ending mode
+        clear_input_state()
         return
 
     # Regular processing
@@ -2795,14 +2824,23 @@ def handle_story_submission(text: str):
         # Update session stats
         update_session_stats()
         
-        # Set flag to clear input on next rerun
-        st.session_state.clear_input = True
+        # Clear input state completely before rerun
+        clear_input_state()
+        
+        # Make sure we don't have any lingering text
+        if 'story_input_area' in st.session_state:
+            st.session_state.story_input_area = ""
+        
+        # Increment clear count to force new input field
+        st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
         
         # Rerun to update UI
         st.rerun()
         
     except Exception as e:
         logging.error(f"Error in story submission: {str(e)}")
+        # Clear input even on error to prevent stuck state
+        clear_input_state()
         raise