Update app.py

app.py

@@ -17,16 +17,14 @@ from fastapi.responses import JSONResponse, FileResponse
 # HARD GOVERNANCE FLAGS (NON-NEGOTIABLE)
 # ======================================================
 
-ENABLE_FAISS_PHASE_4 = False      # HARD DISABLED
-ENABLE_AI = True                 # USER OPT-IN ONLY
-ENABLE_PDF_EXTRACTION = True     # USER OPT-IN ONLY
+ENABLE_FAISS_PHASE_4 = False
+ENABLE_AI = True
 
 # ======================================================
-# SESSION STATE (EPHEMERAL)
+# SESSION STATE
 # ======================================================
 
 LAST_RESULTS = []
-SELECTED_INDEX = None
 
 # ======================================================
 # CRYPTOGRAPHIC CORE
@@ -42,27 +40,24 @@ def citation_hash(r):
 
 def provenance_headers(payload: str):
     return {
-        "Tool-Version": "1.7.1",
+        "Tool-Version": "1.7.2",
         "Generated-UTC": datetime.utcnow().isoformat(),
         "Content-SHA256": sha256_text(payload),
         "Public-Source-Only": "true",
-        "AI-Assisted": "formatting-only",
+        "AI-Assisted": "user-initiated-only",
     }
 
-def render_provenance_block(text: str):
-    return "\n".join(f"{k}: {v}" for k, v in provenance_headers(text).items())
-
 # ======================================================
 # FOIA ADAPTERS (LINK-OUT ONLY)
 # ======================================================
 
 class FOIAAdapter:
-    agency = "UNKNOWN"
+    agency = ""
     search_url = ""
 
-    def search(self, query):
+    def search(self, q):
         start = time.time()
-        url = self.search_url.format(q=quote_plus(query))
+        url = self.search_url.format(q=quote_plus(q))
         latency = round((time.time() - start) * 1000, 1)
         return [{
             "agency": self.agency,
@@ -85,37 +80,21 @@ class DOJ(FOIAAdapter):
     agency = "DOJ"
     search_url = "https://www.justice.gov/foia/library?search={q}"
 
-class DHS(FOIAAdapter):
-    agency = "DHS"
-    search_url = "https://www.dhs.gov/foia-library/search?search={q}"
-
-class STATE(FOIAAdapter):
-    agency = "State Department"
-    search_url = "https://foia.state.gov/Search/Search.aspx?q={q}"
-
-class NSA(FOIAAdapter):
-    agency = "NSA"
-    search_url = "https://www.nsa.gov/resources/everyone/foia/reading-room/?q={q}"
-
 ALL_ADAPTERS = {
     "CIA": CIA(),
     "FBI": FBI(),
     "DOJ": DOJ(),
-    "DHS": DHS(),
-    "State": STATE(),
-    "NSA": NSA(),
 }
 
 # ======================================================
-# PDF RESOLUTION (SAFE)
+# PDF RESOLUTION (SAFE HEAD REQUEST)
 # ======================================================
 
 def resolve_pdf_url(url):
     try:
-        r = requests.get(url, timeout=8, allow_redirects=True)
+        r = requests.head(url, timeout=8, allow_redirects=True)
         ct = r.headers.get("content-type", "").lower()
-        is_pdf = r.url.lower().endswith(".pdf") or "application/pdf" in ct
-        return is_pdf, r.url
+        return ("pdf" in ct or url.lower().endswith(".pdf")), r.url
     except Exception:
         return False, url
 
@@ -124,17 +103,16 @@ def resolve_pdf_url(url):
 # ======================================================
 
 def run_search(query, agencies):
-    global LAST_RESULTS, SELECTED_INDEX
-    SELECTED_INDEX = None
+    global LAST_RESULTS
     LAST_RESULTS = []
     rows = []
 
     for name in agencies:
-        adapter = ALL_ADAPTERS[name]
-        for r in adapter.search(query):
+        for r in ALL_ADAPTERS[name].search(query):
             r["resolved_pdf"], r["resolved_url"] = resolve_pdf_url(r["url"])
             r["hash"] = citation_hash(r)
             LAST_RESULTS.append(r)
+
             rows.append([
                 r["agency"],
                 r["title"],
@@ -143,7 +121,7 @@ def run_search(query, agencies):
                 f"{r['latency_ms']} ms",
             ])
 
-    return rows, render_cards(), "No document selected."
+    return rows, render_cards()
 
 # ======================================================
 # ASK-AI GOVERNANCE GATE
@@ -152,9 +130,9 @@ def run_search(query, agencies):
 def can_enable_ai(r):
     return (
         ENABLE_AI
-        and r.get("resolved_pdf", False)
-        and r.get("resolved_url", "").lower().endswith(".pdf")
-        and not r.get("sealed", False)
+        and r["resolved_pdf"]
+        and r["resolved_url"].lower().endswith(".pdf")
+        and not r["sealed"]
     )
 
 def ask_ai_for_document(index: int):
@@ -168,33 +146,31 @@ def ask_ai_for_document(index: int):
     }
 
 # ======================================================
-# RENDER RESULT CARDS
+# RESULT CARDS
 # ======================================================
 
 def render_cards():
-    cards = []
-    for idx, r in enumerate(LAST_RESULTS):
-        cards.append(f"""
+    html = []
+    for i, r in enumerate(LAST_RESULTS):
+        html.append(f"""
         <div class="card">
           <div class="card-header">
-            <strong>{r['agency']}</strong>
+            <b>{r['agency']}</b>
             <button class="ask-ai"
-              onclick="askAI({idx})"
+              onclick="askAI({i})"
               {"disabled" if not can_enable_ai(r) else ""}>
               Ask AI
             </button>
           </div>
-          <div><b>{r['title']}</b></div>
+          <div>{r['title']}</div>
           <div class="actions">
-            <a href="{r['resolved_url']}" target="_blank">View Source</a>
-            {"<a href='"+r['resolved_url']+"' target='_blank'>Preview PDF</a>" if r["resolved_pdf"] else ""}
-          </div>
-          <div class="helper">
-            Why am I seeing this? This is a publicly released FOIA document.
+            <a href="{r['resolved_url']}" target="_blank">View</a>
+            <a href="{r['resolved_url']}" download>Download</a>
+            <a href="{r['resolved_url']}" target="_blank">Share</a>
           </div>
         </div>
         """)
-    return "".join(cards) or "No results."
+    return "".join(html) or "No results."
 
 # ======================================================
 # COURT BUNDLE
@@ -206,7 +182,7 @@ def generate_court_bundle():
             for i, r in enumerate(LAST_RESULTS, 1):
                 z.writestr(
                     f"Exhibit_{i:03d}.txt",
-                    f"{r['agency']}\n{r['resolved_url']}\n\n{render_provenance_block(r['resolved_url'])}"
+                    f"{r['agency']}\n{r['resolved_url']}"
                 )
                 z.writestr(f"Exhibit_{i:03d}.sha256", r["hash"])
         return tf.name
@@ -228,44 +204,41 @@ def court_bundle():
     path = generate_court_bundle()
     return FileResponse(path, filename="court_bundle.zip")
 
-if os.path.exists("governance-site"):
-    fastapi_app.mount(
-        "/gov",
-        StaticFiles(directory="governance-site", html=True),
-        name="governance",
-    )
-
 # ======================================================
 # UI
 # ======================================================
 
 CSS = """
-.card {border:1px solid #2a2a2a;border-radius:18px;padding:18px;margin-bottom:22px;background:#0f0f0f;}
-.card-header {display:flex;justify-content:space-between;align-items:center;}
-.ask-ai {background:linear-gradient(135deg,#1e88e5,#1565c0);color:white;border:none;padding:6px 16px;border-radius:999px;font-weight:600;}
-.ask-ai:disabled {background:#555;}
-.helper {font-size:0.75rem;color:#aaa;margin-top:6px;}
-.actions {margin-top:12px;display:flex;gap:16px;}
+.card {border:1px solid #2a2a2a;border-radius:16px;padding:16px;margin-bottom:16px;}
+.card-header {display:flex;justify-content:space-between;}
+.ask-ai {background:#1e88e5;color:white;border:none;border-radius:999px;padding:6px 14px;}
+.actions {margin-top:8px;display:flex;gap:16px;}
 """
 
 JS = """
 function askAI(idx){
  fetch('/ask_ai?index=' + idx)
   .then(r=>r.json())
-  .then(d=>alert('AI Enabled for: ' + d.title));
+  .then(d=>alert('AI ready for: ' + d.title));
 }
 """
 
 with gr.Blocks() as gradio_ui:
     gr.Markdown("## Federal FOIA Intelligence Search")
-    agencies = gr.CheckboxGroup(choices=list(ALL_ADAPTERS.keys()), value=list(ALL_ADAPTERS.keys()))
-    query = gr.Textbox(placeholder="Search terms")
+    agencies = gr.CheckboxGroup(
+        choices=list(ALL_ADAPTERS.keys()),
+        value=list(ALL_ADAPTERS.keys())
+    )
+    query = gr.Textbox(placeholder="Search FOIA reading rooms")
     table = gr.Dataframe(headers=["Agency","Title","URL","Hash","Latency"])
     gallery = gr.HTML()
-    status = gr.Textbox(lines=4)
-    gr.Button("Search").click(run_search,[query,agencies],[table,gallery,status])
 
-    gr.Markdown("### Court")
+    gr.Button("Search").click(
+        run_search,
+        inputs=[query, agencies],
+        outputs=[table, gallery]
+    )
+
     gr.Button("Download Court Bundle").click(
         lambda: "/court_bundle",
         None,
@@ -282,6 +255,5 @@ if __name__ == "__main__":
     gradio_ui.launch(
         css=CSS,
         js=JS,
-        title="Federal FOIA Intelligence Search",
         show_error=True,
     )