Update app.py

app.py

@@ -55,13 +55,13 @@ test = str(vectorstore_persisted.get())[:20]
 log_file = Path("logs/") / f"data_{uuid.uuid4()}.json"
 log_folder = log_file.parent
 
-# scheduler = CommitScheduler(
-#     repo_id="MJackman-Project3-logs",
-#     repo_type="dataset",
-#     folder_path=log_folder,
-#     path_in_repo="data",
-#     every=2
-# )
+scheduler = CommitScheduler(
+    repo_id="MJackman-Project3-logs",
+    repo_type="dataset",
+    folder_path=log_folder,
+    path_in_repo="data",
+    every=2
+)
 
 # Define the Q&A system message
 qna_system_message = """
@@ -95,7 +95,7 @@ Answer:
 Source:
 [Source]
 
-Page Number:
+Page Number(s):
 [Page Number]
 """
 
@@ -155,16 +155,16 @@ def predict(user_input,company):
     # While writing to the log file, ensure that the commit scheduler is locked to avoid parallel
     # access
 
-    # with scheduler.lock:
-    #     with log_file.open("a") as f:
-    #         f.write(json.dumps(
-    #             {
-    #                 'user_input': user_input,
-    #                 'retrieved_context': context_for_query,
-    #                 'model_response': prediction
-    #             }
-    #         ))
-    #         f.write("\n")
+    with scheduler.lock:
+        with log_file.open("a") as f:
+            f.write(json.dumps(
+                {
+                    'user_input': user_input,
+                    'retrieved_context': context_for_query,
+                    'model_response': prediction
+                }
+            ))
+            f.write("\n")
 
     return prediction