Update app.py

app.py

@@ -98,13 +98,30 @@ def create_reranking_interface(task_data):
                             label=f"Document {i+1}",
                             interactive=False
                         )
-                        dropdown = gr.Dropdown(
-                            choices=[str(j) for j in range(1, len(samples[0]["candidates"])+1)],
-                            label=f"Rank",
-                            value=""
+                        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
+                            )
+                            with gr.Row():
+                                up_btn = gr.Button("↑", size="sm")
+                                down_btn = gr.Button("↓", size="sm")
+                        
+                        # Connect up/down buttons to increase/decrease rank
+                        up_btn.click(
+                            lambda x: max(1, x-1),  # Lower number = higher rank
+                            inputs=[rank_number],
+                            outputs=[rank_number]
+                        )
+                        down_btn.click(
+                            lambda x: min(len(samples[0]["candidates"]), x+1),
+                            inputs=[rank_number],
+                            outputs=[rank_number]
                         )
-                        doc_containers.append(doc_box)
-                        ranking_dropdowns.append(dropdown)
             
             with gr.Row():
                 prev_btn = gr.Button("← Previous Query", size="sm")