Update app.py

app.py

@@ -181,6 +181,16 @@ def get_stats():
         "dataset_name": DATASET_NAME
     }
 
+def update_stats_event():
+    """Update the stats UI elements"""
+    stats = get_stats()
+    return [
+        stats["conversation_count"],
+        stats["next_save"],
+        stats["last_save"],
+        stats["dataset_name"]
+    ]
+
 # Create a Stanford theme
 theme = gr.themes.Default(
     primary_hue=gr.themes.utils.colors.red,
@@ -215,6 +225,9 @@ css = """
 
 # Set up the Gradio app with Blocks for more control
 with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo:
+    # Create a timer for periodic updates
+    timer = gr.Timer(30, update_stats_event)  # Update every 30 seconds
+    
     with gr.Row():
         with gr.Column(scale=3):
             # Create the chatbot component directly
@@ -242,14 +255,6 @@ with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo
                     label="Top-P"
                 )
             
-            # Create the ChatInterface with properly formatted examples
-            # Each example is a list: [message, temperature, top_p]
-            examples_with_params = [
-                ["Tell me about Stanford University", 0.7, 0.9],
-                ["How can I learn about artificial intelligence?", 0.8, 0.95],
-                ["What's your favorite book?", 0.6, 0.85]
-            ]
-            
             # Create the ChatInterface
             chat_interface = gr.ChatInterface(
                 fn=chat_model,
@@ -258,7 +263,11 @@ with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo
                 type="messages",  # This is important for compatibility
                 title="Stanford Soft Raccoon Chat",
                 description="AI assistant powered by the Soft Raccoon language model",
-                examples=examples_with_params,
+                examples=[
+                    ["Tell me about Stanford University", 0.7, 0.9],
+                    ["How can I learn about artificial intelligence?", 0.8, 0.95],
+                    ["What's your favorite book?", 0.6, 0.85]
+                ],
                 cache_examples=True,
             )
         
@@ -284,27 +293,24 @@ with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo
         [status_output]
     )
     
-    def update_stats():
-        stats = get_stats()
-        return [
-            stats["conversation_count"],
-            stats["next_save"],
-            stats["last_save"],
-            stats["dataset_name"]
-        ]
-    
     refresh_btn.click(
-        update_stats,
+        update_stats_event,
         [],
         [convo_count, next_save, last_save_time_display, dataset_name_display]
     )
     
-    # Auto-update stats every 30 seconds
+    # Connect the timer to update the stats displays
+    timer.stream(
+        fn=lambda: None,  # No-op function
+        inputs=None,
+        outputs=None,
+    )
+    
+    # Initial update of stats on page load
     demo.load(
-        update_stats, 
-        [], 
-        [convo_count, next_save, last_save_time_display, dataset_name_display],
-        every=30  # Refresh every 30 seconds
+        update_stats_event,
+        [],
+        [convo_count, next_save, last_save_time_display, dataset_name_display]
     )
     
     # Ensure we save on shutdown
@@ -313,4 +319,5 @@ with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo
 
 # Launch the app
 if __name__ == "__main__":
+    demo.queue()  # Enable queuing for better performance
     demo.launch(share=True)