Update app.py

app.py

@@ -549,6 +549,8 @@ def init_session_state():
         st.session_state.clear_input = False
     if 'text_input' not in st.session_state:
         st.session_state.text_input = ""
+    if 'story_input_area' not in st.session_state:
+        st.session_state.story_input_area = ""
 
 def init_theme_state():
     """Initialize theme-specific state variables"""
@@ -1882,42 +1884,35 @@ def show_story_input():
             <div class="eng">Your Turn</div>
         </div>
     """, unsafe_allow_html=True)
+
+    # ใช้ callback function เพื่อเคลียร์ input
+    def clear_text():
+        st.session_state.story_input_area = ""
     
-    # 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
+    # Input area with default empty value and callback
     text_input = st.text_area(
         "เขียนต่อจากเรื่องราว | Continue the story:",
-        value=default_value,
+        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:
-        if st.button("📝 ส่งคำตอบ | Submit", use_container_width=True):
+        if st.button("📝 ส่งคำตอบ | Submit", use_container_width=True, on_click=clear_text):
             if not text_input.strip():
                 st.warning("กรุณาเขียนข้อความก่อนส่ง")
                 return
                 
             try:
                 with st.spinner("กำลังวิเคราะห์ประโยค..."):
-                    handle_story_submission(text_input.strip())
+                    if st.session_state.get('ending_mode'):
+                        handle_ending_mode(text_input.strip())
+                    else:
+                        handle_story_submission(text_input.strip())
             except Exception as e:
                 logging.error(f"Error submitting story: {str(e)}")
                 st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
@@ -1961,26 +1956,18 @@ def handle_story_submission(text: str):
         
         # Generate AI continuation
         try:
-            logging.info("Attempting to generate AI continuation...")
-            # ใช้ข้อความที่ถูกต้องสำหรับการต่อเรื่อง
-            text_for_continuation = feedback_data['corrected'] if feedback_data.get('has_errors') else text
-            
-            ai_response = generate_story_continuation(text_for_continuation, st.session_state.level)
-            
-            # Log the AI response
-            logging.info(f"AI Response generated: {ai_response}")
-            
+            ai_response = generate_story_continuation(text, st.session_state.level)
             if ai_response and ai_response.strip():
                 st.session_state.story.append({
                     "role": "AI",
                     "content": ai_response,
                     "timestamp": datetime.now().isoformat()
                 })
-                logging.info("AI response added to story successfully")
             else:
-                logging.error("AI generated empty response")
-                # กรณีที่ AI ไม่สร้างประโยค ให้สร้างประโยคง่ายๆ ตาม theme
-                fallback_response = generate_fallback_response(st.session_state.current_theme, st.session_state.level)
+                fallback_response = generate_fallback_response(
+                    st.session_state.current_theme, 
+                    st.session_state.level
+                )
                 st.session_state.story.append({
                     "role": "AI",
                     "content": fallback_response,
@@ -1988,20 +1975,22 @@ def handle_story_submission(text: str):
                 })
         except Exception as e:
             logging.error(f"Error generating AI continuation: {str(e)}")
-            # สร้าง fallback response เมื่อเกิดข้อผิดพลาด
-            fallback_response = generate_fallback_response(st.session_state.current_theme, st.session_state.level)
+            fallback_response = generate_fallback_response(
+                st.session_state.current_theme, 
+                st.session_state.level
+            )
             st.session_state.story.append({
                 "role": "AI",
                 "content": fallback_response,
                 "timestamp": datetime.now().isoformat()
             })
         
+        # Clear input after successful submission
+        st.session_state.story_input_area = ""
+        
         # Update session stats
         update_session_stats()
         
-        # Set flag to clear input on next rerun
-        st.session_state.clear_input = True
-        
         # Rerun to update UI
         st.rerun()