Update app.py

app.py

@@ -10,6 +10,9 @@ import redis
 import threading
 import asyncio
 import gradio as gr
+import pandas as pd
+import plotly.express as px
+from datetime import datetime, timedelta
 
 # config
 # nest_asyncio.apply()
@@ -101,27 +104,62 @@ def run_alert_listener():
 alert_thread = threading.Thread(target=run_alert_listener)
 alert_thread.start()
 
+# dashboard
+def get_alert_data():
+    global alert_collection
+    pipeline = [
+        {
+            "$group": {
+                "_id": {"$dateToString": {"format": "%Y-%m-%d", "date": {"$toDate": "$CreatedAt"}}},
+                "count": {"$sum": 1}
+            }
+        },
+        {"$sort": {"_id": 1}}
+    ]
+    result = list(alert_collection.aggregate(pipeline))
+    df = pd.DataFrame(result)
+    df.columns = ['Date', 'Count']
+    df['Date'] = pd.to_datetime(df['Date'])
+    return df
+
+# Function to create a plot of alerts over time
+def create_alert_plot():
+    df = get_alert_data()
+    fig = px.line(df, x='Date', y='Count', title='Alerts Over Time')
+    fig.update_layout(xaxis_title='Date', yaxis_title='Number of Alerts')
+    return fig
+
+# Function to get the latest alert and count
+def get_latest_alert():
+    global latest_alert, alert_count
+    return latest_alert, f"Total Alerts: {alert_count}", create_alert_plot()
+
 # gradio interface
 # Gradio interface
 def get_latest_alert():
     global latest_alert, alert_count
     return latest_alert, f"Total Alerts: {alert_count}"
 
-with gr.Blocks() as app:
-    gr.Markdown("# Alert Dashboard 🔔")
+with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🚨 Alert Dashboard 🚨")
     
     with gr.Row():
-        latest_alert_box = gr.Textbox(label="Latest Alert", lines=3, interactive=False)
-        alert_count_box = gr.Textbox(label="Alert Count", interactive=False)
+        with gr.Column():
+            latest_alert_box = gr.Textbox(label="Latest Alert", lines=3, interactive=False)
+            alert_count_box = gr.Textbox(label="Alert Count", interactive=False)
+            refresh_button = gr.Button("Refresh", variant="primary")
+        
+    with gr.Row():
+        alert_plot = gr.Plot(label="Alerts Over Time")
     
-    refresh_button = gr.Button("Refresh", variant="primary")
+    refresh_button.click(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box, alert_plot])
     
-    refresh_button.click(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box])
+    app.load(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box, alert_plot])
     
-    app.load(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box])
+    # Auto-refresh every 30 seconds
+    app.load(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box, alert_plot], every=30)
 
-    # Auto-refresh every 5 seconds
-    app.load(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box], every=5)
+# Launch the app
 
 # Launch the app
 # if __name__ == "__main__":