Upload app_federal.py

app_federal.py

@@ -312,7 +312,6 @@ def run_retrieval(query):
         index = item["index"]
         item["query"] = query
         item["opinion_idx"] = str(ds_paragraphs[index]["idx"])
-
         # only recover one paragraph / opinion
         if item["opinion_idx"] in covered:
             continue
@@ -329,7 +328,17 @@ def run_retrieval(query):
         out_dict.append(item)
     print ("out_dict_before_reranking")
     #print (out_dict[:50])
+    res = {"result_type":"chatgpt_reranking"}
+    res["input_reranking"] = [i["index"]] for i in out_dict
+    res["scores"] = [i["NV_score"] for i in out_dict]
     out_dict = rerank_with_chatGPT(query, out_dict)[:NUM_RESULTS]
+
+    # is that already good?
+    with scheduler.lock:
+        with JSON_DATASET_PATH.open("a") as f:
+            json.dump(res, f)
+            f.write("\n")
+
     print ("RETURNING OUT DICT")
     return out_dict