Added Function to execute shell commands safely

app.py

@@ -3,13 +3,8 @@ import torch
 from transformers import pipeline
 
 # Loading the TTS and Vocoder ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-!git clone https://github.com/saheedniyi02/yarngpt.git
-!pip install -qU outetts uroman
-
-import os
 import re
 import json
-import torch
 import inflect
 import random
 import uroman as ur
@@ -20,6 +15,47 @@ import requests
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from outetts.wav_tokenizer.decoder import WavTokenizer
 
+# Function to execute shell commands safely
+def run_command(command):
+    try:
+        process = subprocess.Popen(
+            command,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            shell=True,
+            universal_newlines=True
+        )
+        stdout, stderr = process.communicate()
+        
+        if process.returncode != 0:
+            print(f"Error executing: {command}")
+            print(stderr)
+            return False
+        else:
+            print(stdout)
+            return True
+    except Exception as e:
+        print(f"Exception during execution of {command}: {e}")
+        return False
+
+# Clone the YarnGPT repository
+if not os.path.exists('yarngpt'):
+    print("Cloning YarnGPT repository...")
+    run_command("git clone https://github.com/saheedniyi02/yarngpt.git")
+else:
+    print("YarnGPT repository already exists")
+
+# Install required packages
+print("Installing required packages...")
+run_command("pip install -q outetts uroman")
+
+# If you need to install the cloned package in development mode
+if os.path.exists('yarngpt'):
+    os.chdir('yarngpt')
+    run_command("pip install -e .")
+    os.chdir('..')  # Go back to the original directory
+    
+
 # Download files using Python's requests library instead of !wget
 def download_file(url, save_path):
     response = requests.get(url, stream=True)