Update app.py

app.py

@@ -105,34 +105,11 @@ 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()
+    return latest_alert, alert_count
 
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🚨 Alert Dashboard 🚨")
@@ -140,18 +117,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     with gr.Row():
         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)
+            alert_count_box = gr.Number(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.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)
+    app.load(get_latest_alert, inputs=None, outputs=[latest_alert_box, alert_count_box], every=30)
 
 # Launch the app