Hugging Face's logo

Spaces:
wzy013
/
hunyuanvideo-foley
Running

App Files Community
hunyuanvideo-foley
File size: 3,543 Bytes 
55d09cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
 
"""
测试音频生成和 Gradio 兼容性
"""
import gradio as gr
import numpy as np
import wave
import tempfile
import os
from loguru import logger

def create_test_audio(text_prompt: str = "test audio") -> str:
    """创建测试音频文件"""
    sample_rate = 44100
    duration = 3.0
    duration_samples = int(duration * sample_rate)
    
    # 使用 numpy 生成音频
    t = np.linspace(0, duration, duration_samples, dtype=np.float32)
    
    # 根据文本生成不同音频
    if "footsteps" in text_prompt.lower():
        audio = 0.4 * np.sin(2 * np.pi * 2 * t) * np.exp(-3 * (t % 0.5))
    elif "rain" in text_prompt.lower():
        audio = 0.3 * np.random.randn(duration_samples)
    else:
        audio = 0.3 * np.sin(2 * np.pi * 440 * t)
    
    # 应用包络
    envelope = np.ones_like(audio)
    fade_samples = int(0.1 * sample_rate)
    envelope[:fade_samples] = np.linspace(0, 1, fade_samples)
    envelope[-fade_samples:] = np.linspace(1, 0, fade_samples)
    audio *= envelope
    
    # 保存为 WAV 文件
    temp_dir = tempfile.mkdtemp()
    audio_path = os.path.join(temp_dir, "test_audio.wav")
    
    # 规范化到 int16
    audio_normalized = np.clip(audio, -1.0, 1.0)
    audio_int16 = (audio_normalized * 32767).astype(np.int16)
    
    # 使用 wave 模块保存
    with wave.open(audio_path, 'w') as wav_file:
        wav_file.setnchannels(1)
        wav_file.setsampwidth(2)
        wav_file.setframerate(sample_rate)
        wav_file.writeframes(audio_int16.tobytes())
    
    logger.info(f"✅ 测试音频已保存: {audio_path}")
    return audio_path

def test_interface(text_input):
    """测试接口"""
    try:
        # 生成音频
        audio_path = create_test_audio(text_input)
        
        # 检查文件是否存在
        if os.path.exists(audio_path):
            file_size = os.path.getsize(audio_path)
            status = f"✅ 成功生成音频！\n文件路径: {audio_path}\n文件大小: {file_size} bytes\n文本: {text_input}"
            return audio_path, status
        else:
            return None, "❌ 音频文件未生成"
            
    except Exception as e:
        logger.error(f"错误: {e}")
        return None, f"❌ 生成失败: {str(e)}"

# 创建 Gradio 界面
def create_test_app():
    with gr.Blocks(title="Audio Test") as app:
        gr.HTML("<h1>🎵 音频兼容性测试</h1>")
        
        with gr.Row():
            with gr.Column():
                text_input = gr.Textbox(
                    label="文本输入",
                    value="footsteps on ground",
                    placeholder="输入文本描述..."
                )
                generate_btn = gr.Button("生成测试音频", variant="primary")
                
            with gr.Column():
                audio_output = gr.Audio(label="生成的音频")
                status_output = gr.Textbox(
                    label="状态信息",
                    lines=5,
                    interactive=False
                )
        
        generate_btn.click(
            fn=test_interface,
            inputs=[text_input],
            outputs=[audio_output, status_output]
        )
    
    return app

if __name__ == "__main__":
    # 设置日志
    logger.remove()
    logger.add(lambda msg: print(msg, end=''), level="INFO")
    
    logger.info("启动音频测试应用...")
    
    app = create_test_app()
    app.launch(
        server_name="0.0.0.0",
        server_port=7861,
        share=False,
        debug=True
    )