Spaces:
Build error
Build error
File size: 1,767 Bytes
efc7a8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from pathlib import Path
from typing import Set, Union
import gradio as gr
class WebhookGradioApp:
"""
```py
from gradio_webhooks import WebhookGradioApp
app = WebhookGradioApp()
@app.add_webhook("/test_webhook")
async def hello():
return {"in_gradio": True}
app.block_thread()
```
"""
def __init__(self, landing_path: Union[str, Path] = "README.md") -> None:
# Use README.md as landing page or provide any markdown file
landing_path = Path(landing_path)
landing_content = landing_path.read_text()
if landing_path.name == "README.md":
landing_content = landing_content.split("---")[-1].strip()
# Simple gradio app with landing content
block = gr.Blocks()
with block:
gr.Markdown(landing_content)
# Launch gradio app:
# - as non-blocking so that webhooks can be added afterwards
# - as shared if launch locally (to receive webhooks)
app, _, _ = block.launch(prevent_thread_lock=True, share=not block.is_space)
self.gradio_app = block
self.fastapi_app = app
self.webhook_paths: Set[str] = set()
def add_webhook(self, path: str):
self.webhook_paths.add(path)
return self.fastapi_app.post(path)
def block_thread(self) -> None:
url = (
self.gradio_app.share_url
if self.gradio_app.share_url is not None
else self.gradio_app.local_url
).strip("/")
print("\nWebhooks are correctly setup and ready to use:")
print("\n".join(f" - POST {url}{webhook}" for webhook in self.webhook_paths))
print(f"Checkout {url}/docs for more details.")
self.gradio_app.block_thread()
|