From 4b536127998c7176463df0f9508161396b83e0e7 Mon Sep 17 00:00:00 2001
From: Thadeu <82273300+thadeuf@users.noreply.github.com>
Date: Tue, 24 Sep 2024 08:29:12 -0300
Subject: [PATCH] Update main.py

---
 main.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/main.py b/main.py
index 4f83a1f..1bb742c 100644
--- a/main.py
+++ b/main.py
@@ -1,14 +1,18 @@
 from typing import Optional
 
-from fastapi import FastAPI
+from fastapi import FastAPI, File, UploadFile
+import pdfplumber
 
 app = FastAPI()
 
 
-@app.get("/")
-async def root():
-    return {"message": "Hello World"}
+@app.post("/extract-text")
+async def extract_text(file: UploadFile = File(...)):
+    # Abre o arquivo PDF
+    with pdfplumber.open(file.file) as pdf:
+        text = ""
+        # Extrai o texto de cada página
+        for page in pdf.pages:
+            text += page.extract_text()
 
-@app.get("/items/{item_id}")
-def read_item(item_id: int, q: Optional[str] = None):
-    return {"item_id": item_id, "q": q}
\ No newline at end of file
+    return {"extracted_text": text}