Mike Wichers
1964c9c2a5
story.py — write_chapter():
- Added POSITION context ("Chapter N of Total") so the AI calibrates narrative
tension correctly (setup vs escalation vs climax/payoff)
- Moved PACING_GUIDE to sit directly after PACING metadata instead of being
buried after 13 quality criteria items where the AI rarely reads it
- Removed duplicate pacing descriptions that appeared after QUALITY_CRITERIA
story.py — refinement loop:
- Capped critique history to last 2 entries (was accumulating all previous
attempts, wasting tokens and confusing the model on attempt 4-5)
- Added TARGET_WORDS and BEATS constraints to the refinement prompt to prevent
chapters from shrinking or losing plot beats during editing passes
- Restructured refinement prompt with explicit HARD_CONSTRAINTS section
story.py — check_and_propagate():
- Increased chapter context from 5000 to 12000 chars for continuity rewrites
(was asking for a full chapter rewrite but only providing a fragment)
- Added explicit word count target to rewrite so chapters are not truncated
- Added conservative decision bias: only rewrite on genuine contradictions
story.py — plan_structure():
- Now passes TARGET_CHAPTERS, TARGET_WORDS, GENRE, and CHARACTERS to the
structure AI — it was planning blindly without knowing the book's scale
marketing.py — generate_blurb():
- Rewrote prompt with 4-part structure: Hook → Stakes → Tension → Close
- Formats plot beats as a readable list instead of raw JSON array
- Extracts protagonist automatically for personalised blurb copy
- Added genre-tone matching, present-tense voice, and no-spoiler rule
marketing.py — generate_cover():
- Added genre-to-visual-style mapping (thriller → cinematic, fantasy → epic
digital painting, romance → painterly, etc.)
- Art prompt instructions now enforce: no text/letters/watermarks, rule-of-thirds
composition, explicit focal point, lighting description, colour palette
- Replaced generic image evaluation with a 5-criteria book-cover rubric:
visual impact, genre fit, composition, quality, and clean image (no text)
- Score penalties: -3 for visible text/watermarks, -2 for blur/deformed anatomy
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
68 lines
2.9 KiB
Python
68 lines
2.9 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
# Ensure .env is loaded from the script's directory (VS Code fix)
|
|
load_dotenv(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".env"))
|
|
|
|
def get_clean_env(key, default=None):
|
|
val = os.getenv(key, default)
|
|
return val.strip() if val else None
|
|
|
|
API_KEY = get_clean_env("GEMINI_API_KEY")
|
|
GCP_PROJECT = get_clean_env("GCP_PROJECT")
|
|
GCP_LOCATION = get_clean_env("GCP_LOCATION", "us-central1")
|
|
MODEL_LOGIC_HINT = get_clean_env("MODEL_LOGIC", "AUTO")
|
|
MODEL_WRITER_HINT = get_clean_env("MODEL_WRITER", "AUTO")
|
|
MODEL_ARTIST_HINT = get_clean_env("MODEL_ARTIST", "AUTO")
|
|
MODEL_IMAGE_HINT = get_clean_env("MODEL_IMAGE", "AUTO")
|
|
DEFAULT_BLUEPRINT = "book_def.json"
|
|
|
|
# --- SECURITY & ADMIN ---
|
|
FLASK_SECRET = get_clean_env("FLASK_SECRET_KEY", "dev-secret-key-change-this")
|
|
ADMIN_USER = get_clean_env("ADMIN_USERNAME")
|
|
ADMIN_PASSWORD = get_clean_env("ADMIN_PASSWORD")
|
|
|
|
if FLASK_SECRET == "dev-secret-key-change-this":
|
|
print("⚠️ WARNING: Using default FLASK_SECRET_KEY. This is insecure for production.")
|
|
|
|
if not API_KEY: raise ValueError("❌ CRITICAL ERROR: GEMINI_API_KEY not found.")
|
|
|
|
# --- DATA DIRECTORIES ---
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
DATA_DIR = os.path.join(BASE_DIR, "data")
|
|
PROJECTS_DIR = os.path.join(DATA_DIR, "projects")
|
|
PERSONAS_DIR = os.path.join(DATA_DIR, "personas")
|
|
PERSONAS_FILE = os.path.join(PERSONAS_DIR, "personas.json")
|
|
FONTS_DIR = os.path.join(DATA_DIR, "fonts")
|
|
|
|
# --- ENSURE DIRECTORIES EXIST ---
|
|
# Critical: Create data folders immediately to prevent DB initialization errors
|
|
for d in [DATA_DIR, PROJECTS_DIR, PERSONAS_DIR, FONTS_DIR]:
|
|
if not os.path.exists(d): os.makedirs(d, exist_ok=True)
|
|
|
|
# --- AUTHENTICATION ---
|
|
GOOGLE_CREDS = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
|
|
if GOOGLE_CREDS:
|
|
# Resolve to absolute path relative to this config file if not absolute
|
|
if not os.path.isabs(GOOGLE_CREDS):
|
|
base = os.path.dirname(os.path.abspath(__file__))
|
|
GOOGLE_CREDS = os.path.join(base, GOOGLE_CREDS)
|
|
|
|
if os.path.exists(GOOGLE_CREDS):
|
|
os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = GOOGLE_CREDS
|
|
else:
|
|
print(f"⚠️ Warning: GOOGLE_APPLICATION_CREDENTIALS file not found at: {GOOGLE_CREDS}")
|
|
|
|
# --- DEFINITIONS ---
|
|
LENGTH_DEFINITIONS = {
|
|
"01": {"label": "Chapter Book", "words": "5,000 - 10,000", "chapters": 10, "depth": 1},
|
|
"1": {"label": "Flash Fiction", "words": "500 - 1,500", "chapters": 1, "depth": 1},
|
|
"2": {"label": "Short Story", "words": "5,000 - 10,000", "chapters": 5, "depth": 1},
|
|
"2b": {"label": "Young Adult", "words": "50,000 - 70,000", "chapters": 25, "depth": 3},
|
|
"3": {"label": "Novella", "words": "20,000 - 40,000", "chapters": 15, "depth": 2},
|
|
"4": {"label": "Novel", "words": "60,000 - 80,000", "chapters": 30, "depth": 3},
|
|
"5": {"label": "Epic", "words": "100,000+", "chapters": 50, "depth": 4}
|
|
}
|
|
|
|
# --- SYSTEM ---
|
|
VERSION = "1.3.0" |