Update app.py

app.py

@@ -292,30 +292,26 @@ body, .gradio-container {
 """
 
 
-# Create a function to update the HTML in a separate thread
-def update_html_periodically(html_component):
-    while True:
+with gr.Blocks(css=custom_css) as demo:
+    # Initialize the HTML output with a default message
+    html_output = gr.HTML("Processing data...")
+    
+    # Define the refresh function
+    def refresh_html():
         if prediction_plot_base64:
-            new_html = (
+            return (
                 f'<img src="data:image/png;base64,{prediction_plot_base64}" '
                 'alt="Prediction Plot" '
                 'style="width: 100vw; height: 100vh; object-fit: contain;">'
             )
         else:
-            new_html = "Processing data..."
-        
-        # Update the HTML component
-        html_component.update(value=new_html)
-        
-        # Wait 10 seconds
-        time.sleep(10)
-
-with gr.Blocks(css=custom_css) as demo:
-    # Initialize the HTML output with a default message
-    html_output = gr.HTML("Processing data...")
+            return "Processing data..."
+    
+    # Use the Timer component according to the documentation
+    timer = gr.Timer(10, refresh_html, [], html_output)
     
-    # Start the update thread when the demo loads
-    demo.load(lambda: threading.Thread(target=update_html_periodically, args=(html_output,), daemon=True).start())
+    # Initial call to set the HTML content when the page loads
+    demo.load(refresh_html, [], html_output)
 
 # Launch the demo
 demo.launch()