Update app.py

app.py

@@ -2112,9 +2112,6 @@ def main():
     except Exception as e:
         handle_application_error(e)
 
-if __name__ == "__main__":
-    main()
-    
 def check_session_status():
     """Check and maintain session status"""
     try:
@@ -2122,16 +2119,23 @@ def check_session_status():
         if 'session_start' not in st.session_state:
             st.session_state.session_start = datetime.now()
         
+        # Calculate session duration
+        current_time = datetime.now()
+        session_duration = (current_time - st.session_state.session_start).total_seconds()
+        
         # Check for session timeout (2 hours)
-        session_duration = (datetime.now() - st.session_state.session_start).total_seconds()
         if session_duration > 7200:  # 2 hours
-            st.warning("เซสชันหมดอายุ กรุณาบันทึกความก้าวหน้าและรีเฟรชหน้าเว็บ")
+            st.warning("⚠️ เซสชันหมดอายุ กรุณาบันทึกความก้าวหน้าและรีเฟรชหน้าเว็บ")
         
         # Check for inactivity (30 minutes)
+        if 'last_interaction' not in st.session_state:
+            st.session_state.last_interaction = datetime.now().isoformat()
+            
         last_interaction = datetime.fromisoformat(st.session_state.last_interaction)
-        inactivity_duration = (datetime.now() - last_interaction).total_seconds()
+        inactivity_duration = (current_time - last_interaction).total_seconds()
+        
         if inactivity_duration > 1800:  # 30 minutes
-            st.info("ไม่มีกิจกรรมเป็นเวลานาน กรุณาบันทึกความก้าวหน้าเพื่อความปลอดภัย")
+            st.info("💤 ไม่มีกิจกรรมเป็นเวลานาน กรุณาบันทึกความก้าวหน้าเพื่อความปลอดภัย")
         
         # Update stats if story exists
         if st.session_state.story: