Update app.py

app.py

@@ -42,8 +42,15 @@ def load_test_data(task):
     with open('/app/test_text.txt', 'r') as file:
             content = file.read()
             print(content)
-    with open(os.path.join(DATA_DIR, f"{task}.json"), "r", encoding='utf-8') as f:
-        test_data = json.load(f)
+    # with open(os.path.join(DATA_DIR, f"{task}.json"), "r", encoding='utf-8') as f:
+    #     test_data = json.load(f)
+
+    try:
+        with open(os.path.join(DATA_DIR, f"{task}.json"), "r", encoding='utf-8') as f:
+            test_data = json.load(f)
+    except FileNotFoundError:
+        return jsonify({"message": "Test data file not found"}), 400
+
     
     # 更新音频路径，将它们指向 Flask 静态文件夹
     for item in test_data:
@@ -79,6 +86,8 @@ def start_test():
 
     # Load the test data
     test_data = load_test_data(task)
+    if not test_data:
+        return jsonify({"message": "No test data available"}), 400
     
     # Shuffle test data for the user
     random.shuffle(test_data)
@@ -95,6 +104,9 @@ def start_test():
 
     task_description = test_data[0].get('task_description', '')
 
+    print(session)
+
+
     return jsonify({"message": "Test started", "total_tests": len(test_data), "task_description": task_description})
 
 
@@ -102,8 +114,12 @@ def start_test():
 @app.route('/next_test', methods=['GET'])
 def next_test():
     """Serve the next test item"""
+    # if 'current_index' not in session or 'test_data' not in session:
+    #     return jsonify({"message": "No active test found"}), 400
+
     if 'current_index' not in session or 'test_data' not in session:
-        return jsonify({"message": "No active test found"}), 400
+        return jsonify({"message": "Session data missing", "session": session}), 400
+
 
     current_index = session['current_index']
     test_data = session['test_data']
@@ -115,6 +131,9 @@ def next_test():
     # 使用 EloRank 的 sample_next_match 来选择两款模型
     selected_models = elo_rank_system.sample_next_match()
 
+    if not selected_models or len(selected_models) != 2:
+        return jsonify({"message": "Error selecting models"}), 500
+
     # Serve test data with the two selected models
     current_test = test_data[current_index]
     session['selected_models'] = selected_models