Update app.py

app.py

@@ -1,16 +1,31 @@
 import os
+import subprocess
 
 os.system("pip install gradio==3.50")
 
-if not os.path.exists('ninja-linux.zip'):
-    os.system("wget https://github.com/ninja-build/ninja/releases/download/v1.10.2/ninja-linux.zip")
-    os.system("unzip ninja-linux.zip -d ninja")  # Unzip it to a local directory
 
-# Set Ninja to the PATH
-os.environ['PATH'] = os.getcwd() + '/ninja:' + os.environ['PATH']
 
-# Check if Ninja is installed correctly
-os.system('ninja --version')
+# Function to download and install Ninja
+def install_ninja():
+    # Check if ninja is already installed
+    if subprocess.run(["ninja", "--version"], capture_output=True).returncode != 0:
+        print("Ninja not found, downloading...")
+        # Download Ninja
+        subprocess.run(["wget", "https://github.com/ninja-build/ninja/releases/download/v1.11.1/ninja-linux.zip"], check=True)
+        
+        # Unzip it
+        subprocess.run(["unzip", "ninja-linux.zip"], check=True)
+        
+        # Move Ninja to a directory in PATH
+        os.rename("ninja", "/usr/local/bin/ninja")  # You may need to change this path
+        print("Ninja installed successfully.")
+
+# Call the function to install Ninja
+install_ninja()
+
+# Now, you can check if Ninja is installed
+os.environ["PATH"] += os.pathsep + "/usr/local/bin"
+print("Ninja version:", subprocess.run(["ninja", "--version"], capture_output=True).stdout.decode().strip())
 
 
 from argparse import Namespace