utils/file_utils.py

# file_utils.py
import os
import tempfile
import shutil
import logging
import streamlit as st

logger = logging.getLogger(__name__)

@st.cache_data
def load_prompt(filepath):
    with open(filepath, "r", encoding="utf-8") as f:
        data = f.read()
    logger.debug("Loaded prompt from %s (length: %d)", filepath, len(data))
    return data

def save_intermediate_output(content, pdf_basename, suffix):
    file_path = os.path.join("promp_tmp", f"{pdf_basename}_{suffix}.txt")
    with open(file_path, "w", encoding="utf-8") as f:
        f.write(content)
    logger.debug("Saved intermediate output to %s", file_path)

def setup_temp_directories():
    if not os.path.exists("promp_tmp"):
        os.makedirs("promp_tmp")
        logger.debug("Created directory 'promp_tmp'.")
    temp_dir = tempfile.mkdtemp(prefix="pdf_digester_")
    logger.debug(f"Created temporary directory: {temp_dir}")
    return temp_dir

def cleanup_temp_files(temp_dir):
    try:
        shutil.rmtree(temp_dir)
        logger.debug(f"Cleaned up temporary directory: {temp_dir}")
    except Exception as e:
        logger.error(f"Failed to cleanup temporary directory: {e}")