Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from PIL import Image | |
| import hopsworks | |
| import threading | |
| import time | |
| # IDE Help | |
| from hopsworks.core.dataset_api import DatasetApi | |
| # Monitor | |
| dir_wine_saves = "latest_wine" | |
| file_dataframe_save = "df_recent.png" | |
| # Images | |
| hopsworks_images_location = f"Resources/images/{dir_wine_saves}" | |
| hopsworks_images = { | |
| "df_recent": | |
| { | |
| "name": file_dataframe_save, | |
| "local_path": "" | |
| } | |
| } | |
| last_refresh_time = None | |
| print("Logging in to Hopsworks...") | |
| project = hopsworks.login() | |
| print("Getting feature store...") | |
| fs = project.get_feature_store() | |
| print("Get database handler from Hopsworks...") | |
| dataset_api: DatasetApi = project.get_dataset_api() | |
| def download_and_refresh(): | |
| global last_refresh_time, dataset_api | |
| refreshed_all = False | |
| error = [] | |
| while not refreshed_all: | |
| error_during_exec = False | |
| for image in hopsworks_images: | |
| # Only execute when no error has currently occurred (so we don't end up in an infinite loop) | |
| if not error_during_exec and len(error) > 0 and error['name'] != hopsworks_images[image]['name']: | |
| print(f"Skipping {hopsworks_images[image]['name']} to only retry failed images.") | |
| continue | |
| elif not error_during_exec: | |
| # Pop current image | |
| print(f"Retrying to get {image}...") | |
| error = [cur_error for cur_error in error if cur_error['name'] != image['name']] | |
| try: | |
| print(f"Downloading {hopsworks_images[image]['name']} from Hopsworks...") | |
| hopsworks_images[image]['local_path'] = dataset_api.download( | |
| f"{hopsworks_images_location}/{hopsworks_images[image]['name']}", | |
| overwrite=True) | |
| print(f"Saved in: {hopsworks_images[image]['local_path']}") | |
| last_refresh_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) | |
| print(f"Image refreshed at {last_refresh_time}") | |
| except Exception as e: | |
| print(f"An error occurred when trying to refresh {hopsworks_images[image]['name']}: {e}") | |
| error_during_exec = True | |
| # Make sure images are not added twice if re-added when retrying to download | |
| error = [cur_error for cur_error in error if cur_error['name'] != hopsworks_images[image]['name']] | |
| error.append(image) | |
| if error_during_exec: | |
| print(f"Will retry fetching failed image(s) in 1 minute...") | |
| time.sleep(60) | |
| else: | |
| print(f"Done getting all images.") | |
| refreshed_all = True | |
| def download_and_refresh_scheduled(): | |
| while True: | |
| print(f"Refreshing in 12 hours...") | |
| # Sleep for 12 hours (12 hours * 60 minutes * 60 seconds) | |
| time.sleep(43200) | |
| download_and_refresh() | |
| def get_label_text(): | |
| if last_refresh_time: | |
| return f"Recent Prediction History (Last Refreshed: {last_refresh_time})" | |
| else: | |
| return "Recent Prediction History" | |
| # Get all once | |
| print("Getting all images...") | |
| download_and_refresh() | |
| thread = threading.Thread(target=download_and_refresh_scheduled) | |
| thread.daemon = True | |
| thread.start() | |
| print("Configuring gradio...") | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| with gr.Column(): | |
| label1 = gr.Label(get_label_text()) | |
| image1 = gr.Image(hopsworks_images['df_recent']['local_path'], | |
| elem_id="recent-predictions", type="pil") | |
| print("Launching gradio...") | |
| demo.launch(debug=True) | |