Bläddra i källkod

Added header line to output

Josh Rosario 8 månader sedan
förälder
incheckning
075c98d6e4
1 ändrade filer med 6 tillägg och 1 borttagningar
  1. 6 1
      main.py

+ 6 - 1
main.py

@@ -1,5 +1,6 @@
 import sys
 import os
+from datetime import datetime, timezone
 from bs4 import BeautifulSoup, NavigableString
 
 
@@ -130,7 +131,11 @@ def convert_chat_html_to_markdown(html_path: str) -> str:
             message = f"**ChatGPT**\n\n{body}"
         messages.append(message)
 
-    markdown = f"# {title}\n\n" + "\n\n---\n\n".join(messages)
+    dd_trace_time = int(int(soup.find("meta", {"name":"dd-trace-time"}).attrs["content"]) / 1000)  # UTC time
+    timestamp = datetime.fromtimestamp(dd_trace_time).strftime("%c")
+    header = f"*ChatGPT conversation saved {timestamp} converted to Markdown*"
+
+    markdown = f"# {title}\n\n{header}\n\n---\n\n" + "\n\n---\n\n".join(messages)
     return filename, markdown