Update app.py

app.py

@@ -113,7 +113,18 @@ def evaluate_predictions(prediction_file, model_name, add_to_leaderboard):
         return "Prediction file not uploaded.", load_leaderboard()
 
     try:
+        #load predition file
         predictions_df = pd.read_csv(prediction_file.name)
+         # Validate required columns in prediction file
+        required_columns = ['question_id', 'predicted_answer']
+        missing_columns = [col for col in required_columns if col not in predictions_df.columns]
+        if missing_columns:
+            return (f"Error: Missing required columns in prediction file: {', '.join(missing_columns)}.",
+                    load_leaderboard())
+
+        # Validate 'Answer' column in ground truth file
+        if 'Answer' not in ground_truth_df.columns:
+            return "Error: 'Answer' column is missing in the ground truth dataset.", load_leaderboard()
         merged_df = pd.merge(predictions_df, ground_truth_df, on='question_id', how='inner')
         merged_df['pred_answer'] = merged_df['predicted_answer'].apply(clean_answer)