app.py

"""Rewrite Rescue — diagnose and revise pasted text. Session-only, no storage."""

from pathlib import Path

import streamlit as st

from src.providers.byo import BYOAdapter
from src.providers.demo import SAMPLE_TEXT, DemoAdapter
from src.providers.errors import ProviderError
from src.services.diagnostics import diagnose
from src.services.revision import run_revision
from src.ui.output_tabs import render_output
from src.ui.report_view import diagnose_only_result, revision_result

TEXT_TYPES = [
    "General prose",
    "Essay / article",
    "Email / message",
    "Report / memo",
    "Web page text",
    "Script / dialogue",
    "Fiction / narrative",
    "Social post",
    "Technical / explanatory",
    "Custom",
]

REGISTERS = [
    "Plain",
    "Conversational",
    "Formal",
    "Technical",
    "Essayistic",
    "Narrative",
    "Custom",
]

OUTPUT_MODES = ["Diagnose + revise", "Revise only", "Diagnose only"]
REVISION_STRENGTHS = ["Light", "Standard", "Heavy", "Surgical"]

DETAIL_MODES = [
    "Ask me for missing details",
    "Only use details already in the text",
    "Leave vague lines alone",
]

STRUCTURE_MODES = [
    "Keep current structure",
    "Lightly reshape structure",
    "Rewrite structure freely",
]

ARTIFACT_MODES = ["Off", "Safe cleanup", "Aggressive cleanup"]

ASSETS_DIR = Path(__file__).parent / "assets"


def inject_neobrutal_css() -> None:
    """Load the Neo-Brutalist skin; a missing file degrades to the base theme."""
    css_path = ASSETS_DIR / "neobrutal.css"
    if css_path.exists():
        st.markdown(f"<style>{css_path.read_text()}</style>", unsafe_allow_html=True)


st.set_page_config(page_title="Rewrite Rescue", page_icon="🛠️", layout="wide")
inject_neobrutal_css()
st.title("Rewrite Rescue")
st.caption(
    "Neutral prose naturalization: diagnose linguistic patterns, revise carefully, "
    "preserve meaning, and invent nothing."
)

with st.sidebar:
    st.subheader("Run configuration")

    provider_name = st.radio(
        "Provider",
        ["Demo (offline)", "BYO API endpoint"],
        help=(
            "Demo runs with no API call. BYO API endpoint sends the text to a "
            "model endpoint you provide for live rewriting."
        ),
    )

    byo_url = ""
    byo_model = ""
    byo_timeout = 60

    if provider_name == "Demo (offline)":
        st.info(
            "Demo mode is a fixed offline sanity check. It can show the app flow, "
            "but it does not use style samples, register, or revision strength. "
            "Use BYO API endpoint for live style-sensitive rewriting."
        )

    if provider_name == "BYO API endpoint":
        byo_url = st.text_input(
            "Base URL",
            placeholder="https://api.example-provider.com/v1",
            help="Use an OpenAI-compatible chat-completions base URL.",
        )
        byo_model = st.text_input(
            "Model name",
            placeholder="provider-model-name",
        )
        st.text_input(
            "API key",
            type="password",
            key="byo_api_key",
            help="Stored only in this Streamlit session.",
        )
        byo_timeout = st.number_input("Timeout seconds", 5, 300, 60)

    st.divider()

    text_type = st.selectbox(
        "Text type",
        TEXT_TYPES,
        index=0,
        help="Helps the live model preserve the right kind of structure and register.",
    )
    custom_text_type = ""
    if text_type == "Custom":
        custom_text_type = st.text_input("Custom text type")

    output_mode = st.radio(
        "Output mode",
        OUTPUT_MODES,
        index=0,
        help=(
            "Diagnose + revise shows both findings and rewritten text. Revise only hides "
            "the findings. Diagnose only never rewrites."
        ),
    )

    revision_strength = st.radio(
        "Revision strength",
        REVISION_STRENGTHS,
        index=1,
        help=(
            "Light = minimal edits. Standard = balanced. Heavy = more structural reshaping. "
            "Surgical = only touch the flagged parts."
        ),
    )

    register = st.selectbox(
        "Register",
        REGISTERS,
        index=0,
        help="The live model uses this to keep the writing formal, plain, technical, etc.",
    )
    custom_register = ""
    if register == "Custom":
        custom_register = st.text_area("Custom register instruction", height=80)

    style_sample_disabled = provider_name == "Demo (offline)"
    style_sample = st.text_area(
        "Style sample",
        height=120,
        placeholder=(
            "Optional. Paste a sample whose rhythm, syntax, and stance should guide "
            "the live rewrite."
        ),
        disabled=style_sample_disabled,
        help=(
            "Live BYO mode uses this as an authorial profile. Demo mode ignores it "
            "because no model call is made."
        ),
    )

    st.divider()

    specificity_mode = st.radio(
        "When the text needs a real detail",
        DETAIL_MODES,
        index=0,
        help=(
            "Ask me = show a question when the text needs a real fact. "
            "Only use details already in the text = sharpen without adding anything. "
            "Leave vague lines alone = preserve abstractions instead of rewriting them."
        ),
    )

    structure_mode = st.radio(
        "Structure handling",
        STRUCTURE_MODES,
        index=1,
        help=(
            "Controls whether the live model keeps the paragraph/list structure or reshapes it."
        ),
    )

    artifact_cleanup = st.select_slider(
        "Pasted-junk cleanup",
        ARTIFACT_MODES,
        value="Safe cleanup",
        help=(
            "Cleans obvious non-prose residue such as placeholder names, broken citation "
            "markers, stray Markdown, copied interface text, or tool-added link junk."
        ),
    )

    audit_mode = st.radio(
        "Audit level",
        ["Standard", "Strict"],
        index=0,
        help=(
            "Strict mode asks the live model to be tougher about meaning drift "
            "and overcorrection."
        ),
    )

    preserve_quotes = st.checkbox(
        "Preserve quoted material",
        value=True,
        help="Keeps quoted text unchanged unless it contains obvious pasted junk.",
    )
    preserve_citations = st.checkbox(
        "Preserve citations and URLs",
        value=True,
        help="Keeps links and citations unless cleanup safely removes obvious tool-added junk.",
    )
    st.checkbox(
        "Do not invent specifics",
        value=True,
        disabled=True,
        help="Always on. Missing real details become questions, not made-up facts.",
    )

    if st.button("Clear session", use_container_width=True):
        st.session_state.clear()
        st.rerun()

text = st.text_area(
    "Text to review",
    height=280,
    placeholder="Paste English prose here…",
)

char_count = len(text)
if char_count > 30_000:
    st.warning(
        f"{char_count:,} characters. Long-document chunking is not part of v1; "
        "shorter sections will usually produce better results."
    )
elif char_count > 10_000:
    st.info(f"{char_count:,} characters. Consider running long text in sections.")
elif char_count:
    st.caption(f"{char_count:,} characters")
else:
    st.caption("Paste text, or leave blank to run the built-in demo sample.")

config = {
    "provider": provider_name,
    "text_type": custom_text_type or text_type,
    "output_mode": output_mode,
    "revision_strength": revision_strength,
    "register": custom_register or register,
    "style_sample": style_sample,
    "specificity_mode": specificity_mode,
    "structure_mode": structure_mode,
    "artifact_cleanup": artifact_cleanup,
    "audit_mode": audit_mode,
    "preserve_quotes": preserve_quotes,
    "preserve_citations": preserve_citations,
    "do_not_invent_specifics": True,
}

if st.button("Run", type="primary"):
    source = text.strip() or SAMPLE_TEXT

    if not text.strip():
        st.info("No text pasted — running the built-in sample through the pipeline.")

    try:
        if output_mode == "Diagnose only":
            report = diagnose(source)
            result = diagnose_only_result(source, report)
        else:
            if provider_name == "BYO API endpoint":
                if not byo_url.strip() or not byo_model.strip():
                    st.error("Enter a base URL and model name, or switch to Demo.")
                    st.stop()
                adapter = BYOAdapter(
                    base_url=byo_url.strip(),
                    model=byo_model.strip(),
                    api_key=st.session_state.get("byo_api_key", ""),
                    timeout=float(byo_timeout),
                )
            else:
                adapter = DemoAdapter()

            report, revision = run_revision(source, config, adapter)
            result = revision_result(source, report, revision, output_mode)

    except ProviderError as err:
        st.error(f"Provider error: {err}")
    else:
        st.session_state["last_result"] = result

if "last_result" in st.session_state:
    render_output(
        st.session_state["last_result"],
        st.session_state["last_result"].get("changes") or [],
    )
else:
    st.caption("Run the app to see revised text, diagnostics, changes, and audit notes.")