Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - HF Spaces compatible version
+# app.py - HF Spaces compatible version (Fixed)
 import os
 import re
 import json
@@ -6,9 +6,6 @@ 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
 
 # Configuration - Use HF Spaces secrets
 WEBHOOK_SECRET = os.getenv("WEBHOOK_SECRET")
@@ -32,12 +29,6 @@ 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()
@@ -240,50 +231,6 @@ async def process_webhook_comment(webhook_data: Dict[str, Any]):
         print(f"❌ {error_msg}")
         return error_msg
 
-# Create FastAPI app for webhook handling
-fastapi_app = FastAPI(title="HF Tagging Bot API")
-fastapi_app.add_middleware(CORSMiddleware, allow_origins=["*"])
-
-@fastapi_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)}
-
-@fastapi_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)
-    }
-
 def create_gradio_interface():
     """Create Gradio interface for monitoring"""
     with gr.Blocks(title="HF Tagging Bot", theme=gr.themes.Soft()) as interface:
@@ -383,12 +330,54 @@ Visit: `https://your-space-name.hf.space/health`
 # Create the Gradio interface
 demo = create_gradio_interface()
 
-# Mount FastAPI to Gradio (for HF Spaces)
-app = gr.mount_gradio_app(fastapi_app, demo, path="/")
+# 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)
+    }
 
-# This is the main app that HF Spaces will use
+# Launch the interface
 if __name__ == "__main__":
     print("🚀 HF Tagging Bot - Gradio interface ready")
     print(f"🔑 HF_TOKEN: {'✅ Configured' if HF_TOKEN else '❌ Missing'}")
     print(f"🔐 Webhook Secret: {'✅ Configured' if WEBHOOK_SECRET else '❌ Missing'}")
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)