from fastapi import FastAPI, File, UploadFile
from fastapi.responses import HTMLResponse, FileResponse
import shutil
import os
app = FastAPI()
# HTML form for file upload
html_content = """
File Upload
Upload a File
Uploaded Files
"""
@app.get("/", response_class=HTMLResponse)
async def read_root():
file_list = get_uploaded_files()
return html_content.format(file_list=file_list)
@app.post("/uploadfile/")
async def upload_file(file: UploadFile = File(...)):
# Save the uploaded file to the uploads directory
upload_directory = "uploads"
os.makedirs(upload_directory, exist_ok=True) # Create the directory if it doesn't exist
file_location = f"{upload_directory}/{file.filename}"
with open(file_location, "wb") as buffer:
shutil.copyfileobj(file.file, buffer)
return {"filename": file.filename}
def get_uploaded_files():
upload_directory = "uploads"
files = os.listdir(upload_directory)
file_links = ""
for file in files:
file_links += f'{file}'
return file_links
@app.get("/files/{filename}")
async def get_file(filename: str):
return FileResponse(f"uploads/{filename}")