1MR commited on
Commit
6025050
·
verified ·
1 Parent(s): aae0bb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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(...)):