Upload app_federal.py

app_federal.py

@@ -205,6 +205,8 @@ def register_review(history, additional_feedback, response_json):
         "additional_feedback": additional_feedback
     }
 
+    # ok, have to add some things, but should be easy
+
     with scheduler.lock:
         with JSON_DATASET_PATH.open("a") as f:
             json.dump(res, f)
@@ -289,7 +291,7 @@ def run_retrieval(query):
 
     query_embeddings = run_dense_retrieval(query)
     #query_embeddings = pca_model.transform(query_embeddings)
-    D, I = faiss_index.search(query_embeddings, 100)
+    D, I = faiss_index.search(query_embeddings, 45)
     scores_embeddings = list(D[0]) 
     indices_embeddings = I[0]
     indices_embeddings = [int(i) for i in indices_embeddings]
@@ -326,7 +328,7 @@ def run_retrieval(query):
             item["metadata_reranking"] = ""
         out_dict.append(item)
     print ("out_dict_before_reranking")
-    print (out_dict[:50])
+    #print (out_dict[:50])
     out_dict = rerank_with_chatGPT(query, out_dict)[:NUM_RESULTS]
     print ("RETURNING OUT DICT")
     return out_dict