Upload app_federal.py

app_federal.py

@@ -196,6 +196,11 @@ def load_NJ_caselaw():
     return ids, titles, chunks
 
 
+#TODO implement this
+def rerank_with_chatGPT(query, search_results)
+    return
+
+
 def run_retrieval(query):
     query = " ".join(query.split())
     print ("query", query)
@@ -215,7 +220,7 @@ def run_retrieval(query):
     for item in results:
         index = item["index"]
         item["query"] = query
-        item["opinion_idx"] = str(ds_paragraphs[i]["idx"])
+        item["opinion_idx"] = str(ds_paragraphs[index]["idx"])
 
         # only recover one paragraph / opinion
         if item["opinion_idx"] in covered: