Create app.py

app.py

@@ -1,27 +1,25 @@
 import gradio as gr
 import hashlib
-import zipfile
 import io
+import zipfile
 from datetime import datetime
 from urllib.parse import quote_plus, urlparse
 from collections import Counter
 
 import plotly.graph_objects as go
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
 
 from bluebook import bluebook_full
-from semantic import SemanticIndex, FAISS_AVAILABLE
+from appendix import build_litigation_appendix
 
 # ======================================================
 # CONFIG
 # ======================================================
 
 ENABLE_AI = True
-ENABLE_JOURNALIST_ZIP = True
+ENABLE_PUBLIC_SHARE = True
 
 # ======================================================
-# FOIA ADAPTER
+# ADAPTER
 # ======================================================
 
 class FOIAAdapter:
@@ -44,15 +42,14 @@ class FBI(FOIAAdapter):
     agency = "FBI"
     search_url = "https://vault.fbi.gov/search?SearchableText={q}"
 
-LIVE_ADAPTERS = [CIA(), FBI()]
+LIVE = [CIA(), FBI()]
 
 # ======================================================
 # STATE
 # ======================================================
 
 RESULTS = []
-SEMANTIC = None
-SELECTED = None
+SHARES = {}
 
 # ======================================================
 # SEARCH
@@ -63,64 +60,94 @@ def run_search(query):
     RESULTS = []
     rows = []
 
-    for a in LIVE_ADAPTERS:
+    for a in LIVE:
         for r in a.search(query):
             r["hash"] = hashlib.sha256(r["url"].encode()).hexdigest()[:16]
             RESULTS.append(r)
             rows.append([r["agency"], r["title"], r["url"], r["hash"]])
 
-    return rows
+    return rows, render_cards()
 
 # ======================================================
-# AI ASK
+# CARDS
 # ======================================================
 
-def ask_ai(opt_in, question):
-    if not opt_in or not SELECTED:
-        return "AI disabled or no document selected."
+def render_cards():
+    cards = []
+    for i, r in enumerate(RESULTS):
+        cards.append(f"""
+        <div style="border:1px solid #ddd;border-radius:12px;padding:14px;margin-bottom:16px">
+          <b>{r['agency']}</b><br>
+          {r['title']}<br><br>
+          <a href="{r['url']}" target="_blank">View</a> |
+          <a href="{r['url']}" download>Download</a> |
+          <a href="#" onclick="share({i})">Share</a> |
+          <i>Ask AI (opt-in)</i>
+        </div>
+        """)
+    return "".join(cards)
+
+# ======================================================
+# SHARE PAGE
+# ======================================================
 
-    r = SELECTED
-    return (
-        f"AI ANALYSIS\n\n"
-        f"{r['title']}\n{r['url']}\n\n"
-        f"Question:\n{question}\n\n"
-        f"Analysis is informational only.\n\n"
-        "AI DISCLOSURE:\n"
-        "• User-initiated\n"
-        "• Public FOIA documents only\n"
-        "• Verify against original source\n"
-    )
+def create_share():
+    sid = hashlib.sha256(str(RESULTS).encode()).hexdigest()[:12]
+    SHARES[sid] = RESULTS.copy()
+    return f"Share ID: `{sid}`"
+
+def load_share(sid):
+    recs = SHARES.get(sid)
+    if not recs:
+        return "Invalid share ID"
+    return "\n".join(bluebook_full(r) for r in recs)
 
 # ======================================================
-# EXPORT
+# EXPORTS
 # ======================================================
 
 def journalist_zip():
     buf = io.BytesIO()
     with zipfile.ZipFile(buf, "w") as z:
         z.writestr("citations.txt", "\n".join(bluebook_full(r) for r in RESULTS))
+        z.writestr(
+            "links.csv",
+            "agency,title,url\n" +
+            "\n".join(f"{r['agency']},{r['title']},{r['url']}" for r in RESULTS)
+        )
     buf.seek(0)
     return buf
 
+def appendix_pdf():
+    return build_litigation_appendix(RESULTS)
+
 # ======================================================
 # UI
 # ======================================================
 
 with gr.Blocks(title="Federal FOIA Intelligence Search") as app:
+    gr.Markdown("""
+    # 🏛️ Federal FOIA Intelligence Search
+    **Public FOIA Electronic Reading Rooms**
+    """)
+
     with gr.Tabs():
-        with gr.Tab("Search"):
-            q = gr.Textbox()
+        with gr.Tab("🔍 Search"):
+            q = gr.Textbox(label="Search FOIA Libraries")
             btn = gr.Button("Search")
             table = gr.Dataframe(headers=["Agency","Title","URL","Hash"])
-            btn.click(run_search, q, table)
+            gallery = gr.HTML()
+            btn.click(run_search, q, [table, gallery])
+
+        with gr.Tab("📤 Share"):
+            gr.Button("Create Share Page").click(create_share, outputs=gr.Textbox())
+            sid = gr.Textbox(label="Load Share ID")
+            gr.Button("Load").click(load_share, sid, gr.Textbox(lines=10))
 
-        with gr.Tab("AI Ask"):
-            opt = gr.Checkbox(label="Enable AI (Opt-In)")
-            question = gr.Textbox(lines=4)
-            answer = gr.Textbox(lines=12)
-            gr.Button("Ask AI").click(ask_ai, [opt, question], answer)
+        with gr.Tab("⚖️ Litigation"):
+            gr.Button("Generate Appendix PDF").click(appendix_pdf, outputs=gr.File())
 
-        with gr.Tab("Exports"):
+        with gr.Tab("🗂 Exports"):
             gr.Button("Journalist ZIP").click(journalist_zip, outputs=gr.File())
 
 app.launch()