Update app.py

app.py

@@ -305,25 +305,27 @@ def create_reranking_interface(task_data):
         """)
         
         def validate_rankings(*rankings):
-            """Validate rankings and update indicators with visual cues."""
+            """Simplified validation with less HTML for better performance."""
             results = []
             all_valid = True
             for rank in rankings:
                 if rank is None or rank == "":
-                    results.append('<span style="color: #dc3545; font-weight: bold;">⚠️ Missing</span>')
+                    # Use simpler HTML with less styling for faster rendering
+                    results.append("⚠️ Missing")
                     all_valid = False
                 else:
-                    results.append('<span style="color: #28a745; font-weight: bold;">✓ Rank ' + str(rank) + '</span>')
+                    # Use simpler HTML with less styling for faster rendering
+                    results.append("✓ Rank " + str(rank))
             
             return results + [all_valid]  # Return validation indicators and validity flag
         
         def on_ranking_change(*rankings):
-            """Validate rankings whenever any ranking dropdown changes."""
+            """Simplified validation for better performance."""
             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."""
+            """Submit rankings with more efficient validation."""
             # Get the last argument (sample_id) and the rankings
             if len(args) < 1:
                 return "Error: No arguments provided", progress_text.value
@@ -340,7 +342,8 @@ def create_reranking_interface(task_data):
             all_valid = validation_results[-1]  # Last item is validity flag
             validation_indicators_values = validation_results[:-1]  # Remove validity flag
             
-            # Update validation indicators
+            # Update validation indicators - less frequently
+            # Only update if really needed
             for i, result in enumerate(validation_indicators_values):
                 if i < len(validation_indicators):
                     validation_indicators[i].update(value=result)
@@ -357,13 +360,13 @@ def create_reranking_interface(task_data):
                             else:
                                 dup_ranks[r] = [i]
                         
-                        # Highlight duplicates with error styling
+                        # Use simpler HTML for duplicate messages
                         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>'
+                                            value=f"⚠️ Duplicate {rank}"
                                         )
                         
                         return "⚠️ Each document must have a unique rank. Please fix duplicate rankings.", progress_text.value
@@ -377,11 +380,11 @@ def create_reranking_interface(task_data):
             # Save the validated rankings
             status, progress = save_ranking(rankings, sample_id)
             
-            # Provide clear success feedback
+            # Provide clear success feedback - with simpler HTML
             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>'
+                        value="✓ Saved"
                     )
             
             return status, progress
@@ -487,14 +490,22 @@ def create_reranking_interface(task_data):
         
         # Define functions for the quick ranking buttons
         def assign_sequential_ranks():
-            return [str(i+1) for i in range(len(samples[0]["candidates"]))]
+            values = [str(i+1) for i in range(len(samples[0]["candidates"]))]
+            # Skip validation until all ranks are assigned
+            return values
         
         def assign_reverse_ranks():
             n = len(samples[0]["candidates"])
-            return [str(n-i) for i in range(n)]
+            values = [str(n-i) for i in range(n)]
+            # Skip validation until all ranks are assigned
+            return values
         
         def clear_rankings():
-            return [""] * len(samples[0]["candidates"])
+            values = [""] * len(samples[0]["candidates"])
+            # Clear validation indicators when clearing rankings
+            for indicator in validation_indicators:
+                indicator.update(value="")
+            return values
         
         # Connect quick ranking buttons
         sequential_btn.click(
@@ -524,23 +535,37 @@ def create_reranking_interface(task_data):
         
         # Auto-save and navigate events
         def handle_next(current_id, auto_save, *rankings):
-            # First, handle auto-save
-            new_id, status, progress = auto_save_and_navigate("next", current_id, auto_save, *rankings)
-            # Then, load the new sample
+            # First, handle auto-save - only if needed
+            if auto_save and any(r != "" for r in rankings):
+                new_id, status, progress = auto_save_and_navigate("next", current_id, auto_save, *rankings)
+            else:
+                new_id = next_sample(current_id)
+                status, progress = "", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
+                
+            # Then, load the new sample with minimal validation
             outputs = load_sample(new_id)
-            # Add the status and progress
-            outputs[-2] = progress if status else outputs[-2]
-            outputs[-1] = status if status else outputs[-1]
+            # Update only status and progress if needed
+            if status:
+                outputs[-2] = progress
+                outputs[-1] = status
+                
             return outputs
         
         def handle_prev(current_id, auto_save, *rankings):
-            # First, handle auto-save
-            new_id, status, progress = auto_save_and_navigate("prev", current_id, auto_save, *rankings)
-            # Then, load the new sample
+            # First, handle auto-save - only if needed
+            if auto_save and any(r != "" for r in rankings):
+                new_id, status, progress = auto_save_and_navigate("prev", current_id, auto_save, *rankings)
+            else:
+                new_id = prev_sample(current_id)
+                status, progress = "", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
+                
+            # Then, load the new sample with minimal validation
             outputs = load_sample(new_id)
-            # Add the status and progress
-            outputs[-2] = progress if status else outputs[-2]
-            outputs[-1] = status if status else outputs[-1]
+            # Update only status and progress if needed
+            if status:
+                outputs[-2] = progress
+                outputs[-1] = status
+                
             return outputs
         
         # Connect navigation with Gradio 3.x syntax
@@ -573,34 +598,14 @@ def create_reranking_interface(task_data):
             outputs=[auto_save_enabled]
         )
         
-        # Connect validation to ranking inputs for real-time feedback
-        for ranking in ranking_inputs:
-            ranking.change(
-                fn=on_ranking_change,
-                inputs=ranking_inputs,
-                outputs=validation_indicators
-            )
-        
-        # Add a real-time validation for the entire set to check for duplicates
-        def validate_all_inputs(*rankings):
-            """Check all inputs for duplicate ranks and provide feedback."""
-            validation_results = validate_rankings(*rankings)
-            all_valid = validation_results[-1]
-            validation_indicators_values = validation_results[:-1]
-            
-            # Show clear button status based on validation
-            submit_status = "Ready to submit" if all_valid else "Please assign unique ranks to all documents"
-            
-            return validation_indicators_values + [submit_status]
+        # Reduce frequency of validation
+        # Only connect validation to the first ranking input to reduce event handlers
+        ranking_inputs[0].change(
+            fn=on_ranking_change,
+            inputs=ranking_inputs,
+            outputs=validation_indicators
+        )
         
-        # Connect this validation to all ranking inputs
-        for ranking in ranking_inputs:
-            ranking.change(
-                fn=validate_all_inputs,
-                inputs=ranking_inputs,
-                outputs=validation_indicators + [status_box]
-            )
-            
         # Helper function for ranking - sort documents by rankings
         def rank_by_relevance(*args):
             """Sorts the documents by their current rankings for a clearer view."""
@@ -991,7 +996,7 @@ def create_main_app():
     
     return app
 
-# Create app.
+# Create the app
 demo = create_main_app()
 
 if __name__ == "__main__":