Update main.py

main.py

@@ -1,8 +1,10 @@
 from fastapi import FastAPI, Request, Header, HTTPException
 from pydantic import BaseModel
 from typing import Optional, Any, Dict, List
-import requests
+import httpx
 import os
+import socket
+from urllib.parse import urlparse
 
 app = FastAPI()
 
@@ -36,22 +38,43 @@ class HFPayload(BaseModel):
     webhook: Optional[Dict[str, Any]] = None
 
 @app.on_event("startup")
-def startup_event():
-    # Simple check to see if the URL loaded correctly
+async def startup_event():
+    print("--- STARTUP DIAGNOSTICS ---")
+    # 1. Check if Variable Exists
     if DISCORD_WEBHOOK_URL:
+        # Mask the URL for safety in logs
         masked = DISCORD_WEBHOOK_URL[:8] + "..." + DISCORD_WEBHOOK_URL[-4:]
         print(f"INFO: Webhook URL loaded: {masked}")
+        
+        # 2. Extract Hostname
+        try:
+            parsed = urlparse(DISCORD_WEBHOOK_URL)
+            hostname = parsed.hostname
+            if not hostname:
+                print("ERROR: Could not extract hostname from URL. Is the format 'https://...'?")
+            else:
+                print(f"DEBUG: Hostname extracted: '{hostname}'")
+                
+                # 3. Test DNS Resolution (The "Network Doctor")
+                try:
+                    # Note: socket.gethostbyname is blocking, but okay for a quick startup check
+                    ip = socket.gethostbyname(hostname)
+                    print(f"SUCCESS: DNS resolved {hostname} to {ip}")
+                except socket.gaierror as e:
+                    print(f"CRITICAL FAILURE: Could not resolve {hostname}. Error: {e}")
+                    print("-> This means the URL is wrong OR the container has no internet access.")
+        except Exception as e:
+            print(f"ERROR parsing URL: {e}")
     else:
         print("WARNING: No DISCORD_WEBHOOK_URL found. Discord messages will not be sent.")
+    print("---------------------------")
 
 @app.get("/")
 def home():
     return {"status": "Webhook listener is running"}
 
-# CHANGED: 'async def' -> 'def'
-# This tells FastAPI to run this in a threadpool so 'requests' doesn't block the server
 @app.post("/webhook")
-def handle_webhook(
+async def handle_webhook(
     payload: HFPayload, 
     x_webhook_secret: Optional[str] = Header(None)
 ):
@@ -67,17 +90,19 @@ def handle_webhook(
     if DISCORD_WEBHOOK_URL:
         print(f"Sending notification for {payload.repo.name}...")
         try:
-            # Using standard requests library
-            response = requests.post(
-                DISCORD_WEBHOOK_URL, 
-                json={"embeds": [embed]},
-                timeout=10 # Good practice to prevent hanging forever
-            )
-            response.raise_for_status()
-            print(f"Sent to Discord: {response.status_code}")
+            # Added a timeout to prevent hanging
+            async with httpx.AsyncClient(timeout=10.0) as client:
+                resp = await client.post(
+                    DISCORD_WEBHOOK_URL, 
+                    json={"embeds": [embed]}
+                )
+                resp.raise_for_status()
+                print(f"Sent to Discord: {resp.status_code}")
+        except httpx.ConnectError as e:
+            print(f"CONNECTION ERROR: Could not connect to Discord. Check DNS/Network. Details: {e}")
+            return {"message": "Connection Failed", "error": str(e)}
         except Exception as e:
             print(f"ERROR connecting to Discord: {e}")
-            # We return 200 to HF so it stops retrying, but log the error
             return {"message": "Error sending to Discord", "error": str(e)}
     
     return {"message": "Processed"}