Update app.py

app.py

@@ -1,20 +1,47 @@
 import os
 import sys
-
-import gradio as gr
-import numpy as np
-import random
 import shutil
+import subprocess
+
+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}")
+    else:
+        print(f"Removing existing repository {repo_dir} and cloning again...")
+        shutil.rmtree(repo_dir)
+        os.system(f"git clone {repo_url}")
+
+    assert os.path.exists(repo_dir), "Failed to clone repository!"
+
+def main():
+    # Install necessary packages
+    install_openblas_and_huggingface_hub()
 
-'''
-if not os.path.exists("sd-ggml-cpp-dp"):
-    os.system("git clone https://huggingface.co/svjack/sd-ggml-cpp-dp")
-else:
-    shutil.rmtree("sd-ggml-cpp-dp")
-    os.system("git clone https://huggingface.co/svjack/sd-ggml-cpp-dp")
-assert os.path.exists("sd-ggml-cpp-dp")
-'''
+    # Clone repository
+    clone_repo()
 
+    print("Setup complete!")
 
-os.system("apt-get install libopenblas-base")
-os.system("pip install huggingface_hub")
+if __name__ == "__main__":
+    main()