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_red.png")
dataset_api.download("Resources/images/actual_wine_red.png")
dataset_api.download("Resources/images/df_recent_wine_red.png")
dataset_api.download("Resources/images/confusion_matrix_wine_red.png")
dataset_api.download("Resources/images/latest_wine_white.png")
dataset_api.download("Resources/images/actual_wine_white.png")
dataset_api.download("Resources/images/df_recent_wine_white.png")
dataset_api.download("Resources/images/confusion_matrix_wine_white.png")

with gr.Blocks() as demo:
    with gr.Row():
      with gr.Column():
          gr.Label("Today's Red Predicted Quality")
          input_img = gr.Image("latest_wine_red.png", elem_id="predicted-img-red")
      with gr.Column():          
          gr.Label("Today's Red Actual Quality")
          input_img = gr.Image("actual_wine_red.png", elem_id="actual-img-red")
      with gr.Column():          
          gr.Label("Today's White Predicted Quality")
          input_img = gr.Image("actual_wine_white.png", elem_id="actual-img-white")
      with gr.Column():          
          gr.Label("Today's White Actual Quality")
          input_img = gr.Image("actual_wine_white.png", elem_id="actual-img-white")        
    with gr.Row():
      with gr.Column():
          gr.Label("Recent Red Prediction History")
          input_img = gr.Image("df_recent_wine_red.png", elem_id="recent-predictions-red")
      with gr.Column():          
          gr.Label("Confusion Maxtrix with Historical Red Prediction Performance")
          input_img = gr.Image("confusion_matrix_wine_red.png", elem_id="confusion-matrix-red")
      with gr.Column():
          gr.Label("Recent White Prediction History")
          input_img = gr.Image("df_recent_wine_white.png", elem_id="recent-predictions-white")
      with gr.Column():          
          gr.Label("Confusion Maxtrix with Historical White Prediction Performance")
          input_img = gr.Image("confusion_matrix_wine_white.png", elem_id="confusion-matrix-white")        

demo.launch()