Update app.py

app.py

@@ -255,7 +255,7 @@ class BackendProcessor:
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - ZEROENGINE - %(message)s', force=True)
 logger = logging.getLogger(__name__)
 
-# Suppress asyncio warnings during shutdown (multiple methods)
+# Suppress asyncio warnings during shutdown (targeted approach)
 import warnings
 warnings.filterwarnings("ignore", category=RuntimeWarning, message=".*asyncio.*")
 warnings.filterwarnings("ignore", category=UserWarning, message=".*asyncio.*")
@@ -263,9 +263,22 @@ warnings.filterwarnings("ignore", category=UserWarning, message=".*asyncio.*")
 # Set environment variable to suppress asyncio warnings
 os.environ['PYTHONASYNCIODEBUG'] = '0'
 
-# Suppress asyncio event loop warnings
+# Store original get_event_loop for restoration
 import asyncio
-asyncio.get_event_loop = lambda: None  # Prevent event loop access during shutdown
+_original_get_event_loop = asyncio.get_event_loop
+
+# Custom get_event_loop that handles shutdown gracefully
+def _safe_get_event_loop():
+    try:
+        return _original_get_event_loop()
+    except RuntimeError as e:
+        if "no running event loop" in str(e).lower() or "closed" in str(e).lower():
+            # Return None gracefully instead of raising
+            return None
+        raise
+
+# Replace with safe version
+asyncio.get_event_loop = _safe_get_event_loop
 
 # --- KERNEL INITIALIZATION ---
 try:
@@ -1900,51 +1913,20 @@ if __name__ == "__main__":
             except Exception as e:
                 logger.error(f"[CLEANUP] Model cleanup error: {e}")
         
-        # Final garbage collection with asyncio cleanup
+        # Final garbage collection with minimal asyncio cleanup
         logger.info("[CLEANUP] Performing final cleanup...")
         try:
-            # More aggressive asyncio cleanup
+            # Simple asyncio cleanup - don't interfere with Gradio
             import asyncio
-            import sys
-            
-            # Force close all event loops
             try:
-                # Get all running loops and close them
-                if hasattr(asyncio, 'all_tasks'):
-                    tasks = asyncio.all_tasks()
-                    logger.info(f"[CLEANUP] Found {len(tasks)} asyncio tasks")
-                    
-                    # Cancel all tasks
-                    for task in tasks:
-                        if not task.done():
-                            task.cancel()
-                    
-                    # Wait for tasks to complete
-                    if tasks:
-                        asyncio.gather(*tasks, return_exceptions=True)
-                        logger.info("[CLEANUP] All tasks cancelled")
-                
-                # Close event loops
-                if hasattr(asyncio, 'get_event_loop'):
-                    try:
-                        loop = asyncio.get_event_loop()
-                        if loop and not loop.is_closed():
-                            logger.info("[CLEANUP] Closing main event loop...")
-                            loop.close()
-                            logger.info("[CLEANUP] Main event loop closed")
-                    except RuntimeError:
-                        logger.info("[CLEANUP] No event loop to close")
-                
-                # Try to close any remaining loops
-                if hasattr(asyncio, '_get_running_loop'):
-                    try:
-                        running_loop = asyncio._get_running_loop()
-                        if running_loop:
-                            logger.info("[CLEANUP] Closing running loop...")
-                            running_loop.close()
-                    except:
-                        pass
-                        
+                # Just try to close the main loop if it exists and is accessible
+                loop = _original_get_event_loop()
+                if loop and hasattr(loop, 'is_closed') and not loop.is_closed():
+                    logger.info("[CLEANUP] Closing event loop...")
+                    loop.close()
+                    logger.info("[CLEANUP] Event loop closed")
+            except (RuntimeError, AttributeError):
+                logger.info("[CLEANUP] No event loop to close")
             except Exception as e:
                 logger.info(f"[CLEANUP] Asyncio cleanup note: {e}")