From c45f5b993ddbd4f81565f7573622dd0ee4df2658 Mon Sep 17 00:00:00 2001
From: Saurab-Shrestha <shresthasaurab030@gmail.com>
Date: Mon, 19 Feb 2024 17:28:29 +0545
Subject: [PATCH] Added api for document listing

---
 .../ocr_components/table_ocr_api.py           |  2 +-
 private_gpt/server/ingest/ingest_router.py    | 29 ++++++++++++++++++-
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/private_gpt/components/ocr_components/table_ocr_api.py b/private_gpt/components/ocr_components/table_ocr_api.py
index da53fd85..088d7b9c 100644
--- a/private_gpt/components/ocr_components/table_ocr_api.py
+++ b/private_gpt/components/ocr_components/table_ocr_api.py
@@ -15,7 +15,7 @@ from private_gpt.server.ingest.ingest_router import common_ingest_logic, IngestR
 from private_gpt.constants import OCR_UPLOAD
 
 
-pdf_router = APIRouter(prefix="/pdf", tags=["ocr"])
+pdf_router = APIRouter(prefix="/v1", tags=["ocr"])
 
 
 @pdf_router.post("/pdf_ocr")
diff --git a/private_gpt/server/ingest/ingest_router.py b/private_gpt/server/ingest/ingest_router.py
index 731ffce8..e5ae468a 100644
--- a/private_gpt/server/ingest/ingest_router.py
+++ b/private_gpt/server/ingest/ingest_router.py
@@ -1,7 +1,7 @@
 import os
 import logging
 from pathlib import Path
-from typing import Literal, Optional
+from typing import Literal, Optional, List
 
 from sqlalchemy.orm import Session
 from fastapi import APIRouter, Depends, HTTPException, Request, UploadFile, File, status, Security, Body
@@ -275,3 +275,30 @@ async def common_ingest_logic(
             status_code=500,
             detail="Internal Server Error: Unable to ingest file.",
         )
+
+from private_gpt.users.schemas import Document
+
+@ingest_router.get("/ingest/list_files", response_model=List[schemas.Document], tags=["Ingestion"])
+def list_files(
+    request: Request,
+    db: Session = Depends(deps.get_db),
+    skip: int = 0,
+    limit: int = 100,
+    current_user: models.User = Security(
+            deps.get_current_user,
+            scopes=[Role.ADMIN["name"], Role.SUPER_ADMIN["name"]],
+
+    )
+):
+    try:
+        docs = crud.documents.get_multi(db, skip=skip, limit=limit)
+        return docs
+    except Exception as e:
+        logger.error(f"There was an error uploading the file(s): {str(e)}")
+        print("ERROR: ", e)
+        raise HTTPException(
+            status_code=500,
+            detail="Internal Server Error: Unable to ingest file.",
+        )
+
+    
\ No newline at end of file