Update app.py

app.py

@@ -6,6 +6,8 @@ import sys
 if not os.path.exists("VV_Clone_1"):
     subprocess.run(["git", "clone", "https://github.com/Socold/VV_Clone_1.git"])
 
+sys.path.insert(0, "VV_Clone_1")
+
 # Installer les dépendances
 subprocess.run([sys.executable, "-m", "pip", "install", "-q", 
     "torch", "transformers>=4.40.0", "accelerate", "gradio>=4.0.0",
@@ -13,57 +15,45 @@ subprocess.run([sys.executable, "-m", "pip", "install", "-q",
 
 subprocess.run([sys.executable, "-m", "pip", "install", "-e", "VV_Clone_1"])
 
-# Récupérer les secrets AVANT tout
-USERNAME = os.environ.get("AUTH_USERNAME")
-PASSWORD = os.environ.get("AUTH_PASSWORD")
-
-if not USERNAME or not PASSWORD:
-    print("❌ ERREUR: Configurez AUTH_USERNAME et AUTH_PASSWORD dans Settings!")
-    exit(1)
-
-print(f"✅ Authentification activée pour: {USERNAME}")
-
-# FORCER l'authentification en modifiant Gradio
+import torch
 import gradio as gr
 
-# Sauvegarder la méthode originale
-original_launch = gr.Blocks.launch
+# Configuration - Changé pour utiliser 7B
+MODEL_PATH = "aoi-ot/VibeVoice-7B"  # ← Modifié ici
 
-# Créer une nouvelle méthode qui force l'auth
-def forced_auth_launch(self, *args, **kwargs):
-    # Forcer l'authentification
-    kwargs['auth'] = (USERNAME, PASSWORD)
-    kwargs['auth_message'] = "Authentification VibeVoice requise"
-    # Configuration spéciale pour HuggingFace Spaces
-    kwargs['server_name'] = "0.0.0.0"
-    kwargs['share'] = False
-    kwargs['root_path'] = os.environ.get("SPACE_ID", "")
-    # Ignorer les erreurs de localhost
-    try:
-        print("🔒 AUTHENTIFICATION FORCÉE APPLIQUÉE")
-        return original_launch(self, *args, **kwargs)
-    except ValueError as e:
-        if "localhost" in str(e):
-            # Forcer le lancement même avec l'erreur
-            kwargs['share'] = True
-            return original_launch(self, *args, **kwargs)
-        raise
-
-# Remplacer la méthode launch
-gr.Blocks.launch = forced_auth_launch
-gr.Interface.launch = forced_auth_launch
+# Récupérer les secrets (invisibles publiquement)
+USERNAME = os.environ.get("AUTH_USERNAME")
+PASSWORD = os.environ.get("AUTH_PASSWORD")
 
-import torch
+print(f"Chargement du modèle: {MODEL_PATH}")
 print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
+print(f"VRAM disponible: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.2f} GB")
 
-# Changer de dossier AVANT l'import
+# Lancer le script principal avec les arguments
 os.chdir("VV_Clone_1")
-sys.path.insert(0, os.getcwd())
-
-# Configurer les arguments SANS --share
-sys.argv = ["gradio_demo.py", "--model_path", "aoi-ot/VibeVoice-7B"]
+sys.argv = ["gradio_demo.py", "--model_path", MODEL_PATH, "--share"]
 
-# Importer et lancer
+# Importer et modifier le main pour ajouter l'auth
 from demo.gradio_demo import main
-print("🚀 Lancement de VibeVoice avec authentification forcée...")
+
+# Si les secrets existent, modifier le fichier pour ajouter l'auth
+if USERNAME and PASSWORD:
+    with open("demo/gradio_demo.py", "r") as f:
+        content = f.read()
+    
+    # Remplacer la ligne launch pour ajouter auth
+    content = content.replace(
+        'demo.launch(share=args.share)',
+        f'demo.launch(share=args.share, auth=("{USERNAME}", "{PASSWORD}"))'
+    )
+    
+    with open("demo/gradio_demo.py", "w") as f:
+        f.write(content)
+    
+    print(f"Authentification activée pour l'utilisateur: {USERNAME}")
+else:
+    print("Pas d'authentification configurée (pas de secrets définis)")
+
+# Lancer le main
+print("Démarrage de l'interface VibeVoice-7B...")
 main()