Update app.py

app.py

@@ -55,6 +55,14 @@ vectorstores = load_chunks()
 # -------------------------------------------------------------
 # Functions
 # -------------------------------------------------------------
+
+def save_logs(logs) -> None:
+    with scheduler.lock:
+        with JSON_DATASET_PATH.open("a") as f:
+            json.dump(logs, f)
+            f.write("\n")
+    print("logging done")
+          
 def make_html_source(source,i):
     """
     takes the text and converts it into html format for display in "source" side tab
@@ -227,6 +235,28 @@ async def chat(query,history,sources,reports,subtype,year):
     history = [tuple(x) for x in history]
         
     yield history,docs_html
+
+
+    try:
+        timestamp = str(datetime.now().timestamp())
+        #file_store = "/data/logs/" + timestamp + ".json"
+        logs = {
+                "system_prompt": SYSTEM_PROMPT,
+                "sources":sources,
+                "reports":reports,
+                "subtype":subtype,
+                "year":year,
+                "question":query,
+                "sources":sources,
+                "retriever":
+                "docs":[doc.page_content for doc in context_retrieved],
+                "answer": history[-1][1],
+                "time": timestamp,
+            }
+        save_logs(file_store,logs)
+    except Exception as e:
+        print(e)
+
 #process_pdf()