import os
import sys
import traceback

print("Starting MuseTalk Debug App...")

try:
    print("1. Testing basic imports...")
    import torch
    print(f"   PyTorch version: {torch.__version__}")
    
    import gradio as gr
    print(f"   Gradio version: {gr.__version__}")
    
    print("2. Testing mmpose import...")
    from mmpose.apis import inference_topdown, init_model
    print("   mmpose imported successfully")
    
    print("3. Testing musetalk imports...")
    from musetalk.utils.blending import get_image
    from musetalk.utils.face_parsing import FaceParsing
    from musetalk.utils.audio_processor import AudioProcessor
    from musetalk.utils.utils import get_file_type, get_video_fps, datagen, load_all_model
    from musetalk.utils.preprocessing import get_landmark_and_bbox, read_imgs, coord_placeholder, get_bbox_range
    print("   musetalk imports successful")
    
    print("4. Testing model loading...")
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    print(f"   Device: {device}")
    
    print("5. Creating simple Gradio interface...")
    
    def test_function():
        return "Test successful!"
    
    with gr.Blocks() as demo:
        gr.Markdown("# MuseTalk Debug Interface")
        test_btn = gr.Button("Test")
        test_output = gr.Textbox()
        test_btn.click(fn=test_function, outputs=test_output)
    
    print("6. Launching Gradio...")
    demo.launch(server_name="0.0.0.0", server_port=7860)
    
except Exception as e:
    print(f"ERROR: {str(e)}")
    print("Full traceback:")
    traceback.print_exc()
    sys.exit(1)