Update app.py

app.py

@@ -17,6 +17,21 @@ def read_root():
     # toolkit = GmailToolkit()
     return {"message": "Connection"}
 
+
+
+UPLOAD_FOLDER = "uploaded_pdfs"
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
+@app.post("/upload-pdf")
+async def upload_pdf(file: UploadFile = File(...)):
+    try:
+        file_location = os.path.join(UPLOAD_FOLDER, file.filename)
+        with open(file_location, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+        return JSONResponse(content={"message": f"Uploaded {file.filename} successfully"}, status_code=200)
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+        
 # # API endpoint for prediction
 # @app.post("/predict")
 # async def predict_image(file: UploadFile = File(...)):