Update app.py

app.py

@@ -100,8 +100,7 @@ def create_reranking_interface(task_data):
             1. Read the query at the top
             2. Review each document carefully
             3. Assign a rank to each document (1 = most relevant, higher numbers = less relevant)
-               - Use the rank buttons for quick ranking
-               - Or use the dropdown menus for more options
+               - Use the dropdown menus to select ranks
             4. Each document must have a unique rank
             5. Click "Submit Rankings" to save rankings for the current query
             6. Use "Previous" and "Next" to navigate between queries
@@ -167,7 +166,7 @@ def create_reranking_interface(task_data):
                         with gr.Column(scale=1, min_width=50):
                             gr.HTML(f"<div class='doc-number'>{i+1}</div>")
                         
-                        with gr.Column(scale=6):
+                        with gr.Column(scale=7):
                             doc_box = gr.Textbox(
                                 value=doc, 
                                 label=f"Document {i+1}",
@@ -176,15 +175,6 @@ def create_reranking_interface(task_data):
                             )
                             doc_containers.append(doc_box)
                         
-                        # Add rank buttons for quick ranking
-                        with gr.Column(scale=2):
-                            # Create row of quick buttons for ranks 1-5
-                            rank_buttons = []
-                            with gr.Row():
-                                for rank in range(1, min(6, len(samples[0]["candidates"])+1)):
-                                    btn = gr.Button(f"{rank}", size="sm")
-                                    rank_buttons.append((btn, rank))
-                        
                         with gr.Column(scale=2):
                             # Dropdown for ranking
                             rank_input = gr.Dropdown(
@@ -651,26 +641,6 @@ def create_reranking_interface(task_data):
             
             return [result]
         
-        # Connect rank buttons to their corresponding documents
-        for i, doc_rank_input in enumerate(ranking_inputs):
-            # Get the buttons for this row
-            row_buttons = rank_buttons[i*5:(i+1)*5] if i*5 < len(rank_buttons) else []
-            for btn, rank in row_buttons:
-                # Create a function to set this specific rank
-                def set_rank(rank_value=rank, doc_index=i):
-                    def set_this_rank():
-                        result = [""] * len(ranking_inputs)
-                        result[doc_index] = str(rank_value)
-                        return result
-                    return set_this_rank
-                    
-                # Connect button to ranking input
-                btn.click(
-                    fn=set_rank(),
-                    inputs=None,
-                    outputs=[doc_rank_input]
-                )
-    
     return demo
 
 # Main app with file upload capability and improved task management