export.py

import json
import os
from datetime import datetime
from memory import get_conversation


from pathlib import Path

BASE_DIR = Path(__file__).resolve().parent
EXPORT_DIR = BASE_DIR / "exports"

EXPORT_DIR.mkdir(exist_ok=True, parents=True)


def _filename(extension):
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    return str(EXPORT_DIR / f"conversation_{timestamp}.{extension}")


def export_to_json():
    conversation = get_conversation()
    file_name = _filename("json")

    with open(
        file_name,
        "w",
        encoding="utf-8"
    ) as file:
        json.dump(
            conversation,
            file,
            indent=4,
            ensure_ascii=False
        )

    return file_name


def export_to_txt():
    conversation = get_conversation()

    file_name = _filename("txt")

    with open(
        file_name,
        "w",
        encoding="utf-8"
    ) as file:
        for message in conversation:
            role = message["role"].upper()
            content = message["content"]

            file.write(f"{role}\n")

            file.write(f"{content}\n")

            file.write("-" * 60 + "\n")

    return file_name


def export_to_markdown():
    conversation = get_conversation()

    file_name = _filename("md")

    with open(
        file_name,
        "w",
        encoding="utf-8"
    ) as file:
        file.write("# Conversation Export\n\n")

        for message in conversation:

            role = message["role"].capitalize()

            content = message["content"]

            file.write(f"## {role}\n\n")

            file.write(f"{content}\n\n")

    return file_name