Update app/main.py

app/main.py

@@ -1,9 +1,12 @@
 from pydantic import BaseModel
 from fastapi import FastAPI, UploadFile, File
-from app.pdf_handler import process_pdf
+from app.pdf_handler import process_pdf, process_image_pdf
 from app.rag_pipeline import generate_answer
 from app.test import query_test
-
+from fastapi import FastAPI, File, UploadFile
+from PIL import Image
+import pytesseract
+import io
 
 app = FastAPI()
 
@@ -23,6 +26,19 @@ async def upload_pdf(file: UploadFile = File(...)):
     return process_pdf(file)
 
 
+@app.post("/upload/imgpdf/")
+async def upload_pdf_image(file: UploadFile = File(...)):
+    return process_image_pdf(file)
+
+
 @app.post("/test/")
 async def chat(query: str, top_k: int = 3):
     return query_test(query)
+
+
+@app.post("/ocr/")
+async def extract_text(file: UploadFile = File(...)):
+    image_data = await file.read()
+    image = Image.open(io.BytesIO(image_data))
+    text = pytesseract.image_to_string(image)
+    return {"extracted_text": text}