Update handler.py

handler.py

@@ -29,10 +29,26 @@ class EndpointHandler:
         sorted_programs = sorted(set(zip(consistent_program_scores, consistent_programs)), reverse=True, key=lambda x: x[0])
         end = time.time()
 
+        top_guess = None
+        top_score = None
+        top_success = False
+        top_10_guesses = None
+        top_10_scores = None
+        top_10_success = False
+        if len(sorted_programs) > 0:
+            top_guess = sorted_programs[0][1]
+            top_score = sorted_programs[0][0]
+            top_success = parse(top_guess).equivalent(parse(true_program))
+            top_10_guesses = [p for s, p in sorted_programs[:10]]
+            top_10_scores = [s for s, p in sorted_programs[:10]]
+            top_10_success = any([parse(p).equivalent(parse(true_program)) for p in top_10_guesses])
+
         return {
-            "guess": sorted_programs[0][1],
-            "top_1_success": parse(sorted_programs[0][1]).equivalent(parse(true_program)),
-            "top_1_score": sorted_programs[0][0],
-            "top_5_success": any([parse(p).equivalent(parse(true_program)) for _, p in sorted_programs[:5]]),
+            "guess": top_guess,
+            "top_1_success": top_success,
+            "top_1_score": top_score,
+            "top_10_guesses": top_10_guesses,
+            "top_10_scores": top_10_scores,
+            "top_10_success": top_10_success,
             "time": end - start
         }