improvement to main.py

main.py

@@ -3,16 +3,23 @@ from pydantic import BaseModel
 from typing import Optional, Any, Dict, List
 import httpx
 import os
+import socket
+from urllib.parse import urlparse
 
 app = FastAPI()
 
-# --- Configuration ---
-# Set these in your Space's "Settings" -> "Variables and secrets"
-DISCORD_WEBHOOK_URL = os.environ.get("DISCORD_WEBHOOK_URL")
-# Optional: Secure your webhook so only HF can trigger it
-WEBHOOK_SECRET = os.environ.get("WEBHOOK_SECRET")
+# --- Configuration & Cleaning ---
+def get_clean_env(key):
+    val = os.environ.get(key)
+    if val:
+        # Remove whitespace, single quotes, and double quotes
+        return val.strip().strip("'").strip('"')
+    return None
 
-# --- Pydantic Models for HF Payload ---
+DISCORD_WEBHOOK_URL = get_clean_env("DISCORD_WEBHOOK_URL")
+WEBHOOK_SECRET = get_clean_env("WEBHOOK_SECRET")
+
+# --- Pydantic Models ---
 class HFRepo(BaseModel):
     type: str
     name: str
@@ -30,6 +37,38 @@ class HFPayload(BaseModel):
     comment: Optional[Dict[str, Any]] = None
     webhook: Optional[Dict[str, Any]] = None
 
+@app.on_event("startup")
+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"}
@@ -41,6 +80,7 @@ async def handle_webhook(
 ):
     # 1. Security Check
     if WEBHOOK_SECRET and x_webhook_secret != WEBHOOK_SECRET:
+        print(f"Unauthorized access attempt. Header: {x_webhook_secret}")
         raise HTTPException(status_code=401, detail="Invalid Secret")
 
     # 2. Format for Discord
@@ -48,12 +88,22 @@ async def handle_webhook(
     
     # 3. Send to Discord
     if DISCORD_WEBHOOK_URL:
-        async with httpx.AsyncClient() as client:
-            resp = await client.post(
-                DISCORD_WEBHOOK_URL, 
-                json={"embeds": [embed]}
-            )
-            print(f"Sent to Discord: {resp.status_code}")
+        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)}
+        except Exception as e:
+            print(f"ERROR connecting to Discord: {e}")
+            return {"message": "Error sending to Discord", "error": str(e)}
     
     return {"message": "Processed"}
 
@@ -82,7 +132,10 @@ def build_discord_embed(data: HFPayload):
         title = f"Discussion Update: {disc_title}"
         
         if data.comment:
-            description = f"**New Comment** on {repo_name}:\n> {data.comment.get('content', '')[:200]}..."
+            content = data.comment.get('content', '')
+            if len(content) > 200:
+                content = content[:200] + "..."
+            description = f"**New Comment** on {repo_name}:\n> {content}"
         else:
             description = f"Discussion status: **{event_action}**"