Add Code for ViTPose Installation

app.py

@@ -1,3 +1,52 @@
+import importlib
+import importlib.util
+import subprocess
+import sys
+import os
+
+def ensure_mmpose_installed(local_path='/data/WHAM/third-party/ViTPose'):
+    """
+    Check if 'mmpose' can be imported; if not, attempt to install it from local_path.
+    local_path should contain setup.py or pyproject.toml so that pip can install it.
+    """
+    package_name = 'mmpose'
+    # Try to find the spec for import
+    spec = importlib.util.find_spec(package_name)
+    if spec is not None:
+        try:
+            module = importlib.import_module(package_name)
+            print(f"'{package_name}' is already installed, version: {getattr(module, '__version__', 'unknown')}")
+            return True
+        except Exception as e:
+            print(f"Found '{package_name}', but import failed: {e}. Will attempt re-install.")
+    else:
+        print(f"'{package_name}' not found, attempting installation...")
+
+    # If we reach here, we need to install or reinstall
+    # Check that the directory exists
+    if not os.path.isdir(local_path):
+        raise FileNotFoundError(f"Specified install directory does not exist: {local_path}")
+
+    # Construct pip install command using the current Python executable
+    cmd = [sys.executable, "-m", "pip", "install", local_path]
+    print("Running command:", " ".join(cmd))
+    try:
+        subprocess.check_call(cmd)
+    except subprocess.CalledProcessError as e:
+        raise RuntimeError(f"Failed to install mmpose: {e}")
+
+    # After installation, try importing again
+    try:
+        module = importlib.import_module(package_name)
+        print(f"'{package_name}' installed and imported successfully, version: {getattr(module, '__version__', 'unknown')}")
+        return True
+    except Exception as e:
+        raise RuntimeError(f"Installed but still cannot import '{package_name}': {e}")
+
+ensure_mmpose_installed('/data/WHAM/third-party/ViTPose')
+
+
+
 import gradio as gr
 import numpy as np
 import cv2
@@ -554,4 +603,6 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True, show_error=True)
+    port = int(os.environ.get("GRADIO_SERVER_PORT", os.environ.get("PORT", 7860)))
+    server_name = str(os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"))
+    demo.launch(server_name=server_name, server_port=port, share=True, show_error=True)