Update app.py

app.py

@@ -3,6 +3,7 @@ import uvicorn
 import pandas as pd
 import gradio as gr
 import plotly.graph_objects as go
+from datetime import datetime
 
 # Import your local files
 import database, schemas, moderator
@@ -33,17 +34,15 @@ def update_dashboard(user, text):
     toxic_count = len(toxic_comments)
     safe_count = total_count - toxic_count
     
-    # Safety Score calculation
     safety_score = int((safe_count / total_count * 100)) if total_count > 0 else 100
     
-    # 4. FIXED CHART LOGIC: Ensure data is never zero if toxic comments exist
+    # 4. Chart Logic
     reasons = {"Identity Hate": 0, "Insult": 0, "Online Harassment": 0, "Threat": 0}
     for c in toxic_comments:
-        # If the AI reason isn't in our list, default to Online Harassment so it shows on chart
         cat = c.flagged_reason if c.flagged_reason in reasons else "Online Harassment"
         reasons[cat] += 1
 
-    # CREATE REAL DONUT CHART
+    # Create Donut Chart
     colors = ['#7B68EE', '#FF4500', '#1E90FF', '#FFA500']
     fig = go.Figure(data=[go.Pie(
         labels=list(reasons.keys()), 
@@ -52,16 +51,30 @@ def update_dashboard(user, text):
         marker_colors=colors,
         textinfo='label+percent'
     )])
-    fig.update_layout(
-        showlegend=True, 
-        margin=dict(t=10, b=10, l=10, r=10),
-        height=300,
-        paper_bgcolor='rgba(0,0,0,0)',
-        plot_bgcolor='rgba(0,0,0,0)',
-        font=dict(color="white") if os.getenv("GRADIO_THEME") == "dark" else dict(color="black")
-    )
+    fig.update_layout(showlegend=True, margin=dict(t=10, b=10, l=10, r=10), height=300, 
+                      paper_bgcolor='rgba(0,0,0,0)', plot_bgcolor='rgba(0,0,0,0)')
+
+    # 5. Create the "Visual Block" HTML (Your requested feature)
+    current_date = datetime.now().strftime("%m/%d/%Y")
+    if analysis["is_toxic"]:
+        block_html = f"""
+        <div style="background-color: #ffebee; border-radius: 10px; padding: 15px; border-left: 5px solid #f44336; margin-top: 20px;">
+            <div style="display: flex; justify-content: space-between; align-items: center;">
+                <strong style="color: #333;">👤 {user}</strong>
+                <span style="color: #888; font-size: 0.8em;">{current_date}</span>
+            </div>
+            <p style="color: #d32f2f; font-weight: bold; margin: 10px 0;">⚠️ Comment flagged as Toxic (Confidence: {analysis['score']*100:.1f}%)</p>
+            <p style="color: #f44336; font-size: 0.9em;">AI Reason: Classified as {analysis['reason'].upper()} with confidence {analysis['score']:.2f}</p>
+        </div>
+        """
+    else:
+        block_html = f"""
+        <div style="background-color: #e8f5e9; border-radius: 10px; padding: 15px; border-left: 5px solid #4caf50; margin-top: 20px;">
+            <strong style="color: #333;">👤 {user}</strong>
+            <p style="color: #2e7d32; margin: 10px 0;">✅ Comment allowed: {text}</p>
+        </div>
+        """
 
-    # Hall of Shame Data (Last 5 toxic)
     shame_query = db.query(database.Comment).filter(database.Comment.is_toxic == True).order_by(database.Comment.id.desc()).limit(5).all()
     shame_data = [[c.user, c.text, f"{c.toxicity_score:.2f}", c.flagged_reason] for c in shame_query]
     
@@ -69,14 +82,7 @@ def update_dashboard(user, text):
 
     status_label = "🔴 High Toxicity" if analysis["is_toxic"] else "🟢 NORMAL"
     
-    return (
-        safety_score,      # gr.Number
-        total_count,       # gr.Number
-        toxic_count,       # gr.Number
-        status_label,      # gr.Label
-        fig,               # gr.Plot (Donut Chart)
-        shame_data         # gr.Dataframe
-    )
+    return safety_score, total_count, toxic_count, status_label, fig, shame_data, block_html
 
 def clear_db():
     db = database.SessionLocal()
@@ -84,7 +90,7 @@ def clear_db():
     db.commit()
     db.close()
     empty_fig = go.Figure(data=[go.Pie(labels=['No Data'], values=[1], hole=.6)])
-    return 100, 0, 0, "🟢 NORMAL", empty_fig, []
+    return 100, 0, 0, "🟢 NORMAL", empty_fig, [], ""
 
 # --- UI LAYOUT ---
 with gr.Blocks(theme=gr.themes.Default(), title="Admin Intelligence Hub") as demo:
@@ -104,10 +110,13 @@ with gr.Blocks(theme=gr.themes.Default(), title="Admin Intelligence Hub") as dem
                 user_input = gr.Textbox(label="User", value="Guest")
                 msg_input = gr.Textbox(label="Message Inference", placeholder="Analyze text or emojis...")
                 submit_btn = gr.Button("ANALYZE SIGNAL", variant="primary")
+            
+            # This is where the red block notification will appear
+            moderation_alert = gr.HTML() 
         
         with gr.Column(scale=1):
             gr.Markdown("### 📊 Toxicity Classification")
-            donut_chart = gr.Plot(label="Toxicity Distribution") # Changed to gr.Plot
+            donut_chart = gr.Plot(label="Toxicity Distribution")
             current_alert = gr.Label(label="Current Threat Level")
 
     gr.Markdown("### 💀 Strategic Insights & History")
@@ -116,12 +125,12 @@ with gr.Blocks(theme=gr.themes.Default(), title="Admin Intelligence Hub") as dem
     submit_btn.click(
         update_dashboard, 
         inputs=[user_input, msg_input], 
-        outputs=[safety_score, total_signals, threats_found, current_alert, donut_chart, shame_table]
+        outputs=[safety_score, total_signals, threats_found, current_alert, donut_chart, shame_table, moderation_alert]
     )
     
     clear_btn.click(
         clear_db,
-        outputs=[safety_score, total_signals, threats_found, current_alert, donut_chart, shame_table]
+        outputs=[safety_score, total_signals, threats_found, current_alert, donut_chart, shame_table, moderation_alert]
     )
 
 if __name__ == "__main__":