Update app.py

app.py

@@ -27,7 +27,7 @@ def create_reranking_interface(task_data):
             if len(set(processed_rankings)) != len(processed_rankings):
                 return "⚠️ Each document must have a unique rank. Please review your rankings.", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
                 
-            # Store this annotation
+            # Store this annotation in memory
             existing_idx = next((i for i, a in enumerate(results["annotations"]) if a["sample_id"] == sample_id), None)
             if existing_idx is not None:
                 results["annotations"][existing_idx] = {
@@ -41,21 +41,19 @@ def create_reranking_interface(task_data):
                 })
             
             completed_samples[sample_id] = True
-            success_msg = f"✅ Rankings for query '{sample_id}' successfully saved!"
-            progress = f"Progress: {sum(completed_samples.values())}/{len(samples)}"
             
-            # Auto-save results after each submission with error handling
+            # Try to save to file, but continue even if it fails
             try:
                 output_path = f"{task_data['task_name']}_human_results.json"
                 with open(output_path, "w") as f:
                     json.dump(results, f, indent=2)
-            except Exception as e:
-                return f"⚠️ Rankings saved in memory but couldn't write to file: {str(e)}", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
-            
-            return success_msg, progress
+                return f"✅ Rankings saved successfully (in memory and to file)", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
+            except:
+                # If file saving fails, still mark as success since we saved in memory
+                return f"✅ Rankings saved in memory (file save failed)", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
         except Exception as e:
-            # Catch-all for any other unexpected errors
-            return f"⚠️ An error occurred: {str(e)}", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
+            # Return specific error message
+            return f"Error: {str(e)}", f"Progress: {sum(completed_samples.values())}/{len(samples)}"
     
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown(f"# {task_data['task_name']} - Human Reranking Evaluation")