Spaces:
Sleeping
Sleeping
File size: 1,353 Bytes
4810f6f 64ce359 2d18665 4810f6f 5235f50 4810f6f 2d18665 4810f6f 5235f50 64ce359 4810f6f 5235f50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
from pydantic import BaseModel
from fastapi import FastAPI, UploadFile, File
from app.pdf_handler import process_pdf, process_image_pdf, process_image_pdf_dummy
from app.rag_pipeline import generate_answer, generate_answer_am
from app.test import query_test
from fastapi import FastAPI, File, UploadFile
from PIL import Image
import pytesseract
import io
app = FastAPI()
class ChatRequest(BaseModel):
query: str
top_k: int = 3
@app.post("/chat/")
async def chat(request: ChatRequest):
return generate_answer(request.query, request.top_k)
@app.post("/am/chat/")
async def chat(request: ChatRequest):
return generate_answer_am(request.query, request.top_k)
@app.post("/upload/")
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("/upload/imgpdf/dummy/")
async def upload_pdf_image_dummy(f):
return process_image_pdf_dummy()
@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}
|