Add asynccontextmanager and lifespan to main.py

main.py

@@ -7,7 +7,7 @@ from huggingface_hub import HfApi
 import os
 from dotenv import load_dotenv
 import json
-
+from contextlib import asynccontextmanager
 from datetime import datetime
 from pathlib import Path
 
@@ -18,6 +18,23 @@ load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_api = HfApi(token=HF_TOKEN)
 
+scheduler = CommitScheduler(
+    repo_id="davanstrien/votes",
+    repo_type="dataset",
+    folder_path="votes",
+    path_in_repo="data",
+    every=1,
+    hf_api=hf_api,
+)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    Path("votes").mkdir(exist_ok=True)
+
+    yield
+
+
 app = FastAPI()
 VOTES_FILE = "votes/votes.jsonl"
 # Configure CORS
@@ -34,15 +51,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-scheduler = CommitScheduler(
-    repo_id="davanstrien/votes",
-    repo_type="dataset",
-    folder_path="votes",
-    path_in_repo="data",
-    every=1,
-    hf_api=hf_api,
-)
-
 
 class Vote(BaseModel):
     dataset: str