Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,12 @@ async def upload_pdf(file: UploadFile = File(...)):
|
|
| 31 |
return JSONResponse(content={"message": f"Uploaded {file.filename} successfully"}, status_code=200)
|
| 32 |
except Exception as e:
|
| 33 |
return JSONResponse(content={"error": str(e)}, status_code=500)
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
# # API endpoint for prediction
|
| 36 |
# @app.post("/predict")
|
| 37 |
# async def predict_image(file: UploadFile = File(...)):
|
|
|
|
| 31 |
return JSONResponse(content={"message": f"Uploaded {file.filename} successfully"}, status_code=200)
|
| 32 |
except Exception as e:
|
| 33 |
return JSONResponse(content={"error": str(e)}, status_code=500)
|
| 34 |
+
|
| 35 |
+
|
| 36 |
+
@app.get("/list-files")
|
| 37 |
+
def list_uploaded_files():
|
| 38 |
+
files = os.listdir(UPLOAD_FOLDER)
|
| 39 |
+
return {"files": files}
|
| 40 |
# # API endpoint for prediction
|
| 41 |
# @app.post("/predict")
|
| 42 |
# async def predict_image(file: UploadFile = File(...)):
|