Update app.py

app.py

@@ -439,7 +439,15 @@ def passive_gc_daemon():
     global passive_gc_active
     while passive_gc_active:
         try:
-            time.sleep(30)
+            # Use shorter sleep intervals to allow faster shutdown
+            for _ in range(30):  # 30 seconds = 30 * 1 second sleeps
+                if not passive_gc_active:
+                    break
+                time.sleep(1)
+            
+            if not passive_gc_active:
+                break
+                
             if AGGRESSIVE_GC:
                 total_collected = 0
                 for pass_num in range(3):
@@ -453,6 +461,7 @@ def passive_gc_daemon():
                     logger.info(f"[PASSIVE-GC] Aggressive cleanup: {total_collected} objects collected")
         except Exception as e:
             logger.error(f"[PASSIVE-GC] Error: {e}")
+            break
 
 def start_passive_gc():
     """Start passive garbage collector thread"""
@@ -467,8 +476,12 @@ def stop_passive_gc():
     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")
+        # Wait for thread to finish with timeout
+        passive_gc_thread.join(timeout=5.0)
+        if passive_gc_thread.is_alive():
+            logger.warning("[PASSIVE-GC] Thread did not shut down gracefully")
+        else:
+            logger.info("[PASSIVE-GC] Background garbage collector stopped")
 
 # Start the passive GC thread
 start_passive_gc()
@@ -1861,7 +1874,7 @@ if __name__ == "__main__":
         """Cleanup function called on application exit"""
         logger.info("[CLEANUP] Starting graceful shutdown...")
         
-        # Stop passive GC
+        # Stop passive GC first
         stop_passive_gc()
         
         # Cleanup kernel resources
@@ -1873,8 +1886,19 @@ if __name__ == "__main__":
             except Exception as e:
                 logger.error(f"[CLEANUP] Model cleanup error: {e}")
         
-        # Final garbage collection
+        # Final garbage collection with asyncio cleanup
         try:
+            import asyncio
+            # Close any remaining event loops
+            try:
+                loop = asyncio.get_event_loop()
+                if loop and not loop.is_closed():
+                    loop.close()
+            except RuntimeError:
+                # No event loop or already closed
+                pass
+            
+            # Final GC
             for _ in range(3):
                 gc.collect(2)
             logger.info("[CLEANUP] Final GC complete")