Update app.py

app.py

@@ -15,11 +15,14 @@ level = st.selectbox("Choose your level:", ["Beginner", "Intermediate", "Advance
 start_button = st.button("Start Creating Your Story!")
 
 # ตรวจสอบเมื่อกดปุ่มเริ่มต้น
+# การเช็คและบันทึกค่าใน session_state
 if start_button:
-    # ไปยังหน้าถัดไปเพื่อสร้างเรื่องราว
     st.session_state["level"] = level  # บันทึกระดับที่เลือกไว้ใน session
     st.session_state["story_started"] = True  # ใช้เพื่อเช็คว่าเริ่มเรื่องหรือยัง
-    st.experimental_rerun()  # รีเฟรชหน้าเพื่อลงไปยังส่วนการแต่งเรื่อง
+    # เช็คว่าค่าใน session_state ถูกบันทึกแล้ว ก่อนเรียก experimental_rerun
+    if st.session_state.get("story_started"):
+        st.experimental_rerun()  # รีเฟรชหน้าเพื่อลงไปยังส่วนการแต่งเรื่อง
+
 
 
 # ตรวจสอบว่าการแต่งเรื่องเริ่มแล้วหรือยัง