Update app.py

app.py

@@ -20,6 +20,7 @@ import asyncio
 from functools import partial
 
 EXAM_DATASET_ID = os.getenv("EXAM_DATASET_ID") or "agents-course/unit_1_quiz"
+CERTIFICATE_DATASET_REPO_ID = os.getenv("CERTIFICATE_DATASET_REPO_ID") or "smol-course/certificates"
 EXAM_MAX_QUESTIONS = os.getenv("EXAM_MAX_QUESTIONS") or 1
 EXAM_PASSING_SCORE = os.getenv("EXAM_PASSING_SCORE") or 0.8
 CERTIFYING_ORG_LINKEDIN_ID = os.getenv("CERTIFYING_ORG_LINKEDIN_ID", "000000")
@@ -27,7 +28,7 @@ COURSE_TITLE = os.getenv("COURSE_TITLE", "Fundamentals of MCP")
 
 ds = load_dataset(EXAM_DATASET_ID, split="train")
 
-DATASET_REPO_URL = "https://huggingface.co/datasets/mcp-course/certificates"
+DATASET_REPO_URL = f"https://huggingface.co/datasets/{CERTIFICATE_DATASET_REPO_ID}"
 
 # Convert dataset to a list of dicts and randomly sort
 quiz_data = ds.to_pandas().to_dict("records")
@@ -120,7 +121,7 @@ def create_linkedin_button(username: str, cert_url: str | None) -> str:
     current_month = date.today().month
 
     # Use the dataset certificate URL if available, otherwise fallback to default
-    certificate_url = cert_url or "https://huggingface.co/mcp-course"
+    certificate_url = cert_url or "https://huggingface.co/smol-course"
 
     linkedin_params = {
         "startTask": "CERTIFICATION_NAME",
@@ -162,7 +163,7 @@ async def upload_certificate_to_hub(username: str, certificate_img) -> str:
                 upload_file,
                 path_or_fileobj=tmp.name,
                 path_in_repo=f"certificates/{username}/{date.today()}.png",
-                repo_id="mcp-course/certificates",
+                repo_id=CERTIFICATE_DATASET_REPO_ID,
                 repo_type="dataset",
                 token=os.getenv("HF_TOKEN"),
             )
@@ -170,7 +171,7 @@ async def upload_certificate_to_hub(username: str, certificate_img) -> str:
 
             # Construct the URL to the image
             cert_url = (
-                f"https://huggingface.co/datasets/mcp-course/certificates/"
+                f"https://huggingface.co/datasets/{CERTIFICATE_DATASET_REPO_ID/"
                 f"resolve/main/certificates/{username}/{date.today()}.png"
             )