Fixed some of the style tts loading issues

app.py

@@ -12,24 +12,53 @@ def ensure_styletts2():
     try:
         import styletts2  # noqa: F401
         return
-    except Exception:
+    except ModuleNotFoundError:
         pass
 
     subprocess.check_call([
-        sys.executable, "-m", "pip", "install", "--no-cache-dir", "--no-deps", "styletts2==0.1.6"
+        sys.executable, "-m", "pip", "install", "--upgrade", "--no-cache-dir", "--no-deps", "styletts2==0.1.6"
     ])
 
+def import_styletts2_class():
+    """
+    styletts2 PyPI package doesn't export StyleTTS2 at top-level.
+    Try a few known module locations and return the class/callable.
+    """
+    import importlib
+
+    # Try common locations seen in forks / packaged builds
+    candidates = [
+        ("styletts2", "StyleTTS2"),
+        ("styletts2.model", "StyleTTS2"),
+        ("styletts2.styletts2", "StyleTTS2"),
+        ("styletts2.api", "StyleTTS2"),
+    ]
+
+    for mod_name, attr in candidates:
+        try:
+            mod = importlib.import_module(mod_name)
+            if hasattr(mod, attr):
+                return getattr(mod, attr)
+        except Exception:
+            pass
+
+    # If none worked, print what's actually inside and fail loudly
+    import styletts2
+    raise ImportError(
+        "Could not locate StyleTTS2 class. "
+        f"styletts2 package loaded from: {getattr(styletts2, '__file__', 'unknown')}. "
+        f"Available attrs: {sorted([a for a in dir(styletts2) if not a.startswith('_')])}"
+    )
+
 ensure_styletts2()
 
 import io
 import uuid
 import soundfile as sf
 import gradio as gr
-
 import torch
 
-from styletts2 import StyleTTS2
-
+StyleTTS2 = import_styletts2_class()
 
 # ---------------------------
 # Global config