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_prediction.gif")
dataset_api.download("resources/images/correct_prediction.gif")
dataset_api.download("resources/images/df_recent.png")
# dataset_api.download("resources/images/confusion_matrix.png")

with gr.Blocks() as demo:
    with gr.Row():
      with gr.Column():
          gr.Label("Today's Predicted Image")
          input_img = gr.Image("latest_prediction.gif", elem_id="predicted-img")
      with gr.Column():          
          gr.Label("Today's Actual Image")
          input_img = gr.Image("correct_prediction.gif", elem_id="actual-img")        
    with gr.Row():
      with gr.Column():
          gr.Label("Recent Prediction History")
          input_img = gr.Image("df_recent.png", elem_id="recent-predictions")
    #   with gr.Column():          
    #       gr.Label("Confusion Maxtrix with Historical Prediction Performance")
    #       input_img = gr.Image("confusion_matrix.png", elem_id="confusion-matrix")        

demo.launch()