Adding Lucain's ideas

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import requests
 from fastapi import BackgroundTasks, Response, status
 from huggingface_hub import WebhookPayload, WebhooksServer
-from huggingface_hub.utils import build_hf_headers
+from huggingface_hub.utils import build_hf_headers, get_session
 
 from src.build_nomic import build_nomic
 from src.my_logger import setup_logger
@@ -71,7 +71,7 @@ app = WebhooksServer(ui=ui.queue(), webhook_secret=WEBHOOK_SECRET)
 
 
 @app.add_webhook("/dataset_repo")
-async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
+async def handle_repository_changes(payload: WebhookPayload, task_queue: BackgroundTasks):
     if not payload.event.scope.startswith("repo"):
         return Response("No task scheduled", status_code=status.HTTP_200_OK)
     # Only run if change is on main branch
@@ -85,10 +85,10 @@ async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
         logger.info(response_content)
         return Response(response_content, status_code=status.HTTP_200_OK)
 
-    # No need to run for README updates
+    # No need to run for README only updates
     try:
         commit_files_url = f"""{payload.repo.url.api}/compare/{payload.updatedRefs[0].oldSha}..{payload.updatedRefs[0].newSha}?raw=true"""
-        response_text = requests.get(commit_files_url, headers=build_hf_headers()).text
+        response_text = get_session().get(commit_files_url, headers=build_hf_headers()).text
         logger.info(f"Git Compare URl: {commit_files_url}")
 
         # Splitting the output into lines
@@ -104,8 +104,8 @@ async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
             logger.info(response_content)
             return Response(response_content, status_code=status.HTTP_200_OK)
     except Exception as e:
-        logger.info(f"{str(e)}")
-        response_content = "Something weird with the webhook?"
+        logger.error(f"{str(e)}")
+        response_content = "Unexpected issue :'("
         logger.info(response_content)
         return Response(response_content, status_code=status.HTTP_501_NOT_IMPLEMENTED)