Update app.py

app.py

@@ -24,9 +24,19 @@ def process_dataset_ui(
     Returns:
         Status message
     """
+    import traceback
+
     try:
+        print("=== Starting Dataset Processing ===")
+        print(f"Original Dataset: {original_dataset}")
+        print(f"Output Dataset: {output_dataset}")
+        print(f"Model Type: {model_type}")
+        print(f"Text Field: {text_field}")
+        print(f"Token provided: {bool(hf_token)}")
+
         # Set HuggingFace token
         os.environ["HF_TOKEN"] = hf_token
+        print("✓ Token set in environment")
 
         # Validate inputs
         if not original_dataset or not output_dataset:
@@ -38,7 +48,10 @@ def process_dataset_ui(
         if model_type not in ["qwen3", "lfm2"]:
             return "❌ Error: Model type must be either 'qwen3' or 'lfm2'"
 
+        print("✓ Input validation passed")
+
         # Process dataset
+        print("Starting dataset processing...")
         process_dataset(
             original_dataset=original_dataset,
             output_dataset=output_dataset,
@@ -49,7 +62,11 @@ def process_dataset_ui(
         return f"✅ Dataset processed successfully and uploaded to: {output_dataset}"
 
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        error_msg = f"❌ Error: {str(e)}\n\n"
+        error_msg += f"Error Type: {type(e).__name__}\n\n"
+        error_msg += f"Traceback:\n{traceback.format_exc()}"
+        print(error_msg)
+        return error_msg
 
 # Create Gradio interface
 with gr.Blocks(title="VyvoTTS Dataset Tokenizer") as demo: