Update app.py

app.py

@@ -37,20 +37,22 @@ os.makedirs(AUDIO_DIR, exist_ok=True)
 audio_cache = {}
 
 def load_chatterbox_model():
-    """Try multiple ways to load ChatterboxTTS"""
+    """Try multiple ways to load ChatterboxTTS from Resemble AI"""
     global MODEL, CHATTERBOX_AVAILABLE
     
-    # Method 1: Try original ChatterboxTTS import
+    # Method 1: Try Resemble AI ChatterboxTTS (most likely)
     try:
         from chatterbox.src.chatterbox.tts import ChatterboxTTS
-        logger.info("✅ Found ChatterboxTTS in chatterbox.src.chatterbox.tts")
+        logger.info("✅ Found Resemble AI ChatterboxTTS in chatterbox.src.chatterbox.tts")
         MODEL = ChatterboxTTS.from_pretrained(DEVICE)
         CHATTERBOX_AVAILABLE = True
         return True
     except ImportError as e:
-        logger.warning(f"Method 1 failed: {e}")
+        logger.warning(f"Method 1 (Resemble AI standard path) failed: {e}")
+    except Exception as e:
+        logger.warning(f"Method 1 failed with error: {e}")
     
-    # Method 2: Try direct chatterbox import
+    # Method 2: Try alternative import path for Resemble AI repo
     try:
         from chatterbox.tts import ChatterboxTTS
         logger.info("✅ Found ChatterboxTTS in chatterbox.tts")
@@ -59,58 +61,77 @@ def load_chatterbox_model():
         return True
     except ImportError as e:
         logger.warning(f"Method 2 failed: {e}")
+    except Exception as e:
+        logger.warning(f"Method 2 failed with error: {e}")
     
-    # Method 3: Try if it's installed as a package
+    # Method 3: Try direct chatterbox import
     try:
-        from chatterbox import ChatterboxTTS
-        logger.info("✅ Found ChatterboxTTS as package")
-        MODEL = ChatterboxTTS.from_pretrained(DEVICE)
+        import chatterbox
+        if hasattr(chatterbox, 'ChatterboxTTS'):
+            MODEL = chatterbox.ChatterboxTTS.from_pretrained(DEVICE)
+        elif hasattr(chatterbox, 'tts') and hasattr(chatterbox.tts, 'ChatterboxTTS'):
+            MODEL = chatterbox.tts.ChatterboxTTS.from_pretrained(DEVICE)
+        else:
+            raise ImportError("ChatterboxTTS not found in chatterbox module")
+        logger.info("✅ Found ChatterboxTTS via direct import")
         CHATTERBOX_AVAILABLE = True
         return True
     except ImportError as e:
         logger.warning(f"Method 3 failed: {e}")
+    except Exception as e:
+        logger.warning(f"Method 3 failed with error: {e}")
     
-    # Method 4: Try alternative package name
+    # Method 4: Try exploring the installed package
     try:
-        import chatterbox_tts
-        logger.info("✅ Found chatterbox_tts package")
-        MODEL = chatterbox_tts.ChatterboxTTS.from_pretrained(DEVICE)
-        CHATTERBOX_AVAILABLE = True
-        return True
+        import chatterbox
+        import inspect
+        
+        # Log what's available in the chatterbox package
+        logger.info(f"Chatterbox module path: {chatterbox.__file__}")
+        logger.info(f"Chatterbox contents: {dir(chatterbox)}")
+        
+        # Try to find ChatterboxTTS class anywhere in the module
+        for name, obj in inspect.getmembers(chatterbox):
+            if name == 'ChatterboxTTS' or (inspect.isclass(obj) and 'TTS' in name):
+                logger.info(f"Found potential TTS class: {name}")
+                MODEL = obj.from_pretrained(DEVICE)
+                CHATTERBOX_AVAILABLE = True
+                return True
+        
+        raise ImportError("ChatterboxTTS class not found in chatterbox package")
+        
     except ImportError as e:
         logger.warning(f"Method 4 failed: {e}")
+    except Exception as e:
+        logger.warning(f"Method 4 failed with error: {e}")
     
-    # Method 5: Try loading from local files
+    # Method 5: Check if the GitHub repo was installed correctly
     try:
-        import sys
-        # Add current directory to path
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        chatterbox_path = os.path.join(current_dir, "chatterbox", "src")
-        if os.path.exists(chatterbox_path):
-            sys.path.insert(0, chatterbox_path)
-            from chatterbox.tts import ChatterboxTTS
-            logger.info("✅ Found ChatterboxTTS in local files")
-            MODEL = ChatterboxTTS.from_pretrained(DEVICE)
-            CHATTERBOX_AVAILABLE = True
-            return True
+        import pkg_resources
+        try:
+            pkg_resources.get_distribution('chatterbox')
+            logger.info("✅ Chatterbox package is installed")
+        except pkg_resources.DistributionNotFound:
+            logger.warning("❌ Chatterbox package not found in installed packages")
+            
+        # Try to import and inspect what we got
+        import chatterbox
+        chatterbox_path = chatterbox.__path__[0] if hasattr(chatterbox, '__path__') else str(chatterbox.__file__)
+        logger.info(f"Chatterbox installed at: {chatterbox_path}")
+        
+        # List all available modules/classes
+        import pkgutil
+        for importer, modname, ispkg in pkgutil.walk_packages(chatterbox.__path__, chatterbox.__name__ + "."):
+            logger.info(f"Available module: {modname}")
+            
     except Exception as e:
-        logger.warning(f"Method 5 failed: {e}")
+        logger.warning(f"Package inspection failed: {e}")
     
-    # Method 6: Check if files exist and guide user
-    possible_locations = [
-        "./chatterbox/src/chatterbox/tts.py",
-        "./chatterbox/tts.py", 
-        "./chatterbox_tts.py",
-        "./src/chatterbox/tts.py"
-    ]
-    
-    existing_files = [loc for loc in possible_locations if os.path.exists(loc)]
-    if existing_files:
-        logger.error(f"❌ Found ChatterboxTTS files at {existing_files} but couldn't import!")
-        logger.error("💡 Check your file structure and __init__.py files")
-    else:
-        logger.error("❌ No ChatterboxTTS files found!")
-        logger.error("💡 Upload your ChatterboxTTS package to the Space")
+    # If we get here, the GitHub repo might have a different structure
+    logger.error("❌ Could not load ChatterboxTTS from Resemble AI repository")
+    logger.error("💡 The GitHub repo might have a different structure than expected")
+    logger.error("🔗 Repository: https://github.com/resemble-ai/chatterbox.git")
+    logger.error("📋 Check the repo's README for correct import instructions")
     
     return False
 
@@ -453,26 +474,20 @@ def create_gradio_interface():
             gr.Markdown("""
             ### 🚨 Currently Using Fallback Model
             
-            You're hearing beep sounds because the real ChatterboxTTS isn't loaded. To fix this:
+            You're hearing beep sounds because the real ChatterboxTTS isn't loaded. 
+            
+            **The Resemble AI ChatterboxTTS from GitHub should auto-install from requirements.txt.**
+            
+            If you're still seeing this message:
+            
+            1. **Check build logs** for any installation errors
+            2. **Verify requirements.txt** contains: `git+https://github.com/resemble-ai/chatterbox.git`
+            3. **Restart the Space** if needed
+            4. **Check logs** for import errors
             
-            1. **Upload your ChatterboxTTS package** to this Space
-            2. **Ensure proper file structure**: `chatterbox/src/chatterbox/tts.py`
-            3. **Add `__init__.py` files** in each directory
-            4. **Restart the Space** after uploading
+            📁 GitHub repo being used: https://github.com/resemble-ai/chatterbox.git
             
-            📁 Expected structure:
-            ```
-            your-space/
-            ├── app.py
-            ├── requirements.txt
-            └── chatterbox/
-                ├── __init__.py
-                └── src/
-                    ├── __init__.py
-                    └── chatterbox/
-                        ├── __init__.py
-                        └── tts.py
-            ```
+            If the GitHub installation fails, you can alternatively upload the package manually.
             """)
         
         with gr.Row():