import gradio as gr
from transformers import pipeline

model_id = "k-habib/autotrain-scram3"
classifier = pipeline("text-classification", model=model_id)

def predict(job_description):
    if not job_description:
        return {"prediction": "N/A", "confidence": 0.0}
    
    result = classifier(job_description[:512])[0]  # Truncate input if needed
    return {
        "prediction": result["label"],
        "confidence": round(result["score"], 3)
    }

demo = gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=5, placeholder="Paste job description here..."),
    outputs="json",
    title="Fake Job Listing Detector",
    description="Paste a job description and get a prediction whether it's a scam."
)

demo.launch()