modified basic and subway

endpoints/basic.py

@@ -21,6 +21,12 @@ async def create_basic_video(
 ):
 	# Save the videos
 	basic_video_path = "./data/basic/videos/video.mp4"
+
+	# create the directory if it doesn't exist
+	if not os.path.exists("./data/basic/videos"):
+		os.makedirs("./data/basic/videos")
+
+	# Save the videos
 	with open(basic_video_path, "wb") as buffer:
 		shutil.copyfileobj(video.file, buffer)
 

endpoints/subway.py

@@ -22,6 +22,11 @@ async def create_subway_video(
 ):
 	top_video_path = "./data/subway/videos/top_video.mp4"
 	bottom_video_path = "./data/subway/videos/bottom_video.mp4"
+
+	# Create the directory if it doesn't exist
+	if not os.path.exists("./data/subway/videos"):
+		os.makedirs("./data/subway/videos")
+
 	# Save the videos
 	with open(top_video_path, "wb") as buffer:
 		shutil.copyfileobj(top_video.file, buffer)

server.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 import uvicorn
 from endpoints import subway_router, minecraft_router, basic_router, video_editor
 
@@ -17,6 +18,12 @@ app.add_middleware(
 # mount static folder
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
+# Define the root route
+@app.get("/")
+async def root():
+	# return the index.html file
+	return FileResponse("static/index.html")
+
 # Include routers
 app.include_router(subway_router, prefix="/generate-subtitles")
 app.include_router(minecraft_router, prefix="/generate-subtitles")