import gradio as gr
from PIL import Image
import hopsworks

project = hopsworks.login()
fs = project.get_feature_store()

dataset_api = project.get_dataset_api()

dataset_api.download("Resources/images/latest_wine.png")
dataset_api.download("Resources/images/actual_wine.png")
dataset_api.download("Resources/images/df_wine_recent.png")
dataset_api.download("Resources/images/confusion_wine_matrix.png")

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            gr.Label("Note: wine quality is given on a scale of 1-3 glasses.")
    with gr.Row():
        with gr.Column():
            gr.Label("Today's Predicted Quality (1-3)")
            input_img = gr.Image("latest_wine.png", elem_id="predicted-img")
        with gr.Column():
            gr.Label("Today's Actual Quality (1-3)")
            input_img = gr.Image("actual_wine.png", elem_id="actual-img")
    with gr.Row():
        with gr.Column():
            gr.Label("Recent Prediction History")
            input_img = gr.Image("df_wine_recent.png", elem_id="recent-predictions")
        with gr.Column():
            gr.Label("Confusion Maxtrix with Historical Prediction Performance")
            input_img = gr.Image(
                "confusion_wine_matrix.png", elem_id="confusion-matrix"
            )

demo.launch()