feat: 🎸 add logging feature for report generation

utils/report_generator.py

@@ -0,0 +1,42 @@
+# a quick report generation script that converts the saved logs file into a pdf.
+import json, os, sys
+import datetime, time
+
+
+def main(file_name):
+    # load the file into json
+    with open(file_name, "r") as f:
+        logs = json.load(f)
+    user_inputs = logs["user"]
+    bot_responses = logs["pentestGPT"]
+    # merge the two list into one. Sort based on the first element (timestamp in str), and add a flag to show source
+    merged_list = []
+    for user_input in user_inputs:
+        merged_list.append([user_input[0], user_input[1], "user"])
+    for bot_response in bot_responses:
+        merged_list.append([bot_response[0], bot_response[1], "pentestGPT"])
+    merged_list.sort(key=lambda x: x[0])
+
+    # now print the conversation
+    output = ""
+    for element in merged_list:
+        # convert the timestamp to a human readable format
+        timestamp = datetime.datetime.fromtimestamp(int(element[0])).strftime(
+            "%Y-%m-%d %H:%M:%S"
+        )
+        output += f"{timestamp} [{element[2]}]: {element[1]}\n"
+        # add an additional line break if the element is from bot
+        if element[2] == "pentestGPT":
+            output += "----------------------------------------\n\n"
+    # print the output
+    print("Conversation log: ")
+
+    print(output)
+
+if __name__ == "__main__":
+    # default filename = "../logs/sample_pentestGPT_log.txt"
+    if len(sys.argv) == 1:
+        file_name = "logs/sample_pentestGPT_log.txt"
+    else:
+        file_name = sys.argv[1]
+    main(file_name)