webhook models

main.py

@@ -1,13 +1,13 @@
 from fastapi import FastAPI, Form, Request, BackgroundTasks
 from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
-#from pydantic import BaseModel
 
 from uuid import uuid4
 import time
 import asyncio
 
 from build_map import load_dataset_and_metadata, upload_dataset_to_atlas
+from models import WebhookPayload, WebhookPayloadRepo, WebhookPayloadEvent
 
 
 app = FastAPI()
@@ -56,3 +56,18 @@ async def read_task(task_id: str):
         return {'status': 'not found'}
     else:
         return tasks[task_id]
+
+@app.post("/webhook")
+async def post_webhook(background_tasks: BackgroundTasks, payload: WebhookPayload):
+    if not (
+        payload.event.action == "update"
+        and payload.event.scope.startswith("repo.content")
+        and payload.repo.type == "dataset"
+    ):
+        return {"processed": False}
+    else:
+        task_id = str(uuid4())
+        tasks[task_id] = {'status': 'running'}
+        #form_data = DatasetForm(dataset_name=dataset_name)
+        background_tasks.add_task(upload_atlas_task, task_id, payload.repo.name)
+        return {'task_id': task_id}

models.py

@@ -0,0 +1,17 @@
+from pydantic import BaseModel
+from typing import Literal
+
+class WebhookPayloadEvent(BaseModel):
+	action: Literal["create", "update", "delete"]
+	scope: str
+
+class WebhookPayloadRepo(BaseModel):
+	type: Literal["dataset", "model", "space"]
+	name: str
+	id: str
+	private: bool
+	headSha: str
+        
+class WebhookPayload(BaseModel):
+	event: WebhookPayloadEvent
+	repo: WebhookPayloadRepo