import gradio as gr
from setfit import SetFitModel

# Download from Hub and run inference
model = SetFitModel.from_pretrained("peter2000/vulnerable-groups-setfit")


group_dict = {
    1: 'Women and girls',
 2: 'Children and youth',
 3: 'Landlocked countries',
 4: 'Outdoor workers',
 5: 'Riverine and flood-prone areas',
 6: 'Small-scale farmers',
 7: 'Men and boys',
 8: 'Small island developing states (SIDS)',
 9: 'Fisherfolk and fishing communities',
 10: 'Children with disabilities',
 11: 'Low-income households',
 12: 'Rural communities',
 13: 'Pregnant women and new mothers',
 14: 'Young adults',
 15: 'Urban slums',
 16: 'Gender non-conforming individuals',
 17: 'Remote communities',
 18: 'Older adults and the elderly',
 19: 'Elderly population',
 20: 'Mountain communities',
 21: 'People with disabilities',
 22: 'Indigenous peoples',
 23: 'Informal settlements and slums',
 24: 'Coastal communities',
 25: 'Informal sector workers',
 26: 'Drought-prone regions',
 27: 'People with pre-existing health conditions',
 28: 'Small-scale farmers and subsistence agriculture',
 29: 'Migrants and displaced populations'}


def predict(text):
    preds = model([text])[0].item()
    return group_dict[preds]

gradio_ui = gr.Interface(
    fn=predict,
    title="Predict reference to vulnerable groups",
    description="This Space showcases.",
    inputs=[
        gr.inputs.Textbox(lines=5, label="Paste some text here"),
    ],
    outputs=[
        gr.outputs.Textbox(label="Vulnerable group"),
   
    ],
    examples=[
        ["To promote gender equality and empower men and boys as agents of change in climate adaptation efforts, we aim to engage 300,000 men and boys in gender-responsive climate change adaptation initiatives by 2030, focusing on capacity building, leadership development, and community engagement."], 
        ["To preserve the traditional knowledge and resources of indigenous peoples in the face of climate change, we commit to protecting and restoring 2 million hectares of indigenous peoples' traditional lands by 2030, focusing on sustainable land management practices, ecosystem restoration, and the conservation of biodiversity"],
        #["Through the Paris Agreement, Parties to the United Nations Framework Convention on Climate Change (UNFCCC) have agreed to limit the increase in the global average temperature to well below 2°C above pre-industrial levels, and pursue efforts to limit the temperature increase to 1.5°C above pre-industrial levels."]
    ],
)

gradio_ui.launch(debug=True)