Add manual clean up of the resources

app.py

@@ -4,7 +4,6 @@ import pandas as pd
 import os
 import threading
 import time
-from pathlib import Path
 from utils.data_loader import get_random_example
 from utils.models import generate_summaries, model_names
 from utils.ui_helpers import toggle_context_display, update_feedback, get_context_html
@@ -97,14 +96,13 @@ def process_generation_result(result):
             "", "", "", "", None, [], False, load_leaderboard_data(),
             gr.update(value="Generation was interrupted or failed. Please try again."),
             gr.update(value="Generation was interrupted or failed. Please try again."),
-            # Keep voting buttons disabled when generation fails or is interrupted
-            gr.update(interactive=False, elem_classes=["vote-button"]),
-            gr.update(interactive=False, elem_classes=["vote-button"]),
-            gr.update(interactive=False, elem_classes=["vote-button"]),
-            gr.update(interactive=False, elem_classes=["vote-button", "vote-button-neither"]),
+            gr.update(interactive=True, elem_classes=["vote-button"]),
+            gr.update(interactive=True, elem_classes=["vote-button"]),
+            gr.update(interactive=True, elem_classes=["vote-button"]),
+            gr.update(interactive=True, elem_classes=["vote-button", "vote-button-neither"]),
             gr.update(choices=[], value=[], interactive=False, visible=False),
             gr.update(visible=False),
-            gr.update(interactive=True, visible=True),
+            gr.update(interactive=False, visible=True),
             gr.update(visible=False),
             gr.update(interactive=True),
             gr.update(elem_classes=[])
@@ -118,7 +116,6 @@ def process_generation_result(result):
         None, [], False, agg_results,
         gr.update(value=result["summary_a"]),
         gr.update(value=result["summary_b"]),
-        # Enable voting buttons only when both summaries are ready
         gr.update(interactive=True, elem_classes=["vote-button"]),
         gr.update(interactive=True, elem_classes=["vote-button"]),
         gr.update(interactive=True, elem_classes=["vote-button"]),
@@ -181,7 +178,6 @@ def show_loading_state():
     return [
         gr.update(value="Loading new question and summaries...", interactive=False),
         gr.update(value="Loading new question and summaries...", interactive=False),
-        # Disable voting buttons during loading
         gr.update(interactive=False),
         gr.update(interactive=False),
         gr.update(interactive=False),
@@ -203,6 +199,12 @@ def update_ui_for_new_context(example):
         False
     ]
 
+# Resource cleanup function for unload event
+def cleanup_on_disconnect():
+    """Clean up resources when browser disconnects"""
+    print(f"Browser disconnected. Cleaning up resources...")
+    generation_interrupt.set()
+
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Default(
     primary_hue=gr.themes.colors.orange,
@@ -210,10 +212,25 @@ with gr.Blocks(theme=gr.themes.Default(
 )) as demo:
     # Load CSS
     css_path = os.path.join(os.getcwd(), 'static', 'styles.css')
+    
+    # Load the files
     with open(css_path, 'r') as f:
         css_content = f.read()
     
+    # Create HTML components with CSS and JavaScript links
     gr.HTML(f"<style>{css_content}</style>")
+    
+    # Add JavaScript to handle browser unload events
+    unload_js = """
+    <script>
+    // This runs when the page is about to be closed or refreshed
+    window.addEventListener('beforeunload', function(e) {
+        // Send a synchronous request to the server
+        navigator.sendBeacon('/cleanup?session_id=' + window.gradioClientState.session_hash);
+    });
+    </script>
+    """
+    gr.HTML(unload_js)
 
     # State Variables
     current_example = gr.State({})
@@ -292,11 +309,10 @@ with gr.Blocks(theme=gr.themes.Default(
                 # Voting section
                 gr.Markdown("### 🏅 Cast Your Vote", elem_classes="section-heading")
                 with gr.Row():
-                    # Start with voting buttons disabled
-                    vote_button_a = gr.Button("⬅️ Summary A is Better", elem_classes=["vote-button"], interactive=False)
-                    vote_button_tie = gr.Button("🤝 Tie / Equally Good", elem_classes=["vote-button"], interactive=False)
-                    vote_button_b = gr.Button("➡️ Summary B is Better", elem_classes=["vote-button"], interactive=False)
-                    vote_button_neither = gr.Button("❌ Neither is Good", elem_classes=["vote-button", "vote-button-neither"], interactive=False)
+                    vote_button_a = gr.Button("⬅️ Summary A is Better", elem_classes=["vote-button"])
+                    vote_button_tie = gr.Button("🤝 Tie / Equally Good", elem_classes=["vote-button"])
+                    vote_button_b = gr.Button("➡️ Summary B is Better", elem_classes=["vote-button"])
+                    vote_button_neither = gr.Button("❌ Neither is Good", elem_classes=["vote-button", "vote-button-neither"])
 
                 # Feedback and Submit sections
                 with gr.Group(elem_classes=["feedback-section"], visible=False) as feedback_section:
@@ -340,6 +356,7 @@ The Elo rating system provides a more accurate ranking than simple win rates:
             
             results_table_display = gr.HTML(label="Model Performance")
 
+    # Event handling
     # Toggle context display
     context_toggle_btn.click(
         fn=toggle_context_display,
@@ -434,6 +451,9 @@ The Elo rating system provides a more accurate ranking than simple win rates:
         outputs=[results_table_display],
         api_name="refresh_leaderboard"
     )
+    
+    # Register unload event for browser disconnections
+    demo.unload(cleanup_on_disconnect)
 
 if __name__ == "__main__":
     demo.launch(debug=True)