Update main.py

main.py

@@ -4,22 +4,22 @@ from typing import Optional
 from huggingface_hub import snapshot_download
 from fastapi import FastAPI, Header, HTTPException, Request, Response
 from huggingface_hub.hf_api import HfApi
-
-import admin
-
-if not admin.isUserAdmin():
-    admin.runAsAdmin()
+from huggingface_hub import whoami
+from huggingface_hub.utils import build_hf_headers, hf_raise_for_status
+from huggingface_hub import delete_repo
 
 app = FastAPI()
 
 api = HfApi() 
-directory = os.getcwd()
 
-print(directory)
+token = "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"
+
 @app.get("/")
 def read_root():
     return {"Hello": "World!"}
 
+
+
 @app.post("/webhook")
 async def webhook(request: Request):
     if request.method == "POST":
@@ -27,9 +27,26 @@ async def webhook(request: Request):
             return Response("Invalid secret", status_code=401)
         data = await request.json()
         if(data["event"]["action"]=="update" and data["event"]["scope"]=="repo.content" and data["repo"]["type"]=="model"):
-            snapshot_download(repo_id="SakethTest/ThirdParty",local_dir="/code/ThirdParty")
-            api.upload_folder(folder_path="/code/ThirdParty",repo_id="shellplc/ThirdParty",repo_type="model",commit_message="uploaded third party model",token="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")
-            return {"processed": True}
-        else:
-            return {"processed": False}
+            try:
+                _ = whoami(token)
+                # ^ this will throw if token is invalid
+    
+                delete_repo(repo_id="shellplc/ThirdParty", token = token, repo_type="model",missing_ok=True)
+                
+                r = requests.post(
+                    f"{ENDPOINT}/api/models/SakethTest/ThirdParty/duplicate",
+                    headers=build_hf_headers(token=token),
+                    json={"repository": "shellplc/ThirdParty"},
+                )
+                hf_raise_for_status(r)
+                
+                return {"processed": True}
+                
+            except Exception as e:
+                return (
+                    f"Error {e}",
+                    None,
+                )
+            
+    return {"processed": False}