import gradio as gr
from covid import Covid
import pandas as pd

# Fonction pour obtenir les données COVID-19 d'un pays
def get_covid_data(country):
    covid = Covid()
    data = covid.get_status_by_country_name(country)
    return data

# Liste de tous les pays disponibles
covid = Covid()
countries = covid.list_countries()

# Interface Gradio avec une liste déroulante
iface = gr.Interface(fn=get_covid_data, 
                     inputs=gr.inputs.Dropdown(countries, label="Select a country"), 
                     outputs=gr.outputs.HTML(),
                     title="COVID-19 Data by Country",
                     description="Select a country to get COVID-19 data.")

# Fonction pour afficher les données sous forme de tableau
def display_table(data):
    df = pd.DataFrame.from_dict(data, orient='index')
    df = df.rename(columns={
        'confirmed': 'Confirmed Cases',
        'active': 'Active Cases',
        'deaths': 'Deaths',
        'recovered': 'Recovered Cases'
    })
    table_html = df.to_html()
    return table_html

# Ajouter la fonction display_table à l'interface Gradio
iface.output(display_table)

# Lancement de l'interface
iface.launch()