import tensorflow as tf
import numpy as np
import gradio as gr

# Carregue o modelo treinado com catarata
loaded_model = tf.keras.models.load_model('modelo_treinado.h5')

# Crie uma lista de classes
class_names = ["Normal", "Cataract"]

# Defina a função de classificação
def classify_image(inp):
    # Reshape a imagem de entrada para o formato esperado pelo modelo
    img = inp.reshape((-1, 224, 224, 3))

    # Faça uma previsão usando o modelo treinado
    prediction = loaded_model.predict(img).flatten()

    # Retorna a classe prevista
    return class_names[np.argmax(prediction)]

# Crie uma interface Gradio
iface = gr.Interface(
    fn=classify_image,
    inputs=gr.inputs.Image(shape=(224, 224)),
    outputs=gr.outputs.Label(),
    capture_session=True
)

# Inicie a interface Gradio
iface.launch(debug=True, share=True)