Update app.py

app.py

@@ -151,7 +151,20 @@ def create_reranking_interface(task_data):
             ranking_inputs = []
             validation_indicators = []
             
-            # Create a better visual layout for documents
+            # Add back the quick ranking tools with improved styling
+            # Add a nicely formatted header for the ranking tools
+            with gr.Box(elem_classes=["tool-container"]):
+                with gr.Row(equal_height=True):
+                    with gr.Column(scale=3):
+                        gr.Markdown("### Quick Ranking Tools", elem_classes=["section-header"])
+                    with gr.Column(scale=2):
+                        sequential_btn = gr.Button("Rank in Order (1,2,3...)", variant="secondary", size="sm")
+                    with gr.Column(scale=2):
+                        reverse_btn = gr.Button("Reverse Order (n,n-1,...)", variant="secondary", size="sm")
+                    with gr.Column(scale=2):
+                        clear_btn = gr.Button("Clear All Rankings", variant="secondary", size="sm")
+            
+            # Adjust the document display for more reasonable textbox size
             with gr.Box():
                 for i, doc in enumerate(samples[0]["candidates"]):
                     row_class = "document-row-even" if i % 2 == 0 else "document-row-odd"
@@ -165,7 +178,7 @@ def create_reranking_interface(task_data):
                                 label=f"Document {i+1}",
                                 interactive=False,
                                 elem_classes=["document-text"],
-                                lines=6,  # Increase number of visible lines
+                                lines=4,  # Reduce to 4 visible lines
                             )
                             doc_containers.append(doc_box)
                         
@@ -217,9 +230,9 @@ def create_reranking_interface(task_data):
                 font-size: 16px !important;
                 line-height: 1.6 !important;
                 padding: 10px !important;
-                min-height: 150px !important;  /* Ensure minimum height */
-                height: auto !important;       /* Allow expansion if needed */
-                overflow-y: visible !important; /* Don't hide overflow */
+                min-height: 100px !important;  /* Reduce minimum height */
+                height: auto !important;       
+                overflow-y: visible !important;
             }
             
             .rank-dropdown select {
@@ -236,11 +249,15 @@ def create_reranking_interface(task_data):
                 box-shadow: 0 0 0 0.2rem rgba(0, 123, 255, 0.25) !important;
             }
             
-            .tool-button button {
-                min-width: 120px !important;
+            .tool-container {
+                margin-bottom: 20px !important;
+                background-color: #f8f9fa !important;
+                border-left: 4px solid #6c757d !important;
+                padding: 10px !important;
             }
             
-            .tool-heading {
+            .section-header {
+                margin: 0 !important;
                 padding-top: 8px !important;
             }
             
@@ -284,6 +301,75 @@ def create_reranking_interface(task_data):
             
             return results + [all_valid]  # Return validation indicators and validity flag
         
+        def on_ranking_change(*rankings):
+            """Validate rankings whenever any ranking dropdown changes."""
+            validation_results = validate_rankings(*rankings)
+            return validation_results[:-1]  # Return only the validation indicators
+        
+        def submit_rankings(*args):
+            """Submit rankings with improved validation and user feedback."""
+            # Get the last argument (sample_id) and the rankings
+            if len(args) < 1:
+                return "Error: No arguments provided", progress_text.value
+            
+            # Verify we have enough rankings
+            if len(args) < len(ranking_inputs) + 1:
+                return "Error: Not enough ranking inputs provided", progress_text.value
+            
+            sample_id = args[-1]
+            rankings = args[:len(ranking_inputs)]
+            
+            # First validate the rankings
+            validation_results = validate_rankings(*rankings)
+            all_valid = validation_results[-1]  # Last item is validity flag
+            validation_indicators_values = validation_results[:-1]  # Remove validity flag
+            
+            # Update validation indicators
+            for i, result in enumerate(validation_indicators_values):
+                if i < len(validation_indicators):
+                    validation_indicators[i].update(value=result)
+            
+            # Check for duplicate rankings
+            if all_valid:
+                try:
+                    processed_rankings = [int(r) for r in rankings]
+                    if len(set(processed_rankings)) != len(processed_rankings):
+                        dup_ranks = {}
+                        for i, r in enumerate(processed_rankings):
+                            if r in dup_ranks:
+                                dup_ranks[r].append(i)
+                            else:
+                                dup_ranks[r] = [i]
+                        
+                        # Highlight duplicates with error styling
+                        for rank, indices in dup_ranks.items():
+                            if len(indices) > 1:
+                                for idx in indices:
+                                    if idx < len(validation_indicators):
+                                        validation_indicators[idx].update(
+                                            value=f'<span style="color: #dc3545; font-weight: bold;">⚠️ Duplicate rank {rank}</span>'
+                                        )
+                        
+                        return "⚠️ Each document must have a unique rank. Please fix duplicate rankings.", progress_text.value
+                except:
+                    pass
+            
+            # If not all valid, return error message
+            if not all_valid:
+                return "⚠️ Please assign a rank to all documents before submitting", progress_text.value
+            
+            # Save the validated rankings
+            status, progress = save_ranking(rankings, sample_id)
+            
+            # Provide clear success feedback
+            if "✅" in status:
+                for i in range(len(validation_indicators)):
+                    validation_indicators[i].update(
+                        value=f'<span style="color: #28a745; font-weight: bold;">✓ Saved</span>'
+                    )
+            
+            return status, progress
+        
         def load_sample(sample_id):
             """Load a specific sample into the interface."""
             sample = next((s for s in samples if s["id"] == sample_id), None)
@@ -383,120 +469,35 @@ def create_reranking_interface(task_data):
             except Exception as e:
                 return f"Error saving results: {str(e)}"
         
-        # Function to assign sequential ranks
+        # Define functions for the quick ranking buttons
         def assign_sequential_ranks():
             return [str(i+1) for i in range(len(samples[0]["candidates"]))]
         
-        # Function to assign reverse ranks
         def assign_reverse_ranks():
             n = len(samples[0]["candidates"])
             return [str(n-i) for i in range(n)]
         
-        # Function to clear all rankings
         def clear_rankings():
             return [""] * len(samples[0]["candidates"])
         
-        # Define a function that collects all ranking values and validates them
-        def submit_rankings(*args):
-            """Submit rankings with improved validation and user feedback."""
-            # Get the last argument (sample_id) and the rankings
-            if len(args) < 1:
-                return "Error: No arguments provided", progress_text.value
-            
-            # Verify we have enough rankings
-            if len(args) < len(ranking_inputs) + 1:
-                return "Error: Not enough ranking inputs provided", progress_text.value
-            
-            sample_id = args[-1]
-            rankings = args[:len(ranking_inputs)]
-            
-            # First validate the rankings
-            validation_results = validate_rankings(*rankings)
-            all_valid = validation_results[-1]  # Last item is validity flag
-            validation_indicators_values = validation_results[:-1]  # Remove validity flag
-            
-            # Update validation indicators
-            for i, result in enumerate(validation_indicators_values):
-                if i < len(validation_indicators):
-                    validation_indicators[i].update(value=result)
-            
-            # Check for duplicate rankings
-            if all_valid:
-                try:
-                    processed_rankings = [int(r) for r in rankings]
-                    if len(set(processed_rankings)) != len(processed_rankings):
-                        dup_ranks = {}
-                        for i, r in enumerate(processed_rankings):
-                            if r in dup_ranks:
-                                dup_ranks[r].append(i)
-                            else:
-                                dup_ranks[r] = [i]
-                        
-                        # Highlight duplicates with error styling
-                        for rank, indices in dup_ranks.items():
-                            if len(indices) > 1:
-                                for idx in indices:
-                                    if idx < len(validation_indicators):
-                                        validation_indicators[idx].update(
-                                            value=f'<span style="color: #dc3545; font-weight: bold;">⚠️ Duplicate rank {rank}</span>'
-                                        )
-                        
-                        return "⚠️ Each document must have a unique rank. Please fix duplicate rankings.", progress_text.value
-                except:
-                    pass
-            
-            # If not all valid, return error message
-            if not all_valid:
-                return "⚠️ Please assign a rank to all documents before submitting", progress_text.value
-            
-            # Save the validated rankings
-            status, progress = save_ranking(rankings, sample_id)
-            
-            # Provide clear success feedback
-            if "✅" in status:
-                for i in range(len(validation_indicators)):
-                    validation_indicators[i].update(
-                        value=f'<span style="color: #28a745; font-weight: bold;">✓ Saved</span>'
-                    )
-            
-            return status, progress
+        # Connect quick ranking buttons
+        sequential_btn.click(
+            fn=assign_sequential_ranks,
+            inputs=None,
+            outputs=ranking_inputs
+        )
         
-        # Update ranking input's event handling for immediate validation
-        def on_ranking_change(*rankings):
-            """Validate rankings whenever any ranking dropdown changes."""
-            validation_results = validate_rankings(*rankings)
-            return validation_results[:-1]  # Return only the validation indicators
+        reverse_btn.click(
+            fn=assign_reverse_ranks,
+            inputs=None,
+            outputs=ranking_inputs
+        )
         
-        # Check for overlapping ranks and duplicate assignments
-        def check_for_duplicates(*rankings):
-            """Highlight duplicate rankings with visual feedback."""
-            clean_rankings = []
-            for r in rankings:
-                if r is not None and r != "":
-                    clean_rankings.append(int(r))
-            
-            if len(clean_rankings) != len(set(clean_rankings)):
-                used_ranks = {}
-                for i, r in enumerate(rankings):
-                    if r is not None and r != "":
-                        rank = int(r)
-                        if rank in used_ranks:
-                            used_ranks[rank].append(i)
-                        else:
-                            used_ranks[rank] = [i]
-            
-            results = []
-            for i, r in enumerate(rankings):
-                if r is not None and r != "":
-                    rank = int(r)
-                    if len(used_ranks[rank]) > 1:
-                        results.append(f'<span style="color: #dc3545; font-weight: bold;">⚠️ Duplicate rank {rank}</span>')
-                    else:
-                        results.append(f'<span style="color: #28a745; font-weight: bold;">✓ Rank {rank}</span>')
-                else:
-                    results.append('<span style="color: #dc3545; font-weight: bold;">⚠️ Missing</span>')
-            
-            return results
+        clear_btn.click(
+            fn=clear_rankings,
+            inputs=None,
+            outputs=ranking_inputs
+        )
         
         # Wire up events (Gradio 3.x syntax)
         submit_btn.click(
@@ -557,7 +558,7 @@ def create_reranking_interface(task_data):
         )
         
         # Connect validation to ranking inputs for real-time feedback
-        for i, ranking in enumerate(ranking_inputs):
+        for ranking in ranking_inputs:
             ranking.change(
                 fn=on_ranking_change,
                 inputs=ranking_inputs,