Initial commit

Dockerfile

@@ -1,43 +1,43 @@
-# --------- Frontend Build Stage ---------
-FROM node:18-alpine AS frontend-build
-
-WORKDIR /app/frontend
-
-COPY frontend/package*.json ./
-RUN npm ci --only=production
-COPY frontend/ ./
-RUN npm run build && npm run export
-
-# --------- Backend Build Stage ---------
-FROM python:3.11-slim AS backend
-
-WORKDIR /app/backend
-
-# System dependencies
-RUN apt-get update && apt-get install -y gcc curl && rm -rf /var/lib/apt/lists/*
-
-# Install Python dependencies
-COPY backend/requirements.txt ./
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy backend code
-COPY backend/ ./
-
-# Copy frontend static build
-COPY --from=frontend-build /app/frontend/out /app/frontend_out
-
-# Ensure necessary directories and DB file exist and have permissions
-RUN mkdir -p /app/backend/chroma_db /app/backend/uploads \
-    && touch /app/backend/pdf_chatbot.db \
-    && chmod -R 777 /app/backend/chroma_db /app/backend/uploads /app/backend/pdf_chatbot.db
-
-# Expose FastAPI port
-EXPOSE 8000
-
-# Add healthcheck
-HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:8000/health || exit 1
-
-# Start FastAPI server
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+# ----------  Front‑end build stage ----------
+    FROM node:18-alpine AS frontend-build
+
+    WORKDIR /app/frontend
+    COPY frontend/package*.json ./
+    RUN npm ci --only=production
+    COPY frontend/ ./
+    RUN npm run build && npm run export
+    
+    
+    # ----------  Back‑end stage ----------
+    FROM python:3.11-slim AS backend
+    
+    WORKDIR /app/backend                       # <─ matches compose volumes
+    
+    # System build deps
+    RUN apt-get update && apt-get install -y gcc curl && rm -rf /var/lib/apt/lists/*
+    
+    # Python deps
+    COPY backend/requirements.txt ./
+    RUN pip install --no-cache-dir -r requirements.txt
+    
+    # App source code
+    COPY backend/ ./
+    
+    # Static export from previous stage
+    COPY --from=frontend-build /app/frontend/out /app/frontend_out
+    
+    # Create folders & DB file and make them writable
+    RUN mkdir -p /app/backend/uploads /app/backend/chroma_db \
+        && touch /app/backend/pdf_chatbot.db \
+        && chmod -R 777 /app/backend/uploads /app/backend/chroma_db /app/backend/pdf_chatbot.db
+    
+    # Expose API port
+    EXPOSE 8000
+    
+    # Small health‑check
+    HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+      CMD curl -f http://localhost:8000/health || exit 1
+    
+    # Start FastAPI
+    CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
     

backend/app/api/endpoints/documents.py

@@ -14,6 +14,8 @@ from app.services.pdf_processor import PDFProcessor
 from app.services.vector_store import VectorStore
 from app.models.document import ChatMessage
 import shutil
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
 
 router = APIRouter()
 pdf_processor = PDFProcessor()
@@ -32,11 +34,11 @@ async def upload_document(
         if doc_count >= 3:
             raise HTTPException(status_code=400, detail="You can only upload up to 3 documents.")
         # Validate file type
-        if not file.filename.lower().endswith('.pdf'):
+        if not file.filename or not file.filename.lower().endswith('.pdf'):
             raise HTTPException(status_code=400, detail="Only PDF files are allowed")
         
         # Generate unique filename
-        file_extension = os.path.splitext(file.filename)[1]
+        file_extension = os.path.splitext(str(file.filename))[1]
         unique_filename = f"{uuid.uuid4()}{file_extension}"
         file_path = os.path.join(settings.UPLOAD_DIR, unique_filename)
         
@@ -150,7 +152,7 @@ def delete_document(document_id: int, db: Session = Depends(get_db)):
         vector_store.delete_document(str(document_id))
         
         # Delete file from filesystem
-        if os.path.exists(document.file_path):
+        if isinstance(document.file_path, str) and os.path.exists(document.file_path):
             os.remove(document.file_path)
         
         # Delete from database
@@ -168,7 +170,7 @@ def delete_document(document_id: int, db: Session = Depends(get_db)):
 
 
 @router.post("/clear_all")
-def clear_all_data(db: Session = Depends(get_db)):
+async def clear_all_data(db: Session = Depends(get_db)):
     """Admin endpoint to clear all documents, chat messages, uploaded files, and vector store."""
     try:
         # Delete all documents and chat messages from DB
@@ -177,17 +179,20 @@ def clear_all_data(db: Session = Depends(get_db)):
         db.commit()
         # Delete all files in uploads directory
         upload_dir = settings.UPLOAD_DIR
-        for filename in os.listdir(upload_dir):
-            file_path = os.path.join(upload_dir, filename)
-            try:
-                if os.path.isfile(file_path) or os.path.islink(file_path):
-                    os.unlink(file_path)
-                elif os.path.isdir(file_path):
-                    shutil.rmtree(file_path)
-            except Exception as e:
-                print(f"Failed to delete {file_path}: {e}")
+        loop = asyncio.get_event_loop()
+        def remove_uploads():
+            for filename in os.listdir(upload_dir):
+                file_path = os.path.join(upload_dir, filename)
+                try:
+                    if os.path.isfile(file_path) or os.path.islink(file_path):
+                        os.unlink(file_path)
+                    elif os.path.isdir(file_path):
+                        shutil.rmtree(file_path)
+                except Exception as e:
+                    print(f"Failed to delete {file_path}: {e}")
+        await loop.run_in_executor(None, remove_uploads)
         # Clear ChromaDB vector store using the singleton
-        vector_store.clear_all()
+        await loop.run_in_executor(None, vector_store.clear_all)
         return {"success": True, "message": "All documents, chat messages, uploads, and vectors cleared."}
     except Exception as e:
         return {"success": False, "message": f"Error clearing data: {str(e)}"}
@@ -199,9 +204,8 @@ def get_document_stats(db: Session = Depends(get_db)):
     try:
         total_documents = db.query(Document).count()
         processed_documents = db.query(Document).filter(Document.processed == True).count()
-        total_size = db.query(Document).with_entities(
-            db.func.sum(Document.file_size)
-        ).scalar() or 0
+        from sqlalchemy import func as sa_func
+        total_size = db.query(sa_func.sum(Document.file_size)).scalar() or 0
         
         vector_stats = vector_store.get_collection_stats()
         

docker-compose.yml

@@ -1,25 +1,34 @@
-version: '3.8'
+version: "3.8"
 
 services:
   backend:
     build:
-      context: ./backend
+      context: ./backend          # <─ your backend folder on host
       dockerfile: Dockerfile
     ports:
       - "8000:8000"
+
+    # ── Environment the code expects ─────────────────────────────
     environment:
-      - DATABASE_URL=sqlite:////app/backend/pdf_chatbot.db
-      - CHROMA_PERSIST_DIRECTORY=/app/backend/chroma_db
-      - UPLOAD_DIR=/app/backend/uploads
+      # ◼ SQLite file lives in the same folder as your code
+      - DATABASE_URL=sqlite:///./pdf_chatbot.db
+      # ◼ Chroma + uploads are relative to the WORKDIR (/app/backend)
+      - CHROMA_PERSIST_DIRECTORY=./chroma_db
+      - UPLOAD_DIR=./uploads
+      # other settings
       - MAX_FILE_SIZE=10485760
       - ALLOWED_EXTENSIONS=[".pdf"]
-      - BACKEND_CORS_ORIGINS=["http://localhost:3000","http://localhost:3001","http://127.0.0.1:3000","http://127.0.0.1:3001"]
+      - BACKEND_CORS_ORIGINS=["http://localhost:3000","http://localhost:3001",
+                              "http://127.0.0.1:3000","http://127.0.0.1:3001"]
     env_file:
-      - ./backend/.env
+      - ./backend/.env        # optional extra secrets
+
+    # ── Persist data on host in the same structure you already have ──
     volumes:
       - ./backend/uploads:/app/backend/uploads
       - ./backend/chroma_db:/app/backend/chroma_db
       - ./backend/pdf_chatbot.db:/app/backend/pdf_chatbot.db
+
     restart: unless-stopped
 
   frontend:

frontend/components/DocumentList.tsx

@@ -14,6 +14,18 @@ export default function DocumentList({ onDocumentChange }: { onDocumentChange?:
     loadStats()
   }, [])
 
+  // Re-fetch stats and documents when onDocumentChange is triggered
+  useEffect(() => {
+    if (!onDocumentChange) return;
+    const handler = () => {
+      loadDocuments();
+      loadStats();
+    };
+    // Listen for custom event
+    window.addEventListener('documentChange', handler);
+    return () => window.removeEventListener('documentChange', handler);
+  }, [onDocumentChange]);
+
   const loadDocuments = async () => {
     try {
       setLoading(true)

frontend/components/DocumentUpload.tsx

@@ -55,6 +55,7 @@ export default function DocumentUpload({ disabled, onDocumentChange }: DocumentU
           )
         )
         if (onDocumentChange) onDocumentChange();
+        window.dispatchEvent(new Event('documentChange'));
       } catch (error: any) {
         setUploadStatuses(prev => 
           prev.map(upload => 
@@ -94,6 +95,7 @@ export default function DocumentUpload({ disabled, onDocumentChange }: DocumentU
     }
     setUploadStatuses(prev => prev.filter(upload => upload.file !== file))
     if (onDocumentChange) onDocumentChange();
+    window.dispatchEvent(new Event('documentChange'));
   }
 
   const formatFileSize = (bytes: number) => {