Update app.py

app.py

@@ -13,7 +13,7 @@ def create_reranking_interface(task_data):
         """Save the current set of rankings."""
         try:
             # Check if all documents have rankings
-            all_ranked = all(r is not None and r != "" and r != 0 for r in rankings)
+            all_ranked = all(r is not None and r != "" for r in rankings)
             if not all_ranked:
                 return "⚠️ Please assign a rank to all documents before submitting", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
             
@@ -98,44 +98,60 @@ def create_reranking_interface(task_data):
                             label=f"Document {i+1}",
                             interactive=False
                         )
-                        doc_containers.append(doc_box)  # Add to doc_containers list
+                        doc_containers.append(doc_box)
                         
-                        with gr.Column():
-                            rank_number = gr.Number(
-                                value=0,  # Start with 0 
-                                label="Rank",
-                                minimum=1,
-                                maximum=len(samples[0]["candidates"]),
-                                step=1,
-                                precision=0  # Integer only
-                            )
-                            ranking_dropdowns.append(rank_number)  # Add to ranking_dropdowns list
-                            
-                            with gr.Row():
-                                up_btn = gr.Button("↑", size="sm")
-                                down_btn = gr.Button("↓", size="sm")
+                        # Use Dropdown instead of Number for ranking
+                        # Ranks from 1 to N (number of candidates)
+                        rank_dropdown = gr.Dropdown(
+                            choices=[str(j) for j in range(1, len(samples[0]["candidates"])+1)],
+                            label=f"Rank",
+                            value=None,
+                            interactive=True
+                        )
+                        ranking_dropdowns.append(rank_dropdown)
+                        
+                        # Add quick rank buttons
+                        with gr.Row():
+                            rank_high_btn = gr.Button("Rank High (1-3)", size="sm")
+                            rank_med_btn = gr.Button("Rank Medium (4-7)", size="sm")
+                            rank_low_btn = gr.Button("Rank Low (8+)", size="sm")
+                        
+                        # Set ranks directly instead of incrementing/decrementing
+                        # Helper functions to quickly set ranks in different ranges
+                        def set_high_rank(i):
+                            def set_rank():
+                                return ["1", "2", "3"][min(i, 2)]  # First 3 docs get ranks 1,2,3
+                            return set_rank
                         
-                        # Create proper closures for the button handlers
-                        def make_up_handler(idx):
-                            def up_handler(x):
-                                return max(1, x-1)
-                            return up_handler
+                        def set_medium_rank(i):
+                            def set_rank():
+                                base = 4
+                                return str(min(base + i % 4, len(samples[0]["candidates"])))
+                            return set_rank
                         
-                        def make_down_handler(idx, max_val):
-                            def down_handler(x):
-                                return min(max_val, x+1)
-                            return down_handler
+                        def set_low_rank(i):
+                            def set_rank():
+                                base = 8
+                                return str(min(base + i % 10, len(samples[0]["candidates"])))
+                            return set_rank
                         
-                        # Connect up/down buttons to increase/decrease rank with proper handlers
-                        up_btn.click(
-                            make_up_handler(i),
-                            inputs=[rank_number],
-                            outputs=[rank_number]
+                        # Connect rank buttons
+                        rank_high_btn.click(
+                            set_high_rank(i),
+                            inputs=[],
+                            outputs=[rank_dropdown]
                         )
-                        down_btn.click(
-                            make_down_handler(i, len(samples[0]["candidates"])),
-                            inputs=[rank_number],
-                            outputs=[rank_number]
+                        
+                        rank_med_btn.click(
+                            set_medium_rank(i),
+                            inputs=[],
+                            outputs=[rank_dropdown]
+                        )
+                        
+                        rank_low_btn.click(
+                            set_low_rank(i),
+                            inputs=[],
+                            outputs=[rank_dropdown]
                         )
             
             with gr.Row():
@@ -149,7 +165,7 @@ def create_reranking_interface(task_data):
             """Load a specific sample into the interface."""
             sample = next((s for s in samples if s["id"] == sample_id), None)
             if not sample:
-                return [query_text.value] + [d.value for d in doc_containers] + [0] * len(ranking_dropdowns) + [current_sample_id.value, progress_text.value, status_box.value]
+                return [query_text.value] + [d.value for d in doc_containers] + [None] * len(ranking_dropdowns) + [current_sample_id.value, progress_text.value, status_box.value]
             
             # Update query
             new_query = sample["query"]
@@ -161,7 +177,7 @@ def create_reranking_interface(task_data):
                     new_docs.append(doc)
                     
             # Initialize rankings
-            new_rankings = [0] * len(ranking_dropdowns)
+            new_rankings = [None] * len(ranking_dropdowns)
             
             # Check if this sample has already been annotated
             existing_annotation = next((a for a in results["annotations"] if a["sample_id"] == sample_id), None)
@@ -169,7 +185,7 @@ def create_reranking_interface(task_data):
                 # Restore previous rankings
                 for i, rank in enumerate(existing_annotation["rankings"]):
                     if i < len(new_rankings) and rank is not None:
-                        new_rankings[i] = rank
+                        new_rankings[i] = str(rank)  # Convert to string for dropdown
             
             # Update progress
             current_idx = samples.index(sample)