main.py

from stt import capture_dream
from da import analyze_emotions
from dreamart import draw_dream_art
import csv
import os
from datetime import datetime
import subprocess

# Step 1: Capture dream text
dream_text = capture_dream()
if not dream_text:
    print("❌ No valid dream captured. Exiting.")
    exit()

# Step 2: Analyze emotions + surrealism
emotions, surreal_score = analyze_emotions(dream_text)

# Step 3: Save to CSV
csv_file = "dream_diary.csv"
fieldnames = ["Date", "Dream", "Happy", "Angry", "Surprise", "Sad", "Fear", "Surrealism"]

new_row = {
    "Date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
    "Dream": dream_text,
    "Happy": emotions["Happy"],
    "Angry": emotions["Angry"],
    "Surprise": emotions["Surprise"],
    "Sad": emotions["Sad"],
    "Fear": emotions["Fear"],
    "Surrealism": surreal_score
}

# If file doesn't exist, write headers
file_exists = os.path.isfile(csv_file)
with open(csv_file, mode="a", newline="", encoding="utf-8") as file:
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    if not file_exists:
        writer.writeheader()
    writer.writerow(new_row)

print(f"📄 Logged dream to {csv_file}.")

# Step 4: Draw the art
draw_dream_art(emotions, surreal_score)
subprocess.run(["xdg-open", "dream_art.png"])  # Open the generated art
subprocess.run(["xdg-open", "dream_diary.csv"])  # Open the CSV file
subprocess.call(["Rscript", "R_analysis.R"])  # Run R script for trend analysis