import gradio as gr
from PIL import Image
import hopsworks
import os


project = hopsworks.login(api_key_value=os.environ['UNI_HOPSWORKS_API_KEY'])


fs = project.get_feature_store()

dataset_api = project.get_dataset_api()

dataset_api.download("Resources/images/wine_df_recent.png")
dataset_api.download("Resources/images/wine_confusion_matrix.png")

with gr.Blocks() as demo:      
    with gr.Row():
      with gr.Column():
          gr.Label("Recent Prediction History")
          input_img = gr.components.Image("wine_df_recent.png", elem_id="wine_recent-predictions")
      with gr.Column():          
          gr.Label("Confusion Maxtrix with Historical Prediction Performance")
          input_img = gr.components.Image("wine_confusion_matrix.png", elem_id="wine_confusion-matrix")        

demo.launch()