Update app.py

app.py

@@ -368,13 +368,71 @@ async def oauth_logout(request: Request):
 
 @fapp.get("/health")
 async def health():
-    # Check SGLang
     try:
         r = requests.get(f"{SGLANG_BASE}/v1/models", timeout=5, verify=False)
         return {"status":"ok","sglang":"connected"}
     except:
         return {"status":"ok","sglang":"disconnected"}
 
+# ── Web Search API (Brave) ──────────────────────────────────────────────
+BRAVE_API_KEY = os.getenv("BRAVE_API_KEY", "")
+
+@fapp.post("/api/search")
+async def api_search(request: Request):
+    body = await request.json()
+    query = body.get("query", "").strip()
+    if not query:
+        return JSONResponse({"error": "empty query"}, status_code=400)
+    key = BRAVE_API_KEY
+    if not key:
+        return JSONResponse({"error": "BRAVE_API_KEY not set"}, status_code=500)
+    try:
+        r = requests.get(
+            "https://api.search.brave.com/res/v1/web/search",
+            headers={"X-Subscription-Token": key, "Accept": "application/json"},
+            params={"q": query, "count": 5}, timeout=10,
+        )
+        r.raise_for_status()
+        results = r.json().get("web", {}).get("results", [])
+        items = []
+        for item in results[:5]:
+            items.append({
+                "title": item.get("title", ""),
+                "desc": item.get("description", ""),
+                "url": item.get("url", ""),
+            })
+        return JSONResponse({"results": items})
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+
+# ── PDF Text Extraction ─────────────────────────────────────────────────
+@fapp.post("/api/extract-pdf")
+async def api_extract_pdf(request: Request):
+    """Base64 PDF → text extraction"""
+    try:
+        body = await request.json()
+        b64 = body.get("data", "")
+        if "," in b64:
+            b64 = b64.split(",", 1)[1]
+        import io
+        pdf_bytes = base64.b64decode(b64)
+        text = ""
+        try:
+            import fitz  # PyMuPDF
+            doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+            for page in doc:
+                text += page.get_text() + "\n"
+        except ImportError:
+            # Fallback: simple text extraction
+            content = pdf_bytes.decode("utf-8", errors="ignore")
+            # Very basic: find text between stream/endstream
+            text = re.sub(r'[^\x20-\x7E\n\r\uAC00-\uD7A3\u3040-\u309F\u30A0-\u30FF]', '', content)
+        
+        text = text.strip()[:8000]  # Max 8000 chars
+        return JSONResponse({"text": text, "chars": len(text)})
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+
 app = gr.mount_gradio_app(fapp, gradio_demo, path="/gradio")
 
 if __name__ == "__main__":