Update app.py

app.py

@@ -1,351 +1,564 @@
 import gradio as gr
-import anthropic
+import json
 import os
+import time
+from pathlib import Path
+from huggingface_hub import InferenceClient
+
+try:
+    from pypdf import PdfReader
+except ImportError:
+    PdfReader = None
+
+SAVE_DIR = Path("/tmp/saved_docs")
+SAVE_DIR.mkdir(exist_ok=True)
+
+MODEL = "meta-llama/Llama-3.3-70B-Instruct"
+
+# ── Client ────────────────────────────────────────────────────────────────────
 
-# ── Anthropic client (key from HF Space secret) ──────────────────────────────
 def get_client():
-    api_key = os.environ.get("ANTHROPIC_API_KEY", "")
-    if not api_key:
-        raise gr.Error("⚠️  ANTHROPIC_API_KEY secret not set. Add it in Space Settings → Secrets.")
-    return anthropic.Anthropic(api_key=api_key)
+    token = os.environ.get("HF_TOKEN", "")
+    return InferenceClient(model=MODEL, token=token if token else None)
+
+
+# ── PDF extraction ────────────────────────────────────────────────────────────
+
+def extract_pdf_text(path: str) -> str:
+    if not path:
+        return ""
+    if PdfReader is None:
+        return "[pypdf not installed — PDF text unavailable]"
+    reader = PdfReader(path)
+    pages = []
+    for i, page in enumerate(reader.pages):
+        text = page.extract_text() or ""
+        if text.strip():
+            pages.append(f"--- Page {i+1} ---\n{text.strip()}")
+    full = "\n\n".join(pages)
+    # Truncate to ~12000 chars to stay within context limits
+    if len(full) > 12000:
+        full = full[:12000] + "\n\n[...document truncated for context limit...]"
+    return full
+
+
+def inject_pdf(prompt: str, pdf_path: str | None) -> str:
+    if not pdf_path:
+        return prompt
+    text = extract_pdf_text(pdf_path)
+    if not text.strip():
+        return prompt
+    return f"Use the following document as your primary reference:\n\n{text}\n\n---\n\n{prompt}"
+
+
+# ── Llama call ────────────────────────────────────────────────────────────────
+
+def call_llama(prompt: str, pdf_path: str | None = None, max_tokens: int = 1200) -> str:
+    full_prompt = inject_pdf(prompt, pdf_path)
+    client = get_client()
+    response = client.chat_completion(
+        messages=[{"role": "user", "content": full_prompt}],
+        max_tokens=max_tokens,
+        temperature=0.7,
+    )
+    return response.choices[0].message.content
+
+
+# ── Document library ──────────────────────────────────────────────────────────
+
+def load_library() -> list:
+    p = SAVE_DIR / "index.json"
+    return json.loads(p.read_text()) if p.exists() else []
+
+
+def save_to_library(title: str, technique: str, content: str) -> list:
+    if not title.strip():
+        title = f"{technique} — {time.strftime('%H:%M:%S')}"
+    library = load_library()
+    entry = {
+        "id": str(int(time.time() * 1000)),
+        "title": title.strip(),
+        "technique": technique,
+        "content": content,
+        "saved_at": time.strftime("%Y-%m-%d %H:%M"),
+    }
+    library.insert(0, entry)
+    (SAVE_DIR / "index.json").write_text(json.dumps(library, indent=2))
+    (SAVE_DIR / f"{entry['id']}.txt").write_text(
+        f"Title: {entry['title']}\nTechnique: {technique}\nSaved: {entry['saved_at']}\n\n{content}"
+    )
+    return library
+
+
+def delete_from_library(doc_id: str) -> list:
+    library = [e for e in load_library() if e["id"] != doc_id]
+    (SAVE_DIR / "index.json").write_text(json.dumps(library, indent=2))
+    fp = SAVE_DIR / f"{doc_id}.txt"
+    if fp.exists():
+        fp.unlink()
+    return library
+
+
+def lib_choices(library: list) -> list:
+    return [f"[{e['technique']}]  {e['title']}  ({e['saved_at']})" for e in library]
+
+
+def choice_to_entry(library: list, choice: str) -> dict | None:
+    choices = lib_choices(library)
+    try:
+        return library[choices.index(choice)]
+    except (ValueError, IndexError):
+        return None
 
 
 # ── Technique runners ─────────────────────────────────────────────────────────
 
-def run_perspective(persona: str, task: str, metaphor: str) -> str:
-    """Technique 1 – Perspective-Shifting Prompts"""
+def run_perspective(persona, task, metaphor, pdf_path):
     if not persona.strip() or not task.strip():
-        return "⚠️  Please fill in both the Persona and the Task fields."
-    prompt = f"Be a {persona} who explains {task}"
+        return "⚠️ Please fill in both Persona and Task."
+    prompt = f"You are a {persona}. Explain {task}"
     if metaphor.strip():
-        prompt += f" using {metaphor} metaphors"
-    prompt += ". Give a rich, vivid explanation."
-    client = get_client()
-    msg = client.messages.create(
-        model="claude-sonnet-4-20250514",
-        max_tokens=1000,
-        messages=[{"role": "user", "content": prompt}],
-    )
-    return msg.content[0].text
+        prompt += f" using {metaphor} metaphors exclusively"
+    prompt += ". Be vivid, specific, and deeply expert in your framing."
+    return call_llama(prompt, pdf_path)
 
 
-def run_constrained(ingredients: str, count: int) -> str:
-    """Technique 2 – Constrained Creativity Boosters"""
+def run_constrained(ingredients, count, pdf_path):
     if not ingredients.strip():
-        return "⚠️  Please enter at least one ingredient / constraint."
+        return "⚠️ Please enter at least one ingredient."
     items = [i.strip() for i in ingredients.split(",") if i.strip()]
     prompt = (
-        f"Suggest {int(count)} innovative startup ideas that MUST incorporate ALL of the "
-        f"following elements: {', '.join(items)}. "
-        "Be wildly creative — find non-obvious connections. "
-        "For each idea: name, one-line pitch, and why the combination is powerful."
+        f"Generate {int(count)} innovative startup ideas that MUST incorporate ALL of these elements: "
+        f"{', '.join(items)}.\n\n"
+        "Rules: Be wildly creative. Find non-obvious connections. "
+        "For each idea provide: (1) Company name, (2) One-line pitch, (3) Why this combination is uniquely powerful."
     )
-    client = get_client()
-    msg = client.messages.create(
-        model="claude-sonnet-4-20250514",
-        max_tokens=1000,
-        messages=[{"role": "user", "content": prompt}],
-    )
-    return msg.content[0].text
+    return call_llama(prompt, pdf_path)
 
 
-def run_meta(goal: str, criteria: str) -> str:
-    """Technique 3 – Meta-Prompts for Precision"""
+def run_meta(goal, criteria, pdf_path):
     if not goal.strip():
-        return "⚠️  Please enter a goal."
-    ranking_criteria = criteria if criteria.strip() else "1. Effectiveness  2. Cost  3. Ease of implementation"
+        return "⚠️ Please enter a goal."
+    ranking = criteria.strip() if criteria.strip() else "1. Scientific validity\n2. Cost-effectiveness\n3. Ease of implementation"
     prompt = (
-        f"First, suggest 3 concrete ways to: {goal}\n\n"
-        f"Then rank them by:\n{ranking_criteria}\n\n"
-        "Finally, critically identify 2–3 potential flaws in your own ranking system and "
-        "explain how those flaws could be mitigated."
-    )
-    client = get_client()
-    msg = client.messages.create(
-        model="claude-sonnet-4-20250514",
-        max_tokens=1200,
-        messages=[{"role": "user", "content": prompt}],
+        f"Step 1: Suggest 3 concrete ways to achieve this goal: {goal}\n\n"
+        f"Step 2: Rank them by these criteria:\n{ranking}\n\n"
+        "Step 3: Critically identify 2-3 flaws in your own ranking system above, "
+        "and explain specifically how each flaw could be mitigated."
     )
-    return msg.content[0].text
+    return call_llama(prompt, pdf_path, max_tokens=1400)
 
 
-def run_cot(problem: str, domain: str) -> str:
-    """Technique 4 – Chain-of-Thought Templates"""
+def run_cot(problem, domain, pdf_path):
     if not problem.strip():
-        return "⚠️  Please describe the problem."
-    domain_hint = f" (domain: {domain})" if domain.strip() else ""
+        return "⚠️ Please describe the problem."
+    domain_hint = f" in the {domain} domain" if domain.strip() else ""
     prompt = (
-        f"Let's solve this complex problem step by step{domain_hint}: {problem}\n\n"
+        f"Solve this problem step by step{domain_hint}: {problem}\n\n"
         "Follow this exact structure:\n"
-        "1. Break down the key components of the problem\n"
-        "2. Identify 3 potential failure points\n"
-        "3. Propose specific countermeasures for each failure point\n"
-        "4. Create a phased implementation roadmap\n"
-        "5. Predict 2 unexpected side effects (positive or negative) and how to handle them\n\n"
-        "Be thorough and specific at every step."
+        "1. COMPONENTS — Break down the key components of the problem\n"
+        "2. FAILURE POINTS — Identify 3 specific ways this could fail\n"
+        "3. COUNTERMEASURES — Propose a countermeasure for each failure point\n"
+        "4. ROADMAP — Create a phased implementation roadmap\n"
+        "5. SIDE EFFECTS — Predict 2 unexpected consequences (positive or negative) and how to handle them"
     )
-    client = get_client()
-    msg = client.messages.create(
-        model="claude-sonnet-4-20250514",
-        max_tokens=1400,
-        messages=[{"role": "user", "content": prompt}],
-    )
-    return msg.content[0].text
+    return call_llama(prompt, pdf_path, max_tokens=1600)
+
 
+def run_pdf_qa(question, pdf_path):
+    if not question.strip():
+        return "⚠��� Please enter a question."
+    if not pdf_path:
+        return "⚠️ Please upload a PDF first."
+    prompt = (
+        f"{question}\n\n"
+        "Answer thoroughly. Cite specific sections or page numbers from the document where relevant. "
+        "If the answer is not in the document, say so clearly."
+    )
+    return call_llama(prompt, pdf_path, max_tokens=1600)
+
+
+# ── Save handlers (one per tab — no lambdas to avoid closure bugs) ────────────
+
+def save_ps(title, content, lib):
+    if not content or not content.strip():
+        return lib, "⚠️ Nothing to save yet."
+    lib = save_to_library(title, "Perspective Shift", content)
+    return lib, "✅ Saved to library."
+
+def save_cc(title, content, lib):
+    if not content or not content.strip():
+        return lib, "⚠️ Nothing to save yet."
+    lib = save_to_library(title, "Constrained Creativity", content)
+    return lib, "✅ Saved to library."
+
+def save_mp(title, content, lib):
+    if not content or not content.strip():
+        return lib, "⚠️ Nothing to save yet."
+    lib = save_to_library(title, "Meta-Prompt", content)
+    return lib, "✅ Saved to library."
+
+def save_cot(title, content, lib):
+    if not content or not content.strip():
+        return lib, "⚠️ Nothing to save yet."
+    lib = save_to_library(title, "Chain of Thought", content)
+    return lib, "✅ Saved to library."
+
+def save_qa(title, content, lib):
+    if not content or not content.strip():
+        return lib, "⚠️ Nothing to save yet."
+    lib = save_to_library(title, "PDF Q&A", content)
+    return lib, "✅ Saved to library."
+
+
+# ── Library handlers ──────────────────────────────────────────────────────────
+
+def lib_refresh(lib):
+    return gr.update(choices=lib_choices(lib), value=None), ""
+
+def lib_view(sel, lib):
+    if not sel:
+        return ""
+    e = choice_to_entry(lib, sel)
+    if not e:
+        return "Document not found."
+    return f"TITLE: {e['title']}\nTECHNIQUE: {e['technique']}\nSAVED: {e['saved_at']}\n\n{'='*60}\n\n{e['content']}"
+
+def lib_download(sel, lib):
+    if not sel:
+        return gr.update(visible=False), "⚠️ Select a document first."
+    e = choice_to_entry(lib, sel)
+    if not e:
+        return gr.update(visible=False), "Not found."
+    fp = SAVE_DIR / f"{e['id']}.txt"
+    if fp.exists():
+        return gr.update(value=str(fp), visible=True), "✅ Ready to download."
+    return gr.update(visible=False), "File missing."
+
+def lib_delete(sel, lib):
+    if not sel:
+        return lib, gr.update(choices=lib_choices(lib), value=None), "⚠️ Select a document."
+    e = choice_to_entry(lib, sel)
+    if not e:
+        return lib, gr.update(choices=lib_choices(lib), value=None), "Not found."
+    lib = delete_from_library(e["id"])
+    return lib, gr.update(choices=lib_choices(lib), value=None), "🗑️ Deleted."
+
+def lib_sync(lib):
+    return gr.update(choices=lib_choices(lib))
+
+
+# ── CSS ───────────────────────────────────────────────────────────────────────
 
-# ── Custom CSS ────────────────────────────────────────────────────────────────
 CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Syne:wght@400;700;800&family=DM+Mono:wght@400;500&display=swap');
+@import url('https://fonts.googleapis.com/css2?family=Syne:wght@400;700;800&family=DM+Mono:ital,wght@0,400;0,500;1,400&display=swap');
 
 :root {
-    --bg: #0b0c10;
-    --surface: #13151a;
-    --border: #1e2230;
-    --accent: #c8f135;
-    --accent2: #35f1c8;
-    --text: #e8eaf0;
-    --muted: #6b7280;
-    --radius: 10px;
+    --bg: #0d0f14;
+    --surface: #151820;
+    --surface2: #1c1f2b;
+    --border: #252836;
+    --accent: #7DF9AA;
+    --accent2: #7DB3F9;
+    --accent3: #F9C97D;
+    --red: #F97D7D;
+    --text: #E2E6F0;
+    --muted: #5A6070;
+    --r: 8px;
 }
 
+*, *::before, *::after { box-sizing: border-box; }
+
 body, .gradio-container {
     background: var(--bg) !important;
     font-family: 'Syne', sans-serif !important;
     color: var(--text) !important;
 }
 
-/* Header */
-.header-wrap {
+/* ── Header ── */
+.hdr {
     text-align: center;
-    padding: 2.5rem 1rem 1.5rem;
+    padding: 2.8rem 1rem 1.8rem;
     border-bottom: 1px solid var(--border);
-    margin-bottom: 1.5rem;
+    margin-bottom: 1.8rem;
+    background: radial-gradient(ellipse at 50% 0%, rgba(125,249,170,0.06) 0%, transparent 70%);
 }
-.header-wrap h1 {
-    font-size: 2.6rem;
+.hdr h1 {
+    font-size: 2.4rem;
     font-weight: 800;
-    letter-spacing: -0.03em;
+    letter-spacing: -0.04em;
     background: linear-gradient(135deg, var(--accent) 0%, var(--accent2) 100%);
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
-    margin: 0;
-}
-.header-wrap p {
-    color: var(--muted);
-    font-family: 'DM Mono', monospace;
-    font-size: 0.85rem;
-    margin-top: 0.5rem;
+    margin: 0 0 0.4rem;
 }
+.hdr p { color: var(--muted); font-family: 'DM Mono', monospace; font-size: 0.82rem; margin: 0; }
 
-/* Tabs */
+/* ── Tabs ── */
 .tab-nav button {
     background: transparent !important;
     color: var(--muted) !important;
     border: 1px solid transparent !important;
-    border-radius: var(--radius) !important;
+    border-radius: var(--r) !important;
     font-family: 'Syne', sans-serif !important;
     font-weight: 700 !important;
-    font-size: 0.9rem !important;
-    padding: 0.5rem 1.2rem !important;
-    transition: all 0.2s !important;
+    font-size: 0.82rem !important;
+    padding: 0.45rem 0.9rem !important;
+    transition: all 0.18s !important;
 }
 .tab-nav button.selected {
-    background: var(--surface) !important;
+    background: var(--surface2) !important;
     color: var(--accent) !important;
     border-color: var(--border) !important;
 }
-.tab-nav button:hover {
-    color: var(--text) !important;
-}
+.tab-nav button:hover { color: var(--text) !important; }
 
-/* Inputs & Textareas */
-input[type="text"], textarea, .gr-textbox textarea {
+/* ── Inputs ── */
+input[type=text], textarea {
     background: var(--surface) !important;
     border: 1px solid var(--border) !important;
-    border-radius: var(--radius) !important;
+    border-radius: var(--r) !important;
     color: var(--text) !important;
     font-family: 'DM Mono', monospace !important;
-    font-size: 0.88rem !important;
-    padding: 0.65rem 0.9rem !important;
-    transition: border-color 0.2s !important;
+    font-size: 0.86rem !important;
+    padding: 0.6rem 0.85rem !important;
+    transition: border-color 0.18s, box-shadow 0.18s !important;
+    resize: vertical !important;
 }
-input[type="text"]:focus, textarea:focus {
+input[type=text]:focus, textarea:focus {
     border-color: var(--accent) !important;
     outline: none !important;
-    box-shadow: 0 0 0 2px rgba(200, 241, 53, 0.12) !important;
+    box-shadow: 0 0 0 3px rgba(125,249,170,0.1) !important;
 }
 
-/* Labels */
-label span, .gr-form label {
-    color: var(--text) !important;
-    font-family: 'Syne', sans-serif !important;
-    font-weight: 700 !important;
-    font-size: 0.82rem !important;
-    letter-spacing: 0.06em !important;
+/* ── Labels ── */
+label > span:first-child {
+    color: var(--muted) !important;
+    font-family: 'DM Mono', monospace !important;
+    font-weight: 500 !important;
+    font-size: 0.75rem !important;
+    letter-spacing: 0.08em !important;
     text-transform: uppercase !important;
 }
 
-/* Buttons */
+/* ── Buttons ── */
 button.primary {
     background: var(--accent) !important;
-    color: #0b0c10 !important;
+    color: #0d0f14 !important;
     border: none !important;
-    border-radius: var(--radius) !important;
+    border-radius: var(--r) !important;
     font-family: 'Syne', sans-serif !important;
     font-weight: 800 !important;
-    font-size: 0.9rem !important;
-    letter-spacing: 0.05em !important;
-    padding: 0.65rem 1.6rem !important;
+    font-size: 0.88rem !important;
+    letter-spacing: 0.04em !important;
+    padding: 0.6rem 1.5rem !important;
+    cursor: pointer !important;
+    transition: opacity 0.15s, transform 0.15s !important;
+}
+button.primary:hover { opacity: 0.85 !important; transform: translateY(-1px) !important; }
+
+button.secondary {
+    background: var(--surface2) !important;
+    color: var(--accent2) !important;
+    border: 1px solid var(--border) !important;
+    border-radius: var(--r) !important;
+    font-family: 'Syne', sans-serif !important;
+    font-weight: 700 !important;
+    font-size: 0.8rem !important;
+    padding: 0.5rem 1rem !important;
     cursor: pointer !important;
-    transition: transform 0.15s, opacity 0.15s !important;
+    transition: border-color 0.15s !important;
 }
-button.primary:hover { opacity: 0.88 !important; transform: translateY(-1px) !important; }
+button.secondary:hover { border-color: var(--accent2) !important; }
 
-/* Output box */
-.output-box textarea {
-    background: #0f1117 !important;
+/* ── Output box ── */
+.out textarea {
+    background: #0a0c10 !important;
     border: 1px solid var(--border) !important;
-    border-left: 3px solid var(--accent2) !important;
+    border-left: 3px solid var(--accent) !important;
+    border-radius: 0 var(--r) var(--r) 0 !important;
     font-family: 'DM Mono', monospace !important;
-    font-size: 0.86rem !important;
-    line-height: 1.65 !important;
+    font-size: 0.85rem !important;
+    line-height: 1.7 !important;
     color: var(--text) !important;
-    border-radius: 0 var(--radius) var(--radius) 0 !important;
 }
 
-/* Technique cards */
-.technique-card {
+/* ── Cards ── */
+.card {
     background: var(--surface);
     border: 1px solid var(--border);
-    border-radius: var(--radius);
-    padding: 1rem 1.2rem;
+    border-radius: var(--r);
+    padding: 0.85rem 1rem;
     margin-bottom: 1rem;
 }
-.technique-card .badge {
+.badge {
     display: inline-block;
     font-family: 'DM Mono', monospace;
-    font-size: 0.72rem;
-    font-weight: 500;
-    letter-spacing: 0.08em;
+    font-size: 0.68rem;
+    letter-spacing: 0.09em;
     text-transform: uppercase;
-    padding: 0.2rem 0.6rem;
+    padding: 0.18rem 0.55rem;
     border-radius: 4px;
-    background: rgba(200,241,53,0.12);
+    background: rgba(125,249,170,0.1);
     color: var(--accent);
-    margin-bottom: 0.4rem;
+    margin-bottom: 0.3rem;
 }
-.technique-card p {
-    color: var(--muted);
-    font-size: 0.84rem;
-    margin: 0;
-    line-height: 1.55;
+.card p { color: var(--muted); font-size: 0.82rem; margin: 0; line-height: 1.55; }
+
+.pdf-note {
+    display: inline-block;
+    font-family: 'DM Mono', monospace;
+    font-size: 0.68rem;
+    letter-spacing: 0.08em;
+    text-transform: uppercase;
+    padding: 0.18rem 0.55rem;
+    border-radius: 4px;
+    background: rgba(249,201,125,0.1);
+    color: var(--accent3);
+    margin: 0.6rem 0 0.3rem;
 }
 
-/* Slider */
-input[type="range"] { accent-color: var(--accent) !important; }
+/* ── Slider ── */
+input[type=range] { accent-color: var(--accent) !important; }
 
-/* Footer */
-.footer-note {
+/* ── File upload ── */
+.file-upload {
+    background: var(--surface) !important;
+    border: 1.5px dashed var(--border) !important;
+    border-radius: var(--r) !important;
+    transition: border-color 0.18s !important;
+}
+.file-upload:hover { border-color: var(--accent3) !important; }
+
+/* ── Footer ── */
+.foot {
     text-align: center;
     padding: 1.5rem;
     color: var(--muted);
     font-family: 'DM Mono', monospace;
-    font-size: 0.75rem;
+    font-size: 0.7rem;
     border-top: 1px solid var(--border);
     margin-top: 2rem;
 }
 """
 
 # ── UI ────────────────────────────────────────────────────────────────────────
+
 with gr.Blocks(css=CSS, title="Prompt Engineering Toolkit") as demo:
 
+    lib_state = gr.State(load_library())
+
     gr.HTML("""
-    <div class="header-wrap">
+    <div class="hdr">
         <h1>⚡ Prompt Engineering Toolkit</h1>
-        <p>4 battle-tested techniques for precision AI output</p>
+        <p>llama-3.3-70b-instruct &nbsp;·&nbsp; 4 techniques &nbsp;·&nbsp; PDF intake &nbsp;·&nbsp; document library</p>
     </div>
     """)
 
     with gr.Tabs():
 
-        # ── Tab 1: Perspective-Shifting ───────────────────────────────────────
+        # ── 01 Perspective Shift ──────────────────────────────────────────────
         with gr.Tab("01 · Perspective Shift"):
-            gr.HTML("""
-            <div class="technique-card">
-                <div class="badge">Formula: Be X who does Y using Z metaphors</div>
-                <p>Forces the AI to adopt a specific expert persona, shaping tone, depth, and vocabulary.
-                The metaphor layer bridges complex domains into intuitive language.</p>
-            </div>
-            """)
+            gr.HTML('<div class="card"><div class="badge">Be X · explain Y · using Z metaphors</div><p>Forces the model into an expert persona. Add a PDF to ground the response in your document.</p></div>')
             with gr.Row():
                 ps_persona = gr.Textbox(label="Persona (X)", placeholder="e.g. neurosurgeon, jazz musician, Viking navigator")
-                ps_task    = gr.Textbox(label="Explain / Do (Y)", placeholder="e.g. quantum entanglement, blockchain consensus")
-            ps_meta   = gr.Textbox(label="Metaphor Domain (Z) — optional", placeholder="e.g. cooking, football, carpentry")
-            ps_btn    = gr.Button("Generate →", variant="primary")
-            ps_out    = gr.Textbox(label="Output", lines=14, elem_classes=["output-box"])
-            ps_btn.click(run_perspective, inputs=[ps_persona, ps_task, ps_meta], outputs=ps_out)
+                ps_task    = gr.Textbox(label="Explain / Do (Y)", placeholder="e.g. quantum entanglement, compound interest")
+            ps_meta = gr.Textbox(label="Metaphor Domain (Z) — optional", placeholder="e.g. cooking, football, carpentry")
+            gr.HTML('<div class="pdf-note">📄 optional pdf context</div>')
+            ps_pdf  = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath", elem_classes=["file-upload"])
+            ps_btn  = gr.Button("Generate →", variant="primary")
+            ps_out  = gr.Textbox(label="Output", lines=14, elem_classes=["out"])
+            with gr.Row():
+                ps_title = gr.Textbox(label="Save as (title) — blank = auto", scale=4)
+                ps_save  = gr.Button("💾 Save", variant="secondary", scale=1)
+            ps_msg = gr.Markdown("")
+            ps_btn.click(run_perspective, inputs=[ps_persona, ps_task, ps_meta, ps_pdf], outputs=ps_out)
+            ps_save.click(save_ps, inputs=[ps_title, ps_out, lib_state], outputs=[lib_state, ps_msg])
 
-        # ── Tab 2: Constrained Creativity ────────────────────────────────────
+        # ── 02 Constrained Creativity ─────────────────────────────────────────
         with gr.Tab("02 · Constrained Creativity"):
-            gr.HTML("""
-            <div class="technique-card">
-                <div class="badge">Concept: Limitations force innovation</div>
-                <p>By restricting the solution space to specific ingredients, the AI must find
-                non-obvious creative bridges — generating ideas that would never emerge from open-ended prompts.</p>
-            </div>
-            """)
-            cc_items = gr.Textbox(
-                label="Ingredients / Constraints (comma-separated)",
-                placeholder="e.g. 3D printers, seaweed, TikTok",
-            )
-            cc_count = gr.Slider(label="Number of Ideas", minimum=2, maximum=8, value=5, step=1)
-            cc_btn   = gr.Button("Generate →", variant="primary")
-            cc_out   = gr.Textbox(label="Output", lines=18, elem_classes=["output-box"])
-            cc_btn.click(run_constrained, inputs=[cc_items, cc_count], outputs=cc_out)
-
-        # ── Tab 3: Meta-Prompts ───────────────────────────────────────────────
+            gr.HTML('<div class="card"><div class="badge">Limitations force innovation</div><p>Locks the model to specific ingredients — forcing non-obvious creative bridges between concepts.</p></div>')
+            cc_items = gr.Textbox(label="Ingredients (comma-separated)", placeholder="e.g. 3D printers, seaweed, TikTok")
+            cc_count = gr.Slider(label="Number of ideas", minimum=2, maximum=8, value=5, step=1)
+            gr.HTML('<div class="pdf-note">📄 optional pdf context</div>')
+            cc_pdf  = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath", elem_classes=["file-upload"])
+            cc_btn  = gr.Button("Generate →", variant="primary")
+            cc_out  = gr.Textbox(label="Output", lines=16, elem_classes=["out"])
+            with gr.Row():
+                cc_title = gr.Textbox(label="Save as (title) — blank = auto", scale=4)
+                cc_save  = gr.Button("💾 Save", variant="secondary", scale=1)
+            cc_msg = gr.Markdown("")
+            cc_btn.click(run_constrained, inputs=[cc_items, cc_count, cc_pdf], outputs=cc_out)
+            cc_save.click(save_cc, inputs=[cc_title, cc_out, lib_state], outputs=[lib_state, cc_msg])
+
+        # ── 03 Meta-Prompts ───────────────────────────────────────────────────
         with gr.Tab("03 · Meta-Prompts"):
-            gr.HTML("""
-            <div class="technique-card">
-                <div class="badge">Concept: Make the AI critique its own output</div>
-                <p>Multi-stage prompting forces the model to suggest, rank, then audit its own
-                reasoning — surfacing blind spots and producing genuinely reliable recommendations.</p>
-            </div>
-            """)
-            mp_goal     = gr.Textbox(label="Goal / Challenge", placeholder="e.g. improve sleep quality, reduce team meeting time")
-            mp_criteria = gr.Textbox(
-                label="Ranking Criteria (one per line) — leave blank for defaults",
-                placeholder="1. Scientific validity\n2. Cost-effectiveness\n3. Ease of implementation",
-                lines=3,
-            )
-            mp_btn = gr.Button("Generate →", variant="primary")
-            mp_out = gr.Textbox(label="Output", lines=18, elem_classes=["output-box"])
-            mp_btn.click(run_meta, inputs=[mp_goal, mp_criteria], outputs=mp_out)
-
-        # ── Tab 4: Chain-of-Thought ───────────────────────────────────────────
-        with gr.Tab("04 · Chain of Thought"):
-            gr.HTML("""
-            <div class="technique-card">
-                <div class="badge">Concept: Force systematic reasoning</div>
-                <p>Breaks the problem into obligatory reasoning stages: decomposition → failure analysis →
-                countermeasures → roadmap → side-effect prediction. Prevents shallow or hand-wavy answers.</p>
-            </div>
-            """)
-            cot_problem = gr.Textbox(
-                label="Complex Problem",
-                placeholder="e.g. launching a product in a saturated market with zero ad budget",
-                lines=3,
-            )
-            cot_domain  = gr.Textbox(label="Domain Hint — optional", placeholder="e.g. B2B SaaS, healthcare, fintech")
-            cot_btn     = gr.Button("Generate →", variant="primary")
-            cot_out     = gr.Textbox(label="Output", lines=22, elem_classes=["output-box"])
-            cot_btn.click(run_cot, inputs=[cot_problem, cot_domain], outputs=cot_out)
+            gr.HTML('<div class="card"><div class="badge">Make the model critique its own output</div><p>Three-stage: suggest → rank → audit own reasoning. Surfaces blind spots automatically.</p></div>')
+            mp_goal = gr.Textbox(label="Goal / Challenge", placeholder="e.g. improve sleep quality, reduce team meeting time")
+            mp_crit = gr.Textbox(label="Ranking criteria (one per line) — blank = defaults", lines=3,
+                                  placeholder="1. Scientific validity\n2. Cost-effectiveness\n3. Ease of implementation")
+            gr.HTML('<div class="pdf-note">📄 optional pdf context</div>')
+            mp_pdf  = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath", elem_classes=["file-upload"])
+            mp_btn  = gr.Button("Generate →", variant="primary")
+            mp_out  = gr.Textbox(label="Output", lines=16, elem_classes=["out"])
+            with gr.Row():
+                mp_title = gr.Textbox(label="Save as (title) — blank = auto", scale=4)
+                mp_save  = gr.Button("💾 Save", variant="secondary", scale=1)
+            mp_msg = gr.Markdown("")
+            mp_btn.click(run_meta, inputs=[mp_goal, mp_crit, mp_pdf], outputs=mp_out)
+            mp_save.click(save_mp, inputs=[mp_title, mp_out, lib_state], outputs=[lib_state, mp_msg])
 
-    gr.HTML("""
-    <div class="footer-note">
-        Powered by Claude · claude-sonnet-4-20250514 · Requires ANTHROPIC_API_KEY secret
-    </div>
-    """)
+        # ── 04 Chain of Thought ───────────────────────────────────────────────
+        with gr.Tab("04 · Chain of Thought"):
+            gr.HTML('<div class="card"><div class="badge">Force systematic reasoning</div><p>Decompose → failure analysis → countermeasures → roadmap → side-effect prediction. No hand-waving.</p></div>')
+            cot_prob = gr.Textbox(label="Complex Problem", lines=3, placeholder="e.g. launching a product in a saturated market with zero ad budget")
+            cot_dom  = gr.Textbox(label="Domain hint — optional", placeholder="e.g. B2B SaaS, healthcare, fintech")
+            gr.HTML('<div class="pdf-note">📄 optional pdf context</div>')
+            cot_pdf  = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath", elem_classes=["file-upload"])
+            cot_btn  = gr.Button("Generate →", variant="primary")
+            cot_out  = gr.Textbox(label="Output", lines=20, elem_classes=["out"])
+            with gr.Row():
+                cot_title = gr.Textbox(label="Save as (title) — blank = auto", scale=4)
+                cot_save  = gr.Button("💾 Save", variant="secondary", scale=1)
+            cot_msg = gr.Markdown("")
+            cot_btn.click(run_cot, inputs=[cot_prob, cot_dom, cot_pdf], outputs=cot_out)
+            cot_save.click(save_cot, inputs=[cot_title, cot_out, lib_state], outputs=[lib_state, cot_msg])
+
+        # ── 05 PDF Q&A ────────────────────────────────────────────────────────
+        with gr.Tab("05 · PDF Q&A"):
+            gr.HTML('<div class="card"><div class="badge">Direct document interrogation</div><p>Upload any PDF — contract, paper, report, case file. Ask anything; the model cites specific sections.</p></div>')
+            qa_pdf = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath", elem_classes=["file-upload"])
+            qa_q   = gr.Textbox(label="Question", lines=3,
+                                 placeholder="e.g. What are the key obligations in section 4?\nSummarise the main argument.\nWhat risks does the author identify?")
+            qa_btn = gr.Button("Ask →", variant="primary")
+            qa_out = gr.Textbox(label="Answer", lines=20, elem_classes=["out"])
+            with gr.Row():
+                qa_title = gr.Textbox(label="Save as (title) — blank = auto", scale=4)
+                qa_save  = gr.Button("💾 Save", variant="secondary", scale=1)
+            qa_msg = gr.Markdown("")
+            qa_btn.click(run_pdf_qa, inputs=[qa_q, qa_pdf], outputs=qa_out)
+            qa_save.click(save_qa, inputs=[qa_title, qa_out, lib_state], outputs=[lib_state, qa_msg])
+
+        # ── 06 Library ────────────────────────────────────────────────────────
+        with gr.Tab("📚 Library"):
+            gr.HTML('<div class="card"><div class="badge">Session document library</div><p>Every output you\'ve saved this session. View, download as .txt, or delete.</p></div>')
+
+            lib_refresh_btn = gr.Button("���� Refresh list", variant="secondary")
+            lib_dd    = gr.Dropdown(label="Saved documents", choices=lib_choices(load_library()), value=None, interactive=True)
+            lib_viewer = gr.Textbox(label="Content", lines=20, elem_classes=["out"], interactive=False)
+            with gr.Row():
+                lib_dl_btn  = gr.Button("⬇️ Download .txt", variant="secondary")
+                lib_del_btn = gr.Button("🗑️ Delete", variant="secondary")
+            lib_dl_file = gr.File(label="Your file", visible=False)
+            lib_msg     = gr.Markdown("")
+
+            lib_refresh_btn.click(lib_refresh,  inputs=[lib_state],               outputs=[lib_dd, lib_msg])
+            lib_dd.change(lib_view,             inputs=[lib_dd, lib_state],        outputs=lib_viewer)
+            lib_dl_btn.click(lib_download,      inputs=[lib_dd, lib_state],        outputs=[lib_dl_file, lib_msg])
+            lib_del_btn.click(lib_delete,       inputs=[lib_dd, lib_state],        outputs=[lib_state, lib_dd, lib_msg])
+            lib_state.change(lib_sync,          inputs=[lib_state],                outputs=lib_dd)
+
+    gr.HTML('<div class="foot">Llama-3.3-70B-Instruct via HF Inference API &nbsp;·&nbsp; Set HF_TOKEN secret for private/gated models &nbsp;·&nbsp; PDFs extracted with pypdf</div>')
 
 if __name__ == "__main__":
     demo.launch()