Update offline warning

src/model_demo.py

@@ -13,14 +13,14 @@ from wandb_data import TIMEZONE
 
 SERVER_API = os.environ["SERVER_API"]
 
-current_model: str
+current_model: str | None = None
 last_current_model_sync: datetime = datetime.fromtimestamp(0, TIMEZONE)
 
 def get_current_model() -> str | None:
     global current_model
     global last_current_model_sync
     now = datetime.now(TIMEZONE)
-    if now - last_current_model_sync < timedelta(minutes=1):
+    if now - last_current_model_sync < timedelta(minutes=5):
         return current_model
     last_current_model_sync = now
 
@@ -66,34 +66,37 @@ def submit(prompt: str, seed: int | str | None, baseline: bool) -> tuple:
 
 
 def create_demo():
-    with gr.Accordion(f"EdgeMaxxing Model Comparison"):
-        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)
+    offline_textbox = gr.Textbox("The server is offline! Please come back later", interactive=False, show_label=False)
+    with gr.Group(visible=get_current_model() is not None):
+        with gr.Group():
             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])
+                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():
+            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,
+            )
+
+        offline_textbox.attach_load_event(lambda: gr.Textbox(visible=get_current_model() is None), None)
+        prompt.attach_load_event(lambda: gr.Textbox(visible=get_current_model() is not None), None)
+        seed_input.attach_load_event(lambda: gr.Textbox(visible=get_current_model() is not None), None)
+
+        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])