Add server offline warning

src/chain_data.py

@@ -5,14 +5,14 @@ from enum import Enum
 from math import ceil
 from typing import TypeAlias
 
+from fiber.chain.commitments import _deserialize_commitment_field
 from fiber.chain.interface import get_substrate
 from fiber.chain.metagraph import Metagraph
 from fiber.chain.models import Node
-from fiber.chain.commitments import _deserialize_commitment_field
 from substrateinterface.storage import StorageKey
 
 from network_commitments import Decoder
-from wandb_data import Hotkey, Uid, TIMEZONE
+from wandb_data import TIMEZONE, Hotkey, Uid
 
 Weight: TypeAlias = float
 Incentive: TypeAlias = float

src/model_demo.py

@@ -2,14 +2,38 @@ import base64
 import os
 import random
 import time
+from datetime import datetime, timedelta
 from io import BytesIO
 
 import gradio as gr
 import requests
 from PIL import Image
 
+from wandb_data import TIMEZONE
+
 SERVER_API = os.environ["SERVER_API"]
 
+current_model: str
+last_current_model_sync: datetime = datetime.fromtimestamp(0, TIMEZONE)
+
+def get_current_model() -> str | None:
+    try:
+        global current_model
+        global last_current_model_sync
+        now = datetime.now(TIMEZONE)
+        if now - last_current_model_sync < timedelta(minutes=5):
+            return current_model
+        last_current_model_sync = now
+
+        response = requests.get(f"{SERVER_API}/model", verify=False)
+        response.raise_for_status()
+        model = response.json()
+        current_model = f"{model['uid']} - {model['url']}"
+        return current_model
+    except:
+        return None
+
+
 
 def image_from_base64(image_data: str) -> Image:
     image_buffer = BytesIO(base64.b64decode(image_data))
@@ -39,40 +63,35 @@ def submit(prompt: str, seed: int | str | None, baseline: bool) -> tuple:
     )
 
 
-def create_textbox() -> gr.Textbox:
-    response = requests.get(f"{SERVER_API}/model", verify=False)
-    response.raise_for_status()
-    model = response.json()
-    return gr.Textbox(f"{model['uid']} - {model['url']}", interactive=False, show_label=False)
-
-
 def create_demo():
     with gr.Accordion(f"EdgeMaxxing Model Comparison"):
-        with gr.Group():
+        gr.Textbox("The server is offline! Please come back later", interactive=False, show_label=False, visible=get_current_model() is None)
+        with gr.Group(visible=get_current_model() is not None):
+            with gr.Group():
+                with gr.Row():
+                    with gr.Column():
+                        gr.Textbox("Baseline", interactive=False, show_label=False)
+                        baseline_image_component = gr.Image(show_label=False)
+
+                    with gr.Column():
+                        textbox = gr.Textbox(interactive=False, show_label=False)
+                        textbox.attach_load_event(lambda: get_current_model(), None)
+                        optimized_image_component = gr.Image(show_label=False)
             with gr.Row():
-                with gr.Column():
-                    gr.Textbox("Baseline", interactive=False, show_label=False)
-                    baseline_image_component = gr.Image(show_label=False)
-
-                with gr.Column():
-                    textbox = gr.Textbox()
-                    textbox.attach_load_event(lambda: create_textbox(), None)
-                    optimized_image_component = gr.Image(show_label=False)
-        with gr.Row():
-            prompt = gr.Textbox(
-                placeholder="Enter prompt...",
-                interactive=True,
-                submit_btn=True,
-                show_label=False,
-                autofocus=True,
-                scale=10,
-            )
-
-            seed_input = gr.Textbox(
-                placeholder="Enter seed...",
-                interactive=True,
-                show_label=False,
-            )
-
-    prompt.submit(lambda prompt, seed: submit(prompt, seed, True), inputs=[prompt, seed_input], outputs=[prompt, baseline_image_component])
-    prompt.submit(lambda prompt, seed: submit(prompt, seed, False), inputs=[prompt, seed_input], outputs=[prompt, optimized_image_component])
+                prompt = gr.Textbox(
+                    placeholder="Enter prompt...",
+                    interactive=True,
+                    submit_btn=True,
+                    show_label=False,
+                    autofocus=True,
+                    scale=10,
+                )
+
+                seed_input = gr.Textbox(
+                    placeholder="Enter seed...",
+                    interactive=True,
+                    show_label=False,
+                )
+
+            prompt.submit(lambda prompt, seed: submit(prompt, seed, True), inputs=[prompt, seed_input], outputs=[prompt, baseline_image_component])
+            prompt.submit(lambda prompt, seed: submit(prompt, seed, False), inputs=[prompt, seed_input], outputs=[prompt, optimized_image_component])