Update app.py

app.py

@@ -1,47 +1,30 @@
-import os
-import sys
-import shutil
 import subprocess
+import os
 
-def install_openblas_and_huggingface_hub():
-    # Step 1: Install OpenBLAS base using apt-get
-    try:
-        print("Installing libopenblas-base...")
-        os.system("sudo apt-get update && sudo apt-get install -y libopenblas-base")
-    except Exception as e:
-        print(f"Error installing libopenblas-base: {e}")
-        sys.exit(1)
-
-    # Step 2: Install huggingface_hub Python package using pip
-    try:
-        print("Installing huggingface_hub Python package...")
-        os.system("pip install huggingface_hub")
-    except Exception as e:
-        print(f"Error installing huggingface_hub: {e}")
-        sys.exit(1)
-
-def clone_repo():
-    repo_url = "https://huggingface.co/svjack/sd-ggml-cpp-dp"
-    repo_dir = "sd-ggml-cpp-dp"
-    
-    if not os.path.exists(repo_dir):
-        print(f"Cloning repository from {repo_url}...")
-        os.system(f"git clone {repo_url}")
+def run_command(command, cwd=None):
+    """Executes a shell command and prints output"""
+    result = subprocess.run(command, cwd=cwd, text=True, capture_output=True, shell=True)
+    if result.returncode != 0:
+        print(f"Error executing command: {command}")
+        print(result.stderr)
     else:
-        print(f"Removing existing repository {repo_dir} and cloning again...")
-        shutil.rmtree(repo_dir)
-        os.system(f"git clone {repo_url}")
+        print(result.stdout)
 
-    assert os.path.exists(repo_dir), "Failed to clone repository!"
+def install_openblas():
+    # Step 1: Clone the OpenBLAS repository
+    print("Cloning the OpenBLAS repository...")
+    run_command("git clone https://github.com/OpenMathLib/OpenBLAS.git")
 
-def main():
-    # Install necessary packages
-    install_openblas_and_huggingface_hub()
+    # Step 2: Change to the OpenBLAS directory
+    os.chdir("OpenBLAS")
 
-    # Clone repository
-    clone_repo()
+    # Step 3: Build OpenBLAS
+    print("Building OpenBLAS...")
+    run_command("make")
 
-    print("Setup complete!")
+    # Step 4: Install OpenBLAS
+    print("Installing OpenBLAS...")
+    run_command("sudo make install")
 
 if __name__ == "__main__":
-    main()
+    install_openblas()