Add logging and improve error handling

main.py

@@ -1,5 +1,5 @@
-import contextlib
 import json
+import logging
 import os
 from contextlib import asynccontextmanager
 from datetime import datetime
@@ -15,7 +15,7 @@ from pydantic import BaseModel, Field
 from starlette.responses import RedirectResponse
 
 load_dotenv()
-
+logger = logging.getLogger(__name__)
 app = FastAPI()
 VOTES_FILE = "votes/votes.jsonl"
 
@@ -39,14 +39,14 @@ async def lifespan(app: FastAPI):
     if not Path("votes").exists():
         Path("votes").mkdir()
     path = hf_hub_download(
-        "davanstrien/summary-ratings",
-        "data/votes.jsonl",
+        repo_id="davanstrien/summary-ratings",
+        filename="data/votes.jsonl",
         repo_type="dataset",
         token=HF_TOKEN,
         local_dir="votes",
         local_dir_use_symlinks=False,
     )
-    print(f"Downloaded votes to {path}")
+    logger.info(f"Downloaded votes.jsonl to {path}")
     yield
 
 
@@ -72,6 +72,7 @@ def save_vote(vote_entry):
         vote_entry["timestamp"] = date_time
         json.dump(vote_entry, file)
         file.write("\n")
+    logger.info(f"Vote saved: {vote_entry}")
 
 
 @app.get("/", include_in_schema=False)
@@ -101,6 +102,7 @@ async def receive_vote(
     background_tasks: BackgroundTasks,
 ):
     if not validate_token(Authorization):
+        logger.error("Invalid token")
         raise HTTPException(status_code=401, detail="Invalid token")
     vote_entry = {
         "dataset": vote.dataset,