log_episode.py

import os
import csv
from datetime import datetime

def log_episode(episode_number, total_reward, actions_taken, fields_filled, submitted):
    if not os.path.exists("logs"):
        os.makedirs("logs")

    now = datetime.now()
    filename = f"logs/agent_log_{now.strftime('%Y-%m-%d')}.csv"

    file_exists = os.path.isfile(filename)

    with open(filename, mode='a', newline='', encoding='utf-8') as f:
        writer = csv.writer(f)
        # Pokud soubor neexistuje, zapíšeme hlavičku
        if not file_exists:
            writer.writerow([
                "Čas",
                "Epizoda",
                "Celková odměna",
                "Počet vyplněných polí",
                "Odesláno",
                "Akce"
            ])

        writer.writerow([
            now.strftime('%Y-%m-%d %H:%M:%S'),
            episode_number,
            total_reward,
            str(fields_filled),
            "ANO" if submitted else "NE",
            ", ".join(actions_taken) if isinstance(actions_taken, (list, tuple)) else str(actions_taken)
        ])