Refactor code and remove unnecessary CORS middleware

main.py

@@ -4,21 +4,21 @@ import os
 from contextlib import asynccontextmanager
 from datetime import datetime
 from pathlib import Path
+from typing import Annotated
 
 from dotenv import load_dotenv
 from fastapi import BackgroundTasks, FastAPI, Header
-from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
-from huggingface_hub import CommitScheduler, HfApi, whoami
-from huggingface_hub.utils._errors import HTTPError
+from huggingface_hub import CommitScheduler, HfApi, hf_hub_download
 from pydantic import BaseModel, Field
 from starlette.responses import RedirectResponse
-from typing import Annotated
-
-from fastapi import FastAPI, Header
 
 load_dotenv()
 
+app = FastAPI()
+VOTES_FILE = "votes/votes.jsonl"
+
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_api = HfApi(token=HF_TOKEN)
 
@@ -35,31 +35,38 @@ scheduler = CommitScheduler(
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    Path("votes").mkdir(exist_ok=True)
+    if not Path("votes").exists():
+        Path("votes").mkdir()
+    path = hf_hub_download(
+        "davanstrien/summary-ratings",
+        "data/votes.jsonl",
+        repo_type="dataset",
+        token=HF_TOKEN,
+        local_dir="votes",
+        local_dir_use_symlinks=False,
+    )
+    print(f"Downloaded votes to {path}")
     yield
 
 
-app = FastAPI()
-VOTES_FILE = "votes/votes.jsonl"
-# Configure CORS
-origins = [
-    "https://huggingface.co",
-    "chrome-extension://ogbhjlfpmjgjbjoiffagjogbhgaipopf",  # Replace with your Chrome plugin ID
-]
-
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=origins,
-    allow_credentials=True,
-    allow_methods=["POST"],
-    allow_headers=["*"],
-)
+# # Configure CORS
+# origins = [
+#     "https://huggingface.co",
+#     "chrome-extension://ogbhjlfpmjgjbjoiffagjogbhgaipopf",  # Replace with your Chrome plugin ID
+# ]
+
+
+# app.add_middleware(
+#     CORSMiddleware,
+#     allow_origins=origins,
+#     allow_credentials=True,
+#     allow_methods=["POST"],
+#     allow_headers=["*"],
+# )
 
 
 def save_vote(vote_entry):
     with open(VOTES_FILE, "a") as file:
-        # add time stamp to the vote entry
         date_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         vote_entry["timestamp"] = date_time
         json.dump(vote_entry, file)
@@ -98,11 +105,4 @@ async def receive_vote(
     }
     # Append the vote entry to the JSONL file
     background_tasks.add_task(save_vote, vote_entry)
-
-    response = JSONResponse(content={"message": "Vote submitted successfully"})
-    response.headers[
-        "Access-Control-Allow-Origin"
-    ] = "chrome-extension://ogbhjlfpmjgjbjoiffagjogbhgaipopf"  # TODO  Replace with your Chrome plugin ID
-    response.headers["Access-Control-Allow-Methods"] = "POST"
-    response.headers["Access-Contr`ol-Allow-Headers"] = "*"
-    return response
+    return JSONResponse(content={"message": "Vote submitted successfully"})