Use Gradio's internal FastAPI app for webhook endpoints

Restructure to use demo.fastapi() instead of separate FastAPI app. This ensures proper integration with HuggingFace Spaces Gradio SDK and prevents exit code 0 issues.

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import Request, HTTPException
 from fastapi.responses import JSONResponse
 import os
 import json
@@ -9,9 +9,6 @@ from datasets import Dataset
 from huggingface_hub import HfApi
 import pandas as pd
 
-# Create FastAPI app
-app = FastAPI()
-
 # Configuration
 DATASET_REPO = "assafvayner/webhook-messages"
 BATCH_SIZE = 100
@@ -26,6 +23,17 @@ latest_batch_file = None
 # HuggingFace API client
 hf_api = HfApi(token=os.environ.get("HF_TOKEN"))
 
+# Ensure dataset repo exists on startup
+try:
+    hf_api.create_repo(
+        repo_id=DATASET_REPO,
+        repo_type="dataset",
+        exist_ok=True
+    )
+    print(f"✅ Dataset repository ready: {DATASET_REPO}")
+except Exception as e:
+    print(f"⚠️  Warning: Could not create/verify dataset repo: {str(e)}")
+
 
 def save_batch_to_dataset(messages, batch_num):
     """Save a batch of webhook messages to the HuggingFace dataset as a parquet file."""
@@ -209,8 +217,8 @@ with gr.Blocks(title="HuggingFace Webhook Processor") as demo:
     timer.tick(fn=get_status, outputs=[status_text, message_count, batch_count, latest_batch])
 
 
-# Add webhook endpoint to FastAPI
-@app.post("/webhooks/hub")
+# Add webhook endpoints to Gradio's internal FastAPI app
+@demo.fastapi().post("/webhooks/hub")
 async def webhook_endpoint(request: Request):
     """
     Webhook endpoint for HuggingFace Hub events.
@@ -252,7 +260,7 @@ async def webhook_endpoint(request: Request):
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@app.get("/webhooks/health")
+@demo.fastapi().get("/webhooks/health")
 async def health_check():
     """Health check endpoint."""
     with message_lock:
@@ -264,17 +272,6 @@ async def health_check():
         }
 
 
-# Ensure dataset repo exists on startup
-try:
-    hf_api.create_repo(
-        repo_id=DATASET_REPO,
-        repo_type="dataset",
-        exist_ok=True
-    )
-    print(f"✅ Dataset repository ready: {DATASET_REPO}")
-except Exception as e:
-    print(f"⚠️  Warning: Could not create/verify dataset repo: {str(e)}")
-
-# Mount Gradio app on FastAPI
-# HuggingFace Spaces will automatically serve this app
-app = gr.mount_gradio_app(app, demo, path="/")
+# Launch the Gradio app (HuggingFace Spaces will handle this automatically)
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)