Update app.py

app.py

@@ -1,35 +1,33 @@
 import gradio as gr
-import sys
+import os
 import traceback
 
 def generate_audio(prompt, duration=10):
-    updates = []
     try:
-        updates.append("🔄 Starting...")
+        updates = []
+        updates.append("🔄 Starting with authentication...")
         
-        # Test basic imports first
-        updates.append("📦 Testing imports...")
-        import torch
-        updates.append(f"✅ PyTorch: {torch.__version__}")
+        # Get token from environment
+        token = os.getenv("HF_TOKEN")
+        if not token:
+            return "❌ No HF_TOKEN found in environment variables"
         
-        import stable_audio_tools
-        updates.append("✅ Stable Audio Tools imported")
+        updates.append("✅ Token found")
         
         from huggingface_hub import hf_hub_download
-        updates.append("✅ HuggingFace Hub ready")
+        updates.append("📦 Downloading with token...")
         
-        updates.append("⬇️ Attempting model download...")
         model_config = hf_hub_download(
             repo_id="stabilityai/stable-audio-open-1.0",
-            filename="model_config.json"
+            filename="model_config.json",
+            token=token
         )
-        updates.append("✅ Config downloaded")
+        updates.append("✅ Model config downloaded!")
         
-        return "\n".join(updates) + f"\n\n🎯 Ready to generate: '{prompt}' for {duration}s"
+        return "\n".join(updates) + f"\n\n🎯 Ready to generate: '{prompt}'"
         
     except Exception as e:
-        error_details = traceback.format_exc()
-        return f"❌ Error: {str(e)}\n\nFull error:\n{error_details}"
+        return f"❌ Error: {str(e)}\n\n{traceback.format_exc()}"
 
 demo = gr.Interface(
     fn=generate_audio,
@@ -37,8 +35,8 @@ demo = gr.Interface(
         gr.Textbox(label="Prompt", value="heavy boots thudding on wet sand"),
         gr.Slider(5, 20, 10, label="Duration")
     ],
-    outputs=gr.Textbox(label="Debug Info", lines=10),
-    title="🔧 Debug Mode"
+    outputs=gr.Textbox(label="Status", lines=8),
+    title="🔧 Token Test"
 )
 
 demo.launch()