Update app.py

app.py

@@ -247,25 +247,29 @@ body, .gradio-container {
 
 
 with gr.Blocks(css=custom_css) as demo:
-    # Initialize the HTML output with a default message
-    html_output = gr.HTML("Processing data...")
+    # Initialize a placeholder for the loaded space
+    space_placeholder = gr.HTML("Loading space...")
     
-    # Define the refresh function
+    # Define the refresh function to call load_space()
     def refresh_html():
-        if prediction_plot_base64:
-            return (
-                f'<img src="data:image/png;base64,{prediction_plot_base64}" '
-                'alt="Prediction Plot" '
-                'style="width: 100vw; height: 100vh; object-fit: contain;">'
-            )
-        else:
-            return "Processing data..."
+        try:
+            # Load the space
+            space = load_space()
+            # Return the rendered HTML of the space
+            return space.render()
+        except Exception as e:
+            return f"Error loading space: {str(e)}"
     
-    # Use the Timer component according to the documentation
-    timer = gr.Timer(3600, refresh_html, [], html_output)
+    # Use the Timer component to refresh the HTML every hour (3600 seconds)
+    timer = gr.Timer(
+        every=3600,  # Interval in seconds (1 hour)
+        fn=refresh_html,  # Function to call
+        inputs=None,  # No inputs
+        outputs=space_placeholder,  # Output to update
+    )
     
     # Initial call to set the HTML content when the page loads
-    demo.load(refresh_html, [], html_output)
+    demo.load(refresh_html, inputs=None, outputs=space_placeholder)
 
 # Launch the demo
 demo.launch()