Update app.py

app.py

@@ -115,6 +115,56 @@ pdf_qa = ConversationalRetrievalChain.from_llm(
 chat_history = []
 
 
+##################################################################################################################
+# Create a log file to store user queries and responses
+log_file = None  # Initialize log_file as None
+
+def setup_log_file():
+    global log_file
+    try:
+        log_file = open("query_log.txt", "a")
+    except Exception as e:
+        print("Error opening log file:", str(e))
+
+def close_log_file():
+    global log_file
+    if log_file is not None:
+        log_file.close()
+
+        
+
+def log_query(query, response):
+    if log_file is not None:
+        timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        log_entry = f"[{timestamp}] User Query: {query}\nSystem Response: {response}\n\n"
+        log_file.write(log_entry)
+        log_file.flush()  # Ensure the log is written immediately
+
+
+# Setup the log file before launching the interface
+setup_log_file()
+
+
+log_file_path = "query_log.txt"
+
+def log_message(message):
+    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    log_entry = f"[{timestamp}] {message}\n"
+    with open(log_file_path, "a") as log_file:
+        log_file.write(log_entry)
+        print("Message logged successfully.")
+        log_file.flush()  # Ensure the log is written immediately
+
+# Test logging
+log_message("This is a test message.")
+
+
+    
+# Close the log file when the application exits
+atexit.register(close_log_file)
+###########################################################################################################
+
+
 def ask_alans_ai(query, vectordb):
     global chat_history
     result = pdf_qa(
@@ -172,52 +222,3 @@ if __name__ == "__main__":
 
 
 
-# Create a log file to store user queries and responses
-log_file = None  # Initialize log_file as None
-
-def setup_log_file():
-    global log_file
-    try:
-        log_file = open("query_log.txt", "a")
-    except Exception as e:
-        print("Error opening log file:", str(e))
-
-def close_log_file():
-    global log_file
-    if log_file is not None:
-        log_file.close()
-
-        
-
-def log_query(query, response):
-    if log_file is not None:
-        timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        log_entry = f"[{timestamp}] User Query: {query}\nSystem Response: {response}\n\n"
-        log_file.write(log_entry)
-        log_file.flush()  # Ensure the log is written immediately
-
-
-# Setup the log file before launching the interface
-setup_log_file()
-
-
-log_file_path = "query_log.txt"
-
-def log_message(message):
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    log_entry = f"[{timestamp}] {message}\n"
-    with open(log_file_path, "a") as log_file:
-        log_file.write(log_entry)
-        print("Message logged successfully.")
-        log_file.flush()  # Ensure the log is written immediately
-
-# Test logging
-log_message("This is a test message.")
-
-
-    
-
-
-
-# Close the log file when the application exits
-atexit.register(close_log_file)