Update app.py

app.py

@@ -293,6 +293,7 @@ gc.enable()
 gc.set_threshold(700, 10, 10)
 
 passive_gc_active = True
+passive_gc_thread = None
 
 def force_gc():
     """Force aggressive garbage collection"""
@@ -322,9 +323,24 @@ def passive_gc_daemon():
         except Exception as e:
             logger.error(f"[PASSIVE-GC] Error: {e}")
 
-passive_gc_thread = threading.Thread(target=passive_gc_daemon, daemon=True)
-passive_gc_thread.start()
-logger.info("[PASSIVE-GC] Background garbage collector started (30s intervals)")
+def start_passive_gc():
+    """Start passive garbage collector thread"""
+    global passive_gc_thread
+    if passive_gc_thread is None or not passive_gc_thread.is_alive():
+        passive_gc_thread = threading.Thread(target=passive_gc_daemon, daemon=True)
+        passive_gc_thread.start()
+        logger.info("[PASSIVE-GC] Background garbage collector started (30s intervals)")
+
+def stop_passive_gc():
+    """Stop passive garbage collector thread gracefully"""
+    global passive_gc_active, passive_gc_thread
+    passive_gc_active = False
+    if passive_gc_thread and passive_gc_thread.is_alive():
+        passive_gc_thread.join(timeout=2.0)
+        logger.info("[PASSIVE-GC] Background garbage collector stopped")
+
+# Start the passive GC thread
+start_passive_gc()
 
 def nuclear_ram_clear():
     """NUCLEAR option: Clear all Python caches and force full GC"""
@@ -903,6 +919,13 @@ class ZeroEngine:
         self.auto_cleanup_thread.start()
         logger.info("[IDLE] Idle monitor started (20s timeout)")
     
+    def stop_idle_monitor(self):
+        """Stop idle monitor thread gracefully"""
+        if self.auto_cleanup_thread and self.auto_cleanup_thread.is_alive():
+            # Set a flag to stop the monitor (would need to add shared flag for proper cleanup)
+            # For now, we'll let it be a daemon thread that exits with the process
+            logger.info("[IDLE] Idle monitor will stop on process exit")
+    
     def update_activity(self):
         """Update last activity timestamp"""
         self.last_activity = time.time()
@@ -1408,7 +1431,10 @@ kernel = ZeroEngine()
 # Session ID for token tracking
 session_id = "turtle170"
 
-with gr.Blocks(title="ZeroEngine V0.2", css=CUSTOM_CSS) as demo:
+with gr.Blocks(title="ZeroEngine V0.2") as demo:
+    demo_css = CUSTOM_CSS
+    if hasattr(demo, 'css'):
+        demo.css = demo_css
     gr.LoginButton()
     # Header with Token Display
     with gr.Row():
@@ -1639,6 +1665,48 @@ with gr.Blocks(title="ZeroEngine V0.2", css=CUSTOM_CSS) as demo:
 
 # --- LAUNCH ---
 if __name__ == "__main__":
+    import atexit
+    import signal
+    
+    def cleanup_on_exit():
+        """Cleanup function called on application exit"""
+        logger.info("[CLEANUP] Starting graceful shutdown...")
+        
+        # Stop passive GC
+        stop_passive_gc()
+        
+        # Cleanup kernel resources
+        if kernel and kernel.llm:
+            try:
+                del kernel.llm
+                kernel.llm = None
+                logger.info("[CLEANUP] Model unloaded")
+            except Exception as e:
+                logger.error(f"[CLEANUP] Model cleanup error: {e}")
+        
+        # Final garbage collection
+        try:
+            for _ in range(3):
+                gc.collect(2)
+            logger.info("[CLEANUP] Final GC complete")
+        except Exception as e:
+            logger.error(f"[CLEANUP] Final GC error: {e}")
+        
+        logger.info("[CLEANUP] Shutdown complete")
+    
+    # Register cleanup functions
+    atexit.register(cleanup_on_exit)
+    
+    def signal_handler(signum, frame):
+        """Handle shutdown signals gracefully"""
+        logger.info(f"[CLEANUP] Received signal {signum}")
+        cleanup_on_exit()
+        import sys
+        sys.exit(0)
+    
+    signal.signal(signal.SIGTERM, signal_handler)
+    signal.signal(signal.SIGINT, signal_handler)
+    
     demo.queue(max_size=20).launch(
         server_name="0.0.0.0",
         server_port=7861,