"""
Gradio app setup and launch module

Author: Jakub Polnis
Copyright: Copyright 2025, Jakub Polnis
License: Apache 2.0
Email: jakubpolnis@gmail.com
"""

import gradio as gr
from predict import prediction
from transformers import AutoImageProcessor, AutoModelForImageClassification
from functools import partial

title = "Audio deepfake Classification"
description = """This space uses fine-tuned kubinooo/convnext-tiny-224-audio-deepfake-classification model to classify audio files.
"""

processor = AutoImageProcessor.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")
model = AutoModelForImageClassification.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")

demo = gr.Interface(
    title=title,
    inputs=gr.Audio( type="filepath",
        interactive=True,  # This prevents users from uploading their own images
        show_label=True,
        label="Select from examples below or upload/record your own audio"),
    fn=partial(prediction, processor=processor, model=model),
    outputs=gr.Label(
        num_top_classes=2,
    ),
    examples=[
        "src/31-MALE-VC-FAKE.wav",
        "src/2152-REAL.wav"
    ],
    description=description
)
demo.launch()