Update app.py

app.py

@@ -62,13 +62,28 @@ for output in outputs:
 
 search_query = sorted(query2outputs)[0]
 
+
+def aggregate(list_of_hits):
+    import numpy as np
+    from permsc import KemenyOptimalAggregator, sum_kendall_tau, ranks_from_preferences
+
+    preferences = []
+    for result in list_of_hits:
+        preferences.append([doc["rank"] for doc in result])
+
+    preferences = np.array(preferences)
+    y_optimal = KemenyOptimalAggregator().aggregate(preferences)
+    rank2doc = {}
+    for doc in list_of_hits[0]:
+        rank2doc[doc["rank"]] = doc
+    return [rank2doc[rank] for rank in y_optimal]
+    
+
 if search_query or button_clicked:
     
     num_results = None
     t_0 = time.time()
     search_results = query2outputs[search_query][0] # first from the 20
-    print(search_results)
-    print()
 
     print(len(search_results), type(search_results[0]))