clean-up

app.py

@@ -139,16 +139,6 @@ async def hf_logo(request: web.Request) -> web.Response:
     content = open(os.path.join(os.path.dirname(__file__), "public", "hf-logo.svg"), "r").read()
     return web.Response(content_type="image/svg+xml", text=content)
 
-async def on_shutdown(app: web.Application):
-    """Cleanup function to be called on server shutdown."""
-    logger.info("Server shutdown initiated, cleaning up resources...")
-
-    if 'engine' in app:
-        await app['engine'].cleanup()
-        logger.info("Engine instance cleaned up")
-
-    logger.info("Server shutdown complete")
-
 async def initialize_app() -> web.Application:
     """Initialize and configure the web application."""
     try:
@@ -161,8 +151,6 @@ async def initialize_app() -> web.Application:
         app = web.Application()
         app['engine'] = engine
 
-        app.on_shutdown.append(on_shutdown)
-
         # Configure routes
         app.router.add_get("/", index)
         app.router.add_get("/index.js", js_index)

client/src/lib/facePoke.ts

@@ -248,49 +248,19 @@ export class FacePoke {
   }
 
   /**
-   * Handles interruption messages from the server.
-   * @param message - The interruption message.
+   * Cleans up resources and closes connections.
    */
-  private handleInterruption(message: string): void {
-    console.warn(`[FacePoke] Interruption: ${message}`);
-    this.emitEvent('interruption', message);
-  }
-
-  /**
-   * Toggles the microphone on or off.
-   * @param isOn - Whether to turn the microphone on (true) or off (false).
-   */
-  public async toggleMicrophone(isOn: boolean): Promise<void> {
-    console.log(`[FacePoke] Attempting to ${isOn ? 'start' : 'stop'} microphone`);
-    try {
-      if (isOn) {
-        await this.startMicrophone();
-      } else {
-        this.stopMicrophone();
-      }
-      this.emitEvent('microphoneToggled', isOn);
-    } catch (error) {
-      console.error(`[FacePoke] Error toggling microphone:`, error);
-      this.emitEvent('microphoneError', error);
-      throw error;
+  public cleanup(): void {
+    console.log('[FacePoke] Starting cleanup process');
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
     }
+    this.eventListeners.clear();
+    console.log('[FacePoke] Cleanup completed');
+    this.emitEvent('cleanup');
   }
 
-
-/**
- * Cleans up resources and closes connections.
- */
-public cleanup(): void {
-  console.log('[FacePoke] Starting cleanup process');
-  if (this.ws) {
-    this.ws.close();
-    this.ws = null;
-  }
-  this.eventListeners.clear();
-  console.log('[FacePoke] Cleanup completed');
-  this.emitEvent('cleanup');
-}
-
   /**
    * Modifies an image based on the provided parameters
    * @param image - The data-uri base64 image to modify.

engine.py

@@ -1,26 +1,20 @@
-import time
 import logging
 import hashlib
-import uuid
 import os
 import io
-import shutil
 import asyncio
 import base64
-from concurrent.futures import ThreadPoolExecutor
 from queue import Queue
-from typing import Dict, Any, List, Optional, AsyncGenerator, Tuple, Union
+from typing import Dict, Any, List, Optional, Union
 from functools import lru_cache
-import av
 import numpy as np
-import cv2
 import torch
 import torch.nn.functional as F
 from PIL import Image
 
 from liveportrait.config.argument_config import ArgumentConfig
 from liveportrait.utils.camera import get_rotation_matrix
-from liveportrait.utils.io import load_image_rgb, load_driving_info, resize_to_limit
+from liveportrait.utils.io import resize_to_limit
 from liveportrait.utils.crop import prepare_paste_back, paste_back
 
 # Configure logging
@@ -62,7 +56,6 @@ class Engine:
 
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-        # cache for the "modify image" workflow
         self.image_cache = {}  # Stores the original images
         self.processed_cache = {}  # Stores the processed image data
 
@@ -273,19 +266,6 @@ class Engine:
         x_d_new[0, 15, 1] -= params.get('pupil_y', 0) * 0.001
         params['eyes'] = params.get('eyes', 0) - params.get('pupil_y', 0) / 2.
 
-    async def cleanup(self):
-        """
-        Perform cleanup operations for the Engine.
-        This method should be called when shutting down the application.
-        """
-        logger.info("Starting Engine cleanup")
-        try:
-            # TODO: Add any additional cleanup operations here
-            logger.info("Engine cleanup completed successfully")
-        except Exception as e:
-            logger.error(f"Error during Engine cleanup: {str(e)}")
-            logger.exception("Full traceback:")
-
 def create_engine(models):
     logger.info("Creating Engine instance...")
 

public/index.js

@@ -29858,25 +29858,6 @@ class FacePoke {
     console.error(`[FacePoke][${this.connectionId}] WebSocket error:`, error);
     this.emitEvent("websocketError", error);
   }
-  handleInterruption(message) {
-    console.warn(`[FacePoke] Interruption: ${message}`);
-    this.emitEvent("interruption", message);
-  }
-  async toggleMicrophone(isOn) {
-    console.log(`[FacePoke] Attempting to ${isOn ? "start" : "stop"} microphone`);
-    try {
-      if (isOn) {
-        await this.startMicrophone();
-      } else {
-        this.stopMicrophone();
-      }
-      this.emitEvent("microphoneToggled", isOn);
-    } catch (error) {
-      console.error(`[FacePoke] Error toggling microphone:`, error);
-      this.emitEvent("microphoneError", error);
-      throw error;
-    }
-  }
   cleanup() {
     console.log("[FacePoke] Starting cleanup process");
     if (this.ws) {