import gradio as gr
from transformers import pipeline
import tempfile

# โหลดโมเดลล่วงหน้า
classifier = pipeline("audio-classification", 
                      model="AmeerHesham/distilhubert-finetuned-baby_cry", 
                      device=-1)

# ฟังก์ชันสำหรับรัน Gradio
def classify_audio(audio):
    with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tmp:
        tmp.write(audio[1])
        tmp_path = tmp.name
    result = classifier(tmp_path)
    return result

# สร้างอินเทอร์เฟซ
iface = gr.Interface(fn=classify_audio, 
                     inputs=gr.Audio(source="upload", type="file"), 
                     outputs="json",
                     title="Baby Cry Detector",
                     description="Upload an audio file to classify whether it contains a baby's cry.")

# รันแอป
iface.launch()