Update app.py

app.py

@@ -1599,10 +1599,10 @@ def show_story_progress():
 
 def show_story_ending_options():
     """Display story ending options and guidance"""
-    if len(st.session_state.story) >= 10:  # แสดงตัวเลือกเมื่อเรื่องยาวพอ
+    if len(st.session_state.story) >= 10:  # Show options when story is long enough
         st.markdown("### 🎭 ต้องการจบเรื่องหรือไม่?")
         
-        # แสดงตัวเลือกวิธีจบเรื่อง
+        # Display ending type options
         ending_type = st.radio(
             "เลือกวิธีจบเรื่อง:",
             options=[
@@ -1619,7 +1619,7 @@ def show_story_ending_options():
             st.session_state.ending_mode = True
             st.session_state.ending_type = ending_type
             st.session_state.sentences_to_end = 5
-            st.rerun()
+            st.rerun()  # Using st.rerun() instead of st.experimental_rerun()
 
 def handle_ending_mode(text: str):
     """Handle story submission during ending mode"""
@@ -1634,12 +1634,12 @@ def handle_ending_mode(text: str):
             "is_corrected": False,
             "is_correct": True,
             "timestamp": datetime.now().isoformat(),
-            "remaining_sentences": remaining  # Include remaining sentences
+            "remaining_sentences": remaining
         })
         
         # Decrease remaining count for user's sentence
         st.session_state.sentences_to_end -= 1
-        remaining = st.session_state.sentences_to_end  # Update local variable
+        remaining = st.session_state.sentences_to_end
         
         if remaining > 0:
             # Generate AI's ending continuation
@@ -1649,7 +1649,7 @@ def handle_ending_mode(text: str):
                 remaining_sentences=remaining
             )
             
-            # Add AI response with remaining count
+            # Add AI response
             st.session_state.story.append({
                 "role": "AI",
                 "content": ai_response,
@@ -1659,7 +1659,7 @@ def handle_ending_mode(text: str):
             
             # Decrease remaining sentences
             st.session_state.sentences_to_end -= 1
-            remaining = st.session_state.sentences_to_end
+            
         else:
             # Generate final ending
             final_response = generate_final_ending(
@@ -1677,18 +1677,16 @@ def handle_ending_mode(text: str):
             
             # Mark story as complete
             st.session_state.story_completed = True
-            complete_story()
-        
+            
         # Clear input and rerun
         st.session_state.text_input = ""
         st.session_state.clear_input = True
-        st.rerun()
+        st.rerun()  # Using st.rerun() instead of st.experimental_rerun()
         
     except Exception as e:
         logging.error(f"Error in ending mode: {str(e)}")
         st.error("เกิดข้อผิดพลาดในโหมดจบเรื่อง กรุณาลองใหม่อีกครั้ง")
 
-
 def generate_final_ending(story: List[dict], ending_type: str) -> str:
     """Generate the final ending sentence based on the story and chosen ending type"""
     try:
@@ -2513,7 +2511,7 @@ def main():
             st.session_state.sentences_to_end = 0
             st.session_state.ending_type = None
             st.session_state.story_completed = False
-            st.experimental_rerun()
+            st.rerun()  # Using st.rerun() instead of st.experimental_rerun()
         
         # Auto-save progress periodically
         if st.session_state.story: