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/df_latest_news.png",overwrite=True)
dataset_api.download("Resources/images/df_recent_tsla_predictions.png",overwrite=True)
dataset_api.download("Resources/images/stock_price_w_pred.png",overwrite=True)

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            gr.Label("Recent Predictions")
            input_img = gr.Image("df_recent_tsla_predictions.png", elem_id="predicted-img")
        with gr.Column():
            gr.Label("Recent Stock Prices")
            input_img = gr.Image("stock_price_w_pred.png", elem_id="actual-img")
    with gr.Row():
        with gr.Column():
            gr.Label("Latest News Articles")
            input_img = gr.Image("df_latest_news.png", elem_id="actual-img")

demo.launch()