Upload app.py

app.py

@@ -1,42 +1,15 @@
 import gradio as gr
 import sys
 import os
+import logging
+import time
+import threading
+import shutil
 
 now_dir = os.getcwd()
 sys.path.append(now_dir)
 
-
-class InstallationError(Exception):
-    def __init__(self, message="InstallationError"):
-        self.message = message
-        super().__init__(self.message)
-
-
-try:
-    system_drive = os.getenv("SystemDrive")
-    current_drive = os.path.splitdrive(now_dir)[0]
-    if current_drive.upper() != system_drive.upper():
-        raise InstallationError(
-            f"Error: Current working directory is not on the default system drive ({system_drive}). Please move Applio in the correct drive."
-        )
-except:
-    pass
-else:
-    if "OneDrive" in now_dir:
-        raise InstallationError(
-            "Error: Current working directory is on OneDrive. Please move Applio in another folder."
-        )
-    elif " " in now_dir:
-        raise InstallationError(
-            "Error: Current working directory contains spaces. Please move Applio in another folder."
-        )
-    try:
-        now_dir.encode("ascii")
-    except UnicodeEncodeError:
-        raise InstallationError(
-            "Error: Current working directory contains non-ASCII characters. Please move Applio in another folder."
-        )
-
+# Tabs
 from tabs.inference.inference import inference_tab
 from tabs.train.train import train_tab
 from tabs.extra.extra import extra_tab
@@ -45,16 +18,21 @@ from tabs.download.download import download_tab
 from tabs.tts.tts import tts_tab
 from tabs.settings.presence import presence_tab
 from tabs.settings.themes import theme_tab
+from tabs.plugins.plugins import plugins_tab
 
+# Assets
 import assets.themes.loadThemes as loadThemes
-
 from assets.i18n.i18n import I18nAuto
-
-i18n = I18nAuto()
-
+import assets.installation_checker as installation_checker
 from assets.discord_presence import RPCManager
+import assets.delete_models as delete_models
 
+delete_models.start_infinite_loop()
+i18n = I18nAuto()
 RPCManager.start_presence()
+installation_checker.check_installation()
+logging.getLogger("uvicorn").disabled = True
+logging.getLogger("fairseq").disabled = True
 
 my_applio = loadThemes.load_json()
 if my_applio:
@@ -77,8 +55,8 @@ with gr.Blocks(theme=my_applio, title="Applio") as Applio:
     with gr.Tab(i18n("Inference")):
         inference_tab()
 
-#    with gr.Tab(i18n("Train")):
-#        train_tab()
+    with gr.Tab(i18n("Train")):
+        train_tab()
 
     with gr.Tab(i18n("TTS")):
         tts_tab()
@@ -86,16 +64,24 @@ with gr.Blocks(theme=my_applio, title="Applio") as Applio:
     with gr.Tab(i18n("Extra")):
         extra_tab()
 
+    with gr.Tab(i18n("Plugins")):
+        plugins_tab()
+
     with gr.Tab(i18n("Download")):
         download_tab()
 
     with gr.Tab(i18n("Report a Bug")):
         report_tab()
 
-#    with gr.Tab(i18n("Settings")):
-#        presence_tab()
-#        theme_tab()
+    with gr.Tab(i18n("Settings")):
+        presence_tab()
+        theme_tab()
 
 
 if __name__ == "__main__":
-    Applio.launch()
+    Applio.launch(
+        favicon_path="assets/ICON.ico",
+        share="--share" in sys.argv,
+        inbrowser="--open" in sys.argv,
+        server_port=6969,
+    )