Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - HF Spaces compatible version (Fixed)
+# app.py - HF Spaces compatible version (Fixed webhook handling)
 import os
 import re
 import json
@@ -6,6 +6,10 @@ from datetime import datetime
 from typing import List, Dict, Any
 
 import gradio as gr
+from fastapi import FastAPI, Request, BackgroundTasks
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+import asyncio
 
 # Configuration - Use HF Spaces secrets
 WEBHOOK_SECRET = os.getenv("WEBHOOK_SECRET")
@@ -29,6 +33,12 @@ RECOGNIZED_TAGS = {
     "llamacpp", "onnx", "mlx"
 }
 
+class WebhookEvent(BaseModel):
+    event: Dict[str, str]
+    comment: Dict[str, Any]
+    discussion: Dict[str, Any]
+    repo: Dict[str, str]
+
 def extract_tags_from_text(text: str) -> List[str]:
     """Extract potential tags from discussion text"""
     text_lower = text.lower()
@@ -231,6 +241,54 @@ async def process_webhook_comment(webhook_data: Dict[str, Any]):
         print(f"❌ {error_msg}")
         return error_msg
 
+# Create FastAPI app for webhook handling
+app = FastAPI(title="HF Tagging Bot API")
+app.add_middleware(CORSMiddleware, allow_origins=["*"])
+
+@app.post("/webhook")
+async def webhook_handler(request: Request, background_tasks: BackgroundTasks):
+    """Handle HF Hub webhooks"""
+    # Verify webhook secret if configured
+    if WEBHOOK_SECRET:
+        webhook_secret = request.headers.get("X-Webhook-Secret")
+        if webhook_secret != WEBHOOK_SECRET:
+            print("❌ Invalid webhook secret")
+            return {"error": "Invalid webhook secret"}
+
+    try:
+        payload = await request.json()
+        print(f"📥 Received webhook: {payload.get('event', {})}")
+
+        event = payload.get("event", {})
+        scope = event.get("scope")
+        action = event.get("action")
+
+        # Only process discussion comment creation (not PRs)
+        if (scope == "discussion" and 
+            action == "create" and 
+            not payload.get("discussion", {}).get("isPullRequest", False)):
+            
+            background_tasks.add_task(process_webhook_comment, payload)
+            return {"status": "processing"}
+
+        return {"status": "ignored"}
+    except Exception as e:
+        print(f"❌ Webhook error: {e}")
+        return {"error": str(e)}
+
+@app.get("/health")
+async def health_check():
+    return {
+        "status": "healthy",
+        "hf_token_configured": bool(HF_TOKEN),
+        "webhook_secret_configured": bool(WEBHOOK_SECRET),
+        "operations_processed": len(tag_operations_store)
+    }
+
+@app.get("/")
+async def root():
+    return {"message": "HF Tagging Bot is running! Visit /gradio for the interface."}
+
 def create_gradio_interface():
     """Create Gradio interface for monitoring"""
     with gr.Blocks(title="HF Tagging Bot", theme=gr.themes.Soft()) as interface:
@@ -330,54 +388,13 @@ Visit: `https://your-space-name.hf.space/health`
 # Create the Gradio interface
 demo = create_gradio_interface()
 
-# Add webhook handling using Gradio's built-in FastAPI integration
-from fastapi import Request
-import asyncio
-
-@demo.app.post("/webhook")
-async def webhook_handler(request: Request):
-    """Handle HF Hub webhooks"""
-    # Verify webhook secret if configured
-    if WEBHOOK_SECRET:
-        webhook_secret = request.headers.get("X-Webhook-Secret")
-        if webhook_secret != WEBHOOK_SECRET:
-            print("❌ Invalid webhook secret")
-            return {"error": "Invalid webhook secret"}
-
-    try:
-        payload = await request.json()
-        print(f"📥 Received webhook: {payload.get('event', {})}")
-
-        event = payload.get("event", {})
-        scope = event.get("scope")
-        action = event.get("action")
-
-        # Only process discussion comment creation (not PRs)
-        if (scope == "discussion" and 
-            action == "create" and 
-            not payload.get("discussion", {}).get("isPullRequest", False)):
-            
-            # Process webhook in background
-            asyncio.create_task(process_webhook_comment(payload))
-            return {"status": "processing"}
-
-        return {"status": "ignored"}
-    except Exception as e:
-        print(f"❌ Webhook error: {e}")
-        return {"error": str(e)}
-
-@demo.app.get("/health")
-async def health_check():
-    return {
-        "status": "healthy",
-        "hf_token_configured": bool(HF_TOKEN),
-        "webhook_secret_configured": bool(WEBHOOK_SECRET),
-        "operations_processed": len(tag_operations_store)
-    }
+# Mount Gradio app to FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")
 
-# Launch the interface
+# This is what HF Spaces will use
 if __name__ == "__main__":
-    print("🚀 HF Tagging Bot - Gradio interface ready")
+    print("🚀 HF Tagging Bot - Starting with FastAPI + Gradio")
     print(f"🔑 HF_TOKEN: {'✅ Configured' if HF_TOKEN else '❌ Missing'}")
     print(f"🔐 Webhook Secret: {'✅ Configured' if WEBHOOK_SECRET else '❌ Missing'}")
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)