Update app.py

app.py

@@ -1,16 +1,55 @@
-import os
 import gradio as gr
+import subprocess
+import os
+import uuid
+
+OUTPUT_DIR = "outputs"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+def generate_music(prompt):
+    uid = str(uuid.uuid4())[:8]
+    out_wav = os.path.join(OUTPUT_DIR, f"{uid}.wav")
+
+    cmd = [
+        "python",
+        "infer/run_infer.py",
+        "--model", "bolshyC/Muse-0.6b",
+        "--prompt", prompt,
+        "--out", out_wav
+    ]
+
+    try:
+        subprocess.run(
+            cmd,
+            check=True,
+            timeout=600
+        )
+    except Exception as e:
+        return None, f"生成失败：{e}"
+
+    if not os.path.exists(out_wav):
+        return None, "生成失败：未找到音频文件"
+
+    return out_wav, "生成完成"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎵 Muse Music Generator (HF Space)")
+    gr.Markdown("使用 Muse-0.6b 在 HF Space 上生成音乐（CLI 接入）")
+
+    prompt = gr.Textbox(
+        label="Prompt",
+        placeholder="A calm piano melody, slow tempo"
+    )
 
-def info():
-    return "Muse Space 已启动。请查看 Logs 使用命令行推理。"
+    btn = gr.Button("Generate")
+    audio = gr.Audio(label="Output", type="filepath")
+    status = gr.Textbox(label="Status")
 
-demo = gr.Interface(
-    fn=info,
-    inputs=[],
-    outputs="text",
-    title="Muse (CLI-based)",
-    description="Muse 官方仓库未提供 Web UI。本 Space 仅用于启动环境。"
-)
+    btn.click(
+        fn=generate_music,
+        inputs=prompt,
+        outputs=[audio, status]
+    )
 
 if __name__ == "__main__":
     demo.launch()