Upload app_federal.py

app_federal.py

@@ -25,6 +25,24 @@ import pickle
 from huggingface_hub import CommitScheduler
 from ast import literal_eval
 import re
+import requests
+
+
+def run_courtlistener_api(casename, citation, court):
+    #casename = individual_response["casename"]
+    params = {"q": casename}
+    url = "https://www.courtlistener.com/api/rest/v4/search/"            
+    response = requests.get(url, params=params)
+
+    if response.status_code == 200:
+        print (response.json()["results"])
+        result = response.json()["results"][0]
+        new_url = "https://www.courtlistener.com" + result["absolute_url"]
+        return f"[Click to see opinion on CourtListener]({new_url})"
+    else:
+        return -1
+
+
 
 JSON_DATASET_DIR = Path("json_dataset")
 JSON_DATASET_DIR.mkdir(parents=True, exist_ok=True)
@@ -85,8 +103,8 @@ def format_metadata_for_reranking(metadata, text, idx):
     out_str = ["<id>" + str(idx) + "</id>"]
     for key in keys:
         i,j = key
-        #print (i,j)
-        #print (metadata[i])
+        print ("i, j", i,j)
+        print ("(metadata[i]", metadata[i])
         out_str.append("<" + j + ">" + str(metadata[i]) + "</" + j + ">")
     out_str.append("<paragraph>" + " ".join(text.split()) + "</paragraph>")
     return "\n".join(out_str) + "\n"