Update app.py

app.py

@@ -18,61 +18,55 @@ subprocess.run([sys.executable, "-m", "pip", "install", "-e", "VV_Clone_1"])
 import torch
 import gradio as gr
 
-# Configuration
-MODEL_PATH = "aoi-ot/VibeVoice-7B"
-
-# DEBUG : Vérifier les secrets
-print("=== VÉRIFICATION AUTHENTIFICATION ===")
+# FORCER la récupération des secrets
 USERNAME = os.environ.get("AUTH_USERNAME")
 PASSWORD = os.environ.get("AUTH_PASSWORD")
-print(f"AUTH_USERNAME présent: {USERNAME is not None}")
-print(f"AUTH_PASSWORD présent: {PASSWORD is not None}")
-if USERNAME:
-    print(f"Utilisateur: {USERNAME}")
-print("====================================")
 
-print(f"Modèle: {MODEL_PATH}")
-print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
+if not USERNAME or not PASSWORD:
+    # Si pas de secrets, bloquer complètement
+    gr.Interface(
+        fn=lambda x: "❌ ERREUR: Configurez AUTH_USERNAME et AUTH_PASSWORD dans Settings → Variables and secrets!",
+        inputs="text", 
+        outputs="text",
+        title="Space non configuré"
+    ).launch()
+    exit()
 
+print(f"✅ Authentification activée pour: {USERNAME}")
+
+# Créer un nouveau script qui wrappe tout avec auth
+wrapper_script = f"""
+import sys
+sys.path.insert(0, "VV_Clone_1")
 os.chdir("VV_Clone_1")
 
-# Lire le fichier original
-with open("demo/gradio_demo.py", "r") as f:
-    original_content = f.read()
-
-# Si auth configurée, modifier le fichier
-if USERNAME and PASSWORD:
-    print("Modification du fichier pour ajouter l'authentification...")
-    
-    # Chercher différents patterns possibles
-    if "demo.launch(share=args.share)" in original_content:
-        new_content = original_content.replace(
-            "demo.launch(share=args.share)",
-            f'demo.launch(share=args.share, auth=("{USERNAME}", "{PASSWORD}"))'
-        )
-    elif "demo.launch()" in original_content:
-        new_content = original_content.replace(
-            "demo.launch()",
-            f'demo.launch(auth=("{USERNAME}", "{PASSWORD}"))'
-        )
-    else:
-        # Pattern plus général
-        import re
-        new_content = re.sub(
-            r'demo\.launch\((.*?)\)',
-            f'demo.launch(\\1, auth=("{USERNAME}", "{PASSWORD}"))',
-            original_content
-        )
-    
-    # Écrire le fichier modifié
-    with open("demo/gradio_demo.py", "w") as f:
-        f.write(new_content)
-    
-    print("✅ Authentification ajoutée")
-else:
-    print("⚠️  ATTENTION: Pas de secrets configurés - Space PUBLIC!")
-
-# Lancer
-sys.argv = ["gradio_demo.py", "--model_path", MODEL_PATH, "--share"]
-from demo.gradio_demo import main
-main()
+from demo.gradio_demo import VibeVoiceDemo
+import gradio as gr
+
+# Créer l'instance
+demo = VibeVoiceDemo(model_path="{MODEL_PATH}")
+
+# Créer l'interface Gradio
+interface = demo.launch()
+
+# Fermer l'instance sans auth
+interface.close()
+
+# Relancer AVEC authentification
+interface.launch(
+    auth=("{USERNAME}", "{PASSWORD}"),
+    auth_message="Entrez vos identifiants VibeVoice",
+    server_name="0.0.0.0",
+    server_port=7860,
+    share=False
+)
+"""
+
+MODEL_PATH = "aoi-ot/VibeVoice-7B"
+
+# Écrire et exécuter le wrapper
+with open("run_with_auth.py", "w") as f:
+    f.write(wrapper_script)
+
+os.chdir("VV_Clone_1")
+exec(open("../run_with_auth.py").read())