Update PDF saver utils

app/db.py

@@ -1,16 +1,33 @@
 # app/db.py
 from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorGridFSBucket
 import os
+import logging
+
+logger = logging.getLogger("book-query")
 
 MONGO_URI = os.getenv("MONGODB_URI")
 MONGO_DB_NAME = os.getenv("MONGODB_DB", "querysearcher")
+TEXTBOOK_URI = os.getenv("TEXTBOOK_URI")
 
+# == QUERY ==
 # Return a fresh client for current event loop
 def get_client():
     return AsyncIOMotorClient(MONGO_URI)
-
 def get_db():
     return get_client()[MONGO_DB_NAME]
-
 def get_gridfs():
     return AsyncIOMotorGridFSBucket(get_db())
+
+# == PDF SAVER ==
+async def save_to_textbook_fs(doc_id: str, file_path: str):
+    try:
+        textbook_client = AsyncIOMotorClient(TEXTBOOK_URI)
+        textbook_db = textbook_client.get_default_database()
+        textbook_fs = AsyncIOMotorGridFSBucket(textbook_db)
+        # Read
+        with open(file_path, "rb") as f:
+            await textbook_fs.upload_from_stream(f"{doc_id}.pdf", f)
+        # Log
+        logger.info(f"📦 PDF also stored to textbook bucket at: {TEXTBOOK_URI}")
+    except Exception as e:
+        logger.warning(f"⚠️ Failed to save to textbook storage: {e}")

app/main.py

@@ -17,6 +17,7 @@
 # ├── docker-compose.yml
 # └── README.md
 # https://binkhoale1812-querysearcher.hf.space/
+# https://binkhoale1812-querysearcher.hf.space/health
 
 # app/main.py
 from fastapi import FastAPI, WebSocket

app/routers/import_doc.py

@@ -1,7 +1,7 @@
 # app/routers/import.py
 from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
-from app.db import get_db, get_gridfs
+from app.db import get_db, get_gridfs, save_to_textbook_fs
 from app.services import google_books, open_library, internet_archive, project_gutenberg
 from app.services.ingest import parse_and_index
 import aiofiles, uuid, os
@@ -31,10 +31,9 @@ async def import_book(req: ImportRequest):
     if req.source not in source_lookup:
         logger.warning(f"❌ Invalid source: {req.source}")
         raise HTTPException(400, "Invalid source")
-
+    # Return all result
     result = await source_lookup[req.source](req.ref)
     logger.debug(f"🔎 Fetch result for ref {req.ref}: {result}")
-    
     # Debugs
     if not result:
         logger.warning(f"⛔️ No fetch result for {req.source} with ref {req.ref}")
@@ -47,7 +46,6 @@ async def import_book(req: ImportRequest):
     download_url = result["download_url"]
     file_path = f"/tmp/{req.candidate_id}.pdf"
     logger.info(f"⬇️ Downloading from: {download_url}")
-
     # Read and write file 
     try:
         async with aiofiles.open(file_path, mode='wb') as f:
@@ -58,15 +56,17 @@ async def import_book(req: ImportRequest):
         logger.info(f"✅ PDF saved to {file_path}")
     except Exception as e:
         logger.error(f"🚨 Failed to download or write PDF: {e}")
-        raise HTTPException(500, "Failed to download PDF")
-    
+        raise HTTPException(500, "Failed to download PDF") 
     # Save to bucket using loop-safe GridFS
     try:
         grid_fs_bucket = get_gridfs()
+        # Save to query bucket
         with open(file_path, "rb") as f:
             await grid_fs_bucket.upload_from_stream(f"{req.candidate_id}.pdf", f)
+        # Save to textbook bucket
+        await save_to_textbook_fs(req.candidate_id, file_path)
         os.remove(file_path)
-
+    # Storage may exceed or fail on writing
     except Exception as e:
         logger.error(f"💥 Failed to upload to GridFS: {e}")
         raise HTTPException(500, "Storage failed")