# run_digs.py

import os
import time
from digest_segment import process_segment
from export_to_spreadsheet import export_to_csv

# Directory containing segment files
SEGMENT_PATH = os.path.join(os.path.dirname(__file__), "digest_paths")

def run_digs_on_all():
    print("Starting DIGS engine...")
    all_results = []

    for filename in os.listdir(SEGMENT_PATH):
        if filename.endswith(".txt"):
            filepath = os.path.join(SEGMENT_PATH, filename)
            print(f"Processing: {filename}")
            try:
                result = process_segment(filepath)
                if result:
                    all_results.append(result)
            except Exception as e:
                print(f"Error processing {filename}: {e}")

    print("Exporting results...")
    export_to_csv(all_results)
    print("DIGS processing complete.")

if __name__ == "__main__":
    run_digs_on_all()