import gradio as gr
import pandas as pd
from joblib import load



def pecho(radio,suavidad,puntos_concavidad,area,compacidad,concavidad):
    model = load('tomatelo_a_pecho_regr.joblib')
    df = pd.DataFrame.from_dict(
        {
            "Radio" : [radio],
            "Suavidad" : [suavidad],
            "Puntos_concavidad" : [puntos_concavidad]
            "Area" : [area]
            "Compacidad" : [compacidad]
            "Concavidad" : [concavidad],
                                 
        }
    )
       
    pred = model.predict(df)[0]
    if pred=='M':
      predicted="Las medidas indican que se trata de un caso maligno"
    else:
      predicted="Las medidas indican que se trata de un caso beningno."
    return predicted
    
iface = gr.Interface(
    pecho,
    [
        gr.inputs.Slider(1,40,label="Radio"),
        gr.inputs.Slider(0.05,0.3,label="Variación local de las longitudes del radio"),
        gr.inputs.Slider(0.01,0.4,label="Número de las porciones cóncavas del contorno"),
        gr.inputs.Slider(100,4000,label="Área"),
        gr.inputs.Slider(0.01,0.5,label="Compacidad"),
        gr.inputs.Slider(0.01,1.5,label="Severidad de las porciones cóncavas del contorno"),
     ],

    "text",
    examples=[
        [40,True,120,80,"High","Normal",0,0,1,168,62,21],
        [35,False,150,60,"Very High","Normal",0,0,1,143,52,31],
        [60,True,160,70,"High","High",1,1,0,185,90,23],
    ],
    interpretation="default",
    title = 'Modelo de clasificación para tumores benignos y malignos de mama',
    description = 'A nivel mundial, la detección de cáncer de mama representa un reto importante para la salud de todas las mujeres. Muchas de ellas, debido a cuestiones económicas (principalmente en países con poco o nulo acceso a la salud pública) no pueden acceder a los métodos de detección confiables como mastografías. Lo llegan a detectar hasta etapas avanzadas y eso suele ser un problema porque el nivel de vida disminuye y los costos de tratamiento suben. https://saturdays.ai/2022/03/16/deteccion-tumores-malignos-cancer-de-mama-con-inteligencia-artificial/',
    theme = 'grass'
)


   
iface.launch()