Update main.py

main.py

@@ -1,10 +1,8 @@
 from fastapi import FastAPI, Request, Header, HTTPException
 from pydantic import BaseModel
 from typing import Optional, Any, Dict, List
-import httpx
+import requests
 import os
-import socket
-from urllib.parse import urlparse
 
 app = FastAPI()
 
@@ -38,43 +36,22 @@ class HFPayload(BaseModel):
     webhook: Optional[Dict[str, Any]] = None
 
 @app.on_event("startup")
-async def startup_event():
-    print("--- STARTUP DIAGNOSTICS ---")
-    # 1. Check if Variable Exists
+def startup_event():
+    # Simple check to see if the URL loaded correctly
     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")
-async def handle_webhook(
+def handle_webhook(
     payload: HFPayload, 
     x_webhook_secret: Optional[str] = Header(None)
 ):
@@ -90,19 +67,17 @@ async def handle_webhook(
     if DISCORD_WEBHOOK_URL:
         print(f"Sending notification for {payload.repo.name}...")
         try:
-            # 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)}
+            # 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}")
         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"}