Update app.py

app.py

@@ -73,8 +73,6 @@ def play_audio(file_path):
 
 
 
-
-
 def download_audio(url):
     ydl_opts = {
         'format': 'bestaudio/best',
@@ -213,30 +211,38 @@ def extract_zip(extraction_folder, zip_name):
         if os.path.isdir(os.path.join(extraction_folder, filepath)):
             shutil.rmtree(os.path.join(extraction_folder, filepath))
 
-def download_online_model(url, dir_name):
+
+def download_online_model(url, dir_name, models_dir='./models'):
     try:
         print(f'[~] Downloading voice model with name {dir_name}...')
         zip_name = url.split('/')[-1]
         extraction_folder = os.path.join(models_dir, dir_name)
+        
         if os.path.exists(extraction_folder):
-            raise Exception(f'Voice model directory {dir_name} already exists! Choose a different name for your voice model.')
+            return f'[!] Voice model directory {dir_name} already exists! Choose a different name for your voice model.'
 
+        # Download from pixeldrain
         if 'pixeldrain.com' in url:
             url = f'https://pixeldrain.com/api/file/{zip_name}'
-        if 'drive.google.com' in url:
-          zip_name = dir_name + ".zip"
-          gdown.download(url, output=zip_name, use_cookies=True, quiet=True, fuzzy=True)
+            urllib.request.urlretrieve(url, zip_name)
+        # Download from Google Drive
+        elif 'drive.google.com' in url:
+            zip_name = dir_name + ".zip"
+            gdown.download(url, output=zip_name, use_cookies=True, quiet=True)
         else:
+            # General URL download
             urllib.request.urlretrieve(url, zip_name)
 
         print(f'[~] Extracting zip file...')
         extract_zip(extraction_folder, zip_name)
         print(f'[+] {dir_name} Model successfully downloaded!')
+        
+        # Return success message after successful download
+        return f"[+] {dir_name} Model successfully downloaded!"
 
     except Exception as e:
-        raise Exception(str(e))
-
-
+        # Return the error message instead of raising an exception
+        return f'[!] Error: {str(e)}'
 
 
 if __name__ == '__main__':