Spaces:
Sleeping
Sleeping
File size: 4,323 Bytes
fabbeef | 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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | import gradio as gr
import time
import pandas as pd
from transformers import pipeline
# =========================
# ADVANCED SECURITY CONFIG
# =========================
# User database: format {user_id: password}
users_db = {
"user1": "pass123",
"user2": "abc456"
}
# Blocked users set
blocked_users = set()
# Activity log
activity_log = pd.DataFrame(columns=["timestamp", "user_id", "message", "status"])
# Rate limiter tracking
user_activity = {}
# Load Hugging Face Toxicity Model
moderator = pipeline("text-classification", model="unitary/toxic-bert", top_k=None)
# =========================
# SECURITY FUNCTIONS
# =========================
def authenticate(user_id, password):
return users_db.get(user_id) == password
def is_user_allowed(user_id):
return user_id not in blocked_users
def can_user_send(user_id):
now = time.time()
if user_id not in user_activity:
user_activity[user_id] = []
# Keep only last 60 seconds
user_activity[user_id] = [t for t in user_activity[user_id] if now - t < 60]
if len(user_activity[user_id]) >= 5: # limit 5 messages/min
return False
user_activity[user_id].append(now)
return True
def is_text_safe(text):
result = moderator(text)[0]
for label in result:
if label["label"] == "TOXIC" and label["score"] > 0.7:
return False
return True
def log_activity(user_id, message, status):
global activity_log
activity_log = pd.concat([activity_log, pd.DataFrame([{
"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
"user_id": user_id,
"message": message,
"status": status
}])], ignore_index=True)
# =========================
# MAIN APP FUNCTIONS
# =========================
def secure_chat(user_id, password, message):
if not authenticate(user_id, password):
log_activity(user_id, message, "Failed Authentication")
return "❌ Invalid user ID or password."
if not is_user_allowed(user_id):
log_activity(user_id, message, "Blocked User")
return "🚫 You are blocked from using this app."
if not can_user_send(user_id):
log_activity(user_id, message, "Rate Limited")
return "⏳ Too many requests. Wait 1 minute."
if not is_text_safe(message):
blocked_users.add(user_id)
log_activity(user_id, message, "Blocked for Toxicity")
return "⚠️ Inappropriate content detected. You are now blocked."
log_activity(user_id, message, "Accepted")
return f"✅ Message Accepted: {message}"
# Admin panel functions
def unblock_user(admin_password, user_id):
if admin_password != "admin123": # simple admin password
return "❌ Invalid admin password."
blocked_users.discard(user_id)
return f"✅ User {user_id} unblocked."
def view_logs(admin_password):
if admin_password != "admin123":
return "❌ Invalid admin password."
return activity_log.tail(20).to_string(index=False)
# =========================
# GRADIO UI
# =========================
with gr.Blocks(theme=gr.themes.Soft()) as app:
gr.Markdown("## 🔐 Advanced Secure AI App")
gr.Markdown("This app includes advanced security features with AI moderation, login, and admin dashboard.")
with gr.Tab("User Interface"):
user_id_input = gr.Textbox(label="User ID")
password_input = gr.Textbox(label="Password", type="password")
message_input = gr.Textbox(label="Enter your message")
output = gr.Textbox(label="Response")
send_btn = gr.Button("Send")
send_btn.click(secure_chat, inputs=[user_id_input, password_input, message_input], outputs=output)
with gr.Tab("Admin Panel"):
admin_pass_input = gr.Textbox(label="Admin Password", type="password")
unblock_user_input = gr.Textbox(label="User ID to Unblock")
unblock_output = gr.Textbox(label="Admin Response")
unblock_btn = gr.Button("Unblock User")
unblock_btn.click(unblock_user, inputs=[admin_pass_input, unblock_user_input], outputs=unblock_output)
log_output = gr.Textbox(label="Recent Activity Logs")
log_btn = gr.Button("View Logs")
log_btn.click(view_logs, inputs=[admin_pass_input], outputs=log_output)
app.launch()
|