Update handler.py

handler.py

@@ -6,6 +6,7 @@ import base64
 import numpy as np
 import librosa
 from scipy.io import wavfile
+import shutil
 
 class EndpointHandler:
     def __init__(self, model_dir=None):
@@ -17,9 +18,15 @@ class EndpointHandler:
             repo_url = "https://huggingface.co/mazalaai/TTS_Mongolian.git"
             os.system(f"git clone {repo_url}")
 
-            # Change directory to the cloned repository
+            # Copy all files from the cloned repository to the /repository directory
             repo_dir = "TTS_Mongolian"
-            os.chdir(repo_dir)
+            dest_dir = "/repository"
+            for item in os.listdir(repo_dir):
+                item_path = os.path.join(repo_dir, item)
+                if os.path.isfile(item_path):
+                    shutil.copy(item_path, dest_dir)
+                elif os.path.isdir(item_path):
+                    shutil.copytree(item_path, os.path.join(dest_dir, item))
 
             # Import the voice_processing module and functions
             from voice_processing import tts, get_model_names, voice_mapping, get_unique_filename
@@ -34,6 +41,17 @@ class EndpointHandler:
             return {"error": str(e)}
         except Exception as e:
             return {"error": str(e)}
+        finally:
+            # Clean up the cloned repository and copied files/directories
+            if os.path.exists(repo_dir):
+                shutil.rmtree(repo_dir)
+            for item in os.listdir(dest_dir):
+                if item.startswith("TTS_Mongolian"):
+                    item_path = os.path.join(dest_dir, item)
+                    if os.path.isfile(item_path):
+                        os.remove(item_path)
+                    elif os.path.isdir(item_path):
+                        shutil.rmtree(item_path)
 
     def process_json_input(self, json_data):
         if all(key in json_data for key in ["model_name", "tts_text", "selected_voice", "slang_rate", "use_uploaded_voice"]):