Update app.py

app.py

@@ -247,29 +247,30 @@ body, .gradio-container {
 
 
 with gr.Blocks(css=custom_css) as demo:
-    # Initialize a placeholder for the loaded space
-    space_placeholder = gr.HTML("Loading space...")
+    # Initialize the HTML output with a default message
+    html_output = gr.HTML("Processing data...")
     
-    # Define the refresh function to call load_space()
+    # Define the refresh function
     def refresh_html():
-        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)}"
+        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..."
     
-    # 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
-    )
+    # Use the Timer component according to the documentation
+    timer = gr.Timer(3600, refresh_html, [], html_output)
+
+
+
+
+
     
     # Initial call to set the HTML content when the page loads
-    demo.load(refresh_html, inputs=None, outputs=space_placeholder)
+    demo.load(refresh_html, [], html_output)
 
 # Launch the demo
 demo.launch()