import os
import sys
from src.reddit_client import get_next_post, mark_processed
from src.processor import optimize_text
from src.voice_gen import generate_audio
from src.subtitler import transcribe
from src.video_engine import render_video

WHISPER_MODEL = os.getenv("WHISPER_MODEL", "base")
TEMP_AUDIO = "temp_audio.mp3"


def run():
    print("[main] Fetching next Reddit post...")
    post = get_next_post()
    if not post:
        print("[main] No new viral posts found. Try again later.")
        sys.exit(0)

    print(f"[main] Found: r/{post['subreddit']} — {post['title'][:60]} (score: {post['score']})")

    print("[main] Optimizing text with Groq...")
    script = optimize_text(post["title"], post["text"])
    print(f"[main] Script preview: {script[:120]}...")

    print("[main] Generating voice...")
    generate_audio(script, TEMP_AUDIO)

    print("[main] Transcribing with Whisper...")
    words = transcribe(TEMP_AUDIO, model_name=WHISPER_MODEL)
    print(f"[main] Got {len(words)} word timestamps.")

    print("[main] Rendering video...")
    output_path = render_video(TEMP_AUDIO, words, post["id"])

    mark_processed(post["id"])
    print(f"[main] Done! Video saved to: {output_path}")

    if os.path.exists(TEMP_AUDIO):
        os.remove(TEMP_AUDIO)


if __name__ == "__main__":
    run()