Update app.py

app.py

@@ -1,19 +1,56 @@
 import sys
 import subprocess
+import pkg_resources
+
+# List of required packages
+required_packages = {
+    "gradio": None,  # None means any version is acceptable
+    "torch": None,
+    "transformers": None,
+    "accelerate": None,
+    "pydantic": None,
+    "einops": None,
+    "timm": None,
+    "av": None
+}
+
+# Function to check if package is installed
+def is_package_installed(package_name):
+    try:
+        pkg_resources.get_distribution(package_name)
+        return True
+    except pkg_resources.DistributionNotFound:
+        return False
+
+# Install missing packages
+for package, version in required_packages.items():
+    if not is_package_installed(package):
+        print(f"Installing {package}...")
+        if version:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={version}"])
+        else:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
-# Try to import OpenCV; if not available, install it
+# Try to import OpenCV, install if missing
 try:
     import cv2
     print("OpenCV already installed.")
 except ImportError:
-    print("OpenCV not found. Installing opencv-python-headless...")
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python-headless"])
+    print("Installing OpenCV...")
     try:
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python-headless"])
         import cv2
         print("OpenCV successfully installed.")
-    except ImportError:
-        print("Failed to install OpenCV. Continuing without it...")
+    except:
+        print("Failed to install opencv-python-headless, trying opencv-python...")
+        try:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python"])
+            import cv2
+            print("OpenCV successfully installed.")
+        except:
+            print("Failed to install OpenCV. Continuing without it...")
 
+# Now import the rest of the packages
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -33,11 +70,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
     print("경고: HF_TOKEN 환경 변수가 설정되지 않았습니다. 비공개 모델에 접근할 수 없을 수 있습니다.")
 
-# --- 선택 사항: CPU 스레드 설정 ---
-# torch.set_num_threads(CPU_THREAD_COUNT)
-# os.environ["OMP_NUM_THREADS"] = str(CPU_THREAD_COUNT)
-# os.environ["MKL_NUM_THREADS"] = str(CPU_THREAD_COUNT)
-
+# --- 환경 설정 ---
 print("--- 환경 설정 ---")
 print(f"PyTorch 버전: {torch.__version__}")
 print(f"실행 장치: {torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")