import gradio as gr
from inference import EnsembleInference

model = EnsembleInference("Final_Model.pt")

def predict_text(text):
    result = model.predict(text)
    return f"{result['prediction']} (Confidence: {result['confidence']})\n\nDetails:\n- RoBERTa: {result['details']['roberta_confidence']}\n- DeBERTa: {result['details']['deberta_confidence']}"

iface = gr.Interface(
    fn=predict_text,
    inputs=gr.Textbox(lines=5, placeholder="Enter your text here..."),
    outputs="text",
    title="AI Detector",
    description="This model detects whether the text is AI-generated or human-written using an ensemble of RoBERTa and DeBERTa."
)

iface.launch()