Removed health and added webhook

app.py

@@ -1,22 +1,28 @@
 import os
 from pathlib import Path
-from datetime import datetime
 
 import gradio as gr
 from bs4 import BeautifulSoup
+from huggingface_hub import WebhookPayload, WebhooksServer
 from rich.console import Console
 from rich.syntax import Syntax
 
+from utilities.my_logger import setup_logger
+
 proj_dir = Path(__name__).parent
 
-subreddit = os.environ["SUBREDDIT"]
-username = os.environ["USERNAME"]
-dataset_name = f"{username}/dataset-creator-reddit-{subreddit}"
+SUBREDDIT = os.environ["SUBREDDIT"]
+USERNAME = os.environ["USERNAME"]
+DATASET_NAME = f"{USERNAME}/dataset-creator-reddit-{SUBREDDIT}"
 
-frequency = os.environ.get("FREQUENCY", '').lower()
-if frequency not in ["daily", "hourly"]:
+FREQUENCY = os.environ.get("FREQUENCY", '').lower()
+if FREQUENCY not in ["daily", "hourly"]:
     raise gr.Error("FREQUENCY environment variable must be 'daily' or 'hourly'")
 
+SECRET = os.getenv("HF_WEBHOOK_SECRET")
+
+logger = setup_logger(__name__)
+
 
 def log_file_to_html_string():
     log_file = "mylog.log"
@@ -63,8 +69,8 @@ pre, code {
 
 intro_md = f"""
 # Reddit Dataset Creator
-This is a reddit dataset creator which builds and updates [{dataset_name}](https://huggingface.co/datasets/{dataset_name}) 
-which pulls from [/r/{subreddit}](http://www.reddit.com/r/{subreddit}). Check the dataset for more details. 
+This is a reddit dataset creator which builds and updates [{DATASET_NAME}](https://huggingface.co/datasets/{DATASET_NAME}) 
+which pulls from [/r/{SUBREDDIT}](http://www.reddit.com/r/{SUBREDDIT}). Check the dataset for more details. 
 
 As shown in the below diagram this space pulls data from reddit via [PRAW](https://praw.readthedocs.io/en/stable/), processes it, and puts it in a corresponding dataset.
 """
@@ -98,22 +104,15 @@ log files. I use gradio for `app` and map that to the open port of huggingface s
 
 The only communication between `app` and `main` is the log file.
 """
-def health(text):
-    # Get the current date and time
-    current_time = datetime.now()
-
-    # Print it in the format YYYY-MM-DD HH:MM:SS
-    print(current_time.strftime("%Y-%m-%d %H:%M:%S"))
-    return "Healthy"
 
-with gr.Blocks() as demo:
+with gr.Blocks() as ui:
     with gr.Tab("Application"):
         gr.Markdown(intro_md)
         gr.Image(proj_dir / 'media' / 'reddit_scraper.drawio.png')
         gr.Markdown("# Logs")
         output = gr.HTML(log_file_to_html_string, every=1)
-        demo.load(None,
-              _js="""
+        ui.load(None,
+                _js="""
         () => {
             document.body.classList.toggle('dark');
             document.querySelector('gradio-app').style.backgroundColor = 'var(--color-background-primary)'
@@ -130,7 +129,15 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_text = gr.Textbox(label="Output Text")
 
-        health_btn.click(health, inputs=input_text, outputs=output_text, api_name="health")
+app = WebhooksServer(ui=ui, webhook_secret=SECRET)
+
+
+@app.add_webhook("/community")
+async def community(payload: WebhookPayload):
+    if payload.event.scope.startswith("repo"):
+        logger.info(f"Webhook received from {DATASET_NAME} indicating a repo {payload.event.action}")
+
 
 if __name__ == '__main__':
-    demo.queue().launch(server_name="0.0.0.0", show_error=True, server_port=7860)
+    app.run()
+    # ui.queue().launch(server_name="0.0.0.0", show_error=True, server_port=7860)