bookapp/ai_blueprint.md

AI Blueprint: Modularization Plan

This blueprint details the strategy to break down the monolithic files (main.py, wizard.py, modules/story.py, modules/web_app.py, modules/marketing.py, modules/web_tasks.py, modules/ai.py) into a small-file, Single Responsibility architecture.

Proposed Folder Structure

c:/Users/thethreemagi/OneDrive/Gemini/BookApp/
├── core/
│   ├── config.py             
│   └── utils.py              
├── ai/
│   ├── setup.py              
│   └── models.py             
├── story/
│   ├── planner.py            
│   ├── writer.py             
│   ├── editor.py             
│   ├── style_persona.py      
│   └── bible_tracker.py      
├── marketing/
│   ├── blurb.py              
│   ├── cover.py              
│   ├── fonts.py              
│   └── assets.py             
├── export/
│   └── exporter.py           
├── web/
│   ├── app.py                
│   ├── db.py                 
│   ├── tasks.py              
│   └── routes/
│       ├── auth.py           
│       ├── project.py        
│       ├── run.py            
│       ├── admin.py          
│       └── persona.py        
├── cli/
│   ├── engine.py             
│   └── wizard.py             

Step-by-Step Migration Details

1. core/ Module

New File: core/config.py

• Moves config.py unchanged.

New File: core/utils.py

• Moves all functions from modules/utils.py.
• Exact Imports:

  import os, json, datetime, time, threading, re
  from core import config
  

2. ai/ Module (Extracting from modules/ai.py)

New File: ai/models.py

• Extract ResilientModel class, and global model variables (model_logic, model_writer, etc.).
• Exact Imports:

  import google.generativeai as genai
  from core import utils
  

New File: ai/setup.py

• Extract init_models, select_best_models, get_optimal_model, get_default_models.
• Exact Imports:

  import os, json, time, warnings
  import google.generativeai as genai
  from core import config, utils
  from ai import models
  

3. story/ Module (Extracting from modules/story.py)

New File: story/planner.py

• Extract enrich, plan_structure, expand, create_chapter_plan.
• Exact Imports:

  import json, random
  from core import utils
  from ai import models as ai_models
  from story.bible_tracker import filter_characters
  

New File: story/writer.py

• Extract write_chapter.
• Exact Imports:

  import json, os
  from core import config, utils
  from ai import models as ai_models
  from story.style_persona import get_style_guidelines
  from story.editor import evaluate_chapter_quality
  

New File: story/editor.py

• Extract evaluate_chapter_quality, check_pacing, analyze_consistency, rewrite_chapter_content, check_and_propagate.
• Exact Imports:

  import json, os
  from core import config, utils
  from ai import models as ai_models
  from story.style_persona import get_style_guidelines
  

New File: story/style_persona.py

• Extract get_style_guidelines, refresh_style_guidelines, create_initial_persona, refine_persona, update_persona_sample.
• Exact Imports:

  import json, os, time
  from core import config, utils
  from ai import models as ai_models
  

New File: story/bible_tracker.py

• Extract merge_selected_changes, filter_characters, update_tracking, harvest_metadata, refine_bible.
• Exact Imports:

  import json
  from core import utils
  from ai import models as ai_models
  

4. marketing/ Module (Extracting from modules/marketing.py)

New File: marketing/fonts.py

• Extract download_font.
• Exact Imports:

  import os, requests
  from core import config, utils
  

New File: marketing/cover.py

• Extract generate_cover, evaluate_image_quality.
• Exact Imports:

  import os, json, shutil, textwrap, subprocess, sys
  from core import config, utils
  from ai import models as ai_models
  from marketing.fonts import download_font
  from rich.prompt import Confirm
  try:
      from PIL import Image, ImageDraw, ImageFont
      HAS_PIL = True
  except ImportError:
      HAS_PIL = False
  

New File: marketing/blurb.py

• Extract generate_blurb.
• Exact Imports:

  import os
  from core import utils
  from ai import models as ai_models
  

New File: marketing/assets.py

• Extract create_marketing_assets.
• Exact Imports:

  from marketing.blurb import generate_blurb
  from marketing.cover import generate_cover
  

5. export/ Module (Extracting from modules/export.py)

New File: export/exporter.py

• Extract create_readme, compile_files.
• Exact Imports:

  import os, markdown
  from docx import Document
  from ebooklib import epub
  from core import utils
  

6. web/ Module (Extracting from modules/web_app.py and modules/web_tasks.py)

New File: web/db.py

• Extract models User, Project, Run, LogEntry from modules/web_db.py.
• Exact Imports:

  from flask_sqlalchemy import SQLAlchemy
  from flask_login import UserMixin
  from datetime import datetime
  db = SQLAlchemy()
  

New File: web/tasks.py

• Extract Huey configuration and tasks generate_book_task, regenerate_artifacts_task, rewrite_chapter_task, refine_bible_task.
• Exact Imports:

  import os, json, time, sqlite3, shutil
  from datetime import datetime
  from huey import SqliteHuey
  from core import config, utils
  from ai import setup as ai_setup
  from story import planner, writer, editor, style_persona, bible_tracker
  from marketing import cover, assets
  from export import exporter
  from web.db import db, Run, User, Project
  

New File: web/routes/auth.py

• Extract /login, /register, /logout.
• Exact Imports:

  from flask import Blueprint, render_template, request, redirect, url_for, flash
  from flask_login import login_user, login_required, logout_user, current_user
  from werkzeug.security import generate_password_hash, check_password_hash
  from web.db import db, User
  

New File: web/routes/project.py

• Extract /, /project/setup, /project/setup/refine, /project/create, /project/import, /project/<id>, /project/<id>/update, /project/<id>/clone, /project/<id>/add_book, /project/<id>/book/<book_num>/update, /project/<id>/delete_book/<book_num>, /project/<id>/import_characters, /project/<id>/set_persona.
• Exact Imports:

  import os, json
  from datetime import datetime
  from flask import Blueprint, render_template, request, redirect, url_for, flash
  from flask_login import login_required, current_user
  from core import config, utils
  from ai import setup as ai_setup, models as ai_models
  from web.db import db, Project, Run
  from story import planner, bible_tracker
  

New File: web/routes/run.py

• Extract /project/<id>/run, /run/<id>, /run/<id>/status, /run/<id>/stop, /run/<id>/restart, /project/<run_id>/regenerate_artifacts, /project/<run_id>/revise_book/<book>, /project/<run_id>/read/<book>, /project/<run_id>/save_chapter, /project/<run_id>/check_consistency/<book>, /project/<run_id>/sync_book/<book>, /project/<run_id>/rewrite_chapter, /project/<run_id>/download, /task_status/<task_id>.
• Exact Imports:

  import os, json, markdown
  from flask import Blueprint, render_template, request, redirect, url_for, flash, send_from_directory, session
  from flask_login import login_required, current_user
  from web.db import db, Project, Run, LogEntry
  from web.tasks import huey, generate_book_task, regenerate_artifacts_task, rewrite_chapter_task
  from core import config, utils
  from story import editor, bible_tracker, style_persona
  from export import exporter
  

New File: web/routes/persona.py

• Extract /personas, /persona/new, /persona/<name>, /persona/save, /persona/delete/<name>, /persona/analyze.
• Exact Imports:

  import os, json
  from flask import Blueprint, render_template, request, redirect, url_for, flash
  from flask_login import login_required
  from core import config, utils
  from ai import setup as ai_setup, models as ai_models
  

New File: web/routes/admin.py

• Extract /admin, /admin/user/<id>/delete, /admin/project/<id>/delete, /admin/reset, /admin/spend, /admin/style, /admin/impersonate/<id>, /admin/stop_impersonate, /debug/routes, /system/optimize_models, /system/status.
• Exact Imports:

  import os, json, shutil
  from datetime import datetime, timedelta
  from flask import Blueprint, render_template, request, redirect, url_for, flash, session
  from flask_login import login_required, current_user, login_user
  from sqlalchemy import func
  from web.db import db, User, Project, Run
  from core import config, utils
  from story import style_persona
  from ai import setup as ai_setup, models as ai_models
  

New File: web/app.py

• Flask initialization, registering blueprints, and the __main__ entry point.
• Exact Imports:

  import os
  from flask import Flask
  from flask_login import LoginManager
  from core import config
  from web.db import db, User
  

7. cli/ Module

New File: cli/engine.py

• Extract process_book and run_generation from main.py.
• Exact Imports:

  import json, os, time, sys, shutil
  from rich.prompt import Confirm
  from core import config, utils
  from ai import setup as ai_setup
  from story import planner, writer, editor, style_persona, bible_tracker
  from marketing import cover, assets
  from export import exporter
  

New File: cli/wizard.py

• Move wizard.py here and update imports.
• Exact Imports:

  import os, sys, json
  from rich.console import Console
  from rich.panel import Panel
  from rich.prompt import Prompt, IntPrompt, Confirm
  from rich.table import Table
  from flask import Flask
  from core import config, utils
  from ai import setup as ai_setup, models as ai_models
  from web.db import db, User, Project
  

HTML Template Changes

The templates/ folder contains HTML files. Ensure any url_for() calls reference the correct Blueprint names (e.g., url_for('auth.login') instead of url_for('login')).

Awaiting approval to begin creating files and making structural changes.