Update app.py

app.py

@@ -1,3 +1,47 @@
+import os
+import subprocess
+import sys
+import importlib
+
+def check_and_install_package(package_name):
+    try:
+        importlib.import_module(package_name)
+        print(f"{package_name} is already installed")
+    except ImportError:
+        print(f"Installing {package_name}...")
+        subprocess.check_call([sys.executable, '-m', 'pip', 'install', package_name])
+
+def install_requirements():
+    try:
+        # Upgrade pip first
+        subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', 'pip'])
+        
+        # Install requirements from file
+        requirements_path = os.path.join(os.path.dirname(__file__), 'requirement.txt')
+        if os.path.exists(requirements_path):
+            print("Installing requirements from requirements.txt...")
+            subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-r', requirements_path])
+            print("Successfully installed requirements")
+        else:
+            print(f"requirements.txt not found at {requirements_path}")
+            
+        # Explicitly check and install critical packages
+        critical_packages = ['sentencepiece', 'transformers', 'torch', 'gradio']
+        for package in critical_packages:
+            check_and_install_package(package)
+            
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing requirements: {e}")
+        raise
+    except Exception as e:
+        print(f"Unexpected error during installation: {e}")
+        raise
+
+# Install requirements first
+print("Starting installation of requirements...")
+install_requirements()
+print("Finished installing requirements")
+
 import gradio as gr
 import torch
 import os