working containerized build

2026-06-13 13:27:29 -07:00
parent 34690b3691
commit 530fd93a0b
12 changed files with 441 additions and 10 deletions
@@ -1,13 +1,15 @@
 import json
 import asyncio
+import os
 import httpx
 from loguru import logger
 from typing import List, Dict

-OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
-MODEL_NAME = "gemma4:e2b"
-INPUT_FILE = "data/output.json"
-OUTPUT_FILE = "data/scripts.json"
+OLLAMA_URL = os.environ.get("OLLAMA_URL", "http://127.0.0.1:11434/api/generate")
+MODEL_NAME = os.environ.get("MODEL_NAME", "gemma4:e2b")
+DATA_DIR = os.environ.get("DATA_DIR", "data")
+INPUT_FILE = os.path.join(DATA_DIR, "output.json")
+OUTPUT_FILE = os.path.join(DATA_DIR, "scripts.json")

 async def generate_script(client: httpx.AsyncClient, headline: str, content: str, attempt: int = 1) -> str:
    system_prompt = (