Spaces:
Sleeping
Sleeping
File size: 5,093 Bytes
9f88035 | 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 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 | import uuid
import sqlite3
import gradio as gr
from datetime import datetime
from passlib.context import CryptContext
# ================= DATABASE =================
conn = sqlite3.connect("chat.db", check_same_thread=False)
cursor = conn.cursor()
cursor.execute("""CREATE TABLE IF NOT EXISTS users(email TEXT PRIMARY KEY,password TEXT)""")
cursor.execute("""CREATE TABLE IF NOT EXISTS invites(code TEXT PRIMARY KEY,user1 TEXT,user2 TEXT)""")
cursor.execute("""CREATE TABLE IF NOT EXISTS messages(id TEXT PRIMARY KEY,room TEXT,sender TEXT,message TEXT,time TEXT)""")
conn.commit()
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
# ================= AUTH =================
def register_or_login(email, password):
cursor.execute("SELECT password FROM users WHERE email=?", (email,))
data = cursor.fetchone()
if data:
if pwd_context.verify(password, data[0]):
return email, "Welcome back ✅"
return None, "Wrong password ❌"
hashed = pwd_context.hash(password)
cursor.execute("INSERT INTO users VALUES (?,?)", (email, hashed))
conn.commit()
return email, "Account created ✅"
# ================= INVITES =================
def create_invite(user):
code = str(uuid.uuid4())[:8]
cursor.execute("INSERT INTO invites VALUES (?,?,?)", (code, user, ""))
conn.commit()
return code, f"Invite Code: {code}"
def join_invite(user, code):
cursor.execute("SELECT user1,user2 FROM invites WHERE code=?", (code,))
data = cursor.fetchone()
if not data:
return None, "Invalid Code"
if data[1]:
return None, "Room already full"
cursor.execute("UPDATE invites SET user2=? WHERE code=?", (user, code))
conn.commit()
return code, f"Connected with {data[0]}"
# ================= CHAT =================
def send_message(user, room, msg):
if not msg or not room:
return
cursor.execute("INSERT INTO messages VALUES (?,?,?,?,?)",
(str(uuid.uuid4()), room, user, msg,
datetime.now().strftime("%H:%M")))
conn.commit()
def fetch_messages(room, current_user):
if not room:
return "<div class='system'>Join a room to start chatting.</div>"
cursor.execute("SELECT sender,message,time FROM messages WHERE room=? ORDER BY rowid", (room,))
chats = cursor.fetchall()
html = ""
for sender, msg, t in chats:
side = "me" if sender == current_user else "other"
html += f"""
<div class='msg {side}'>
<div class='bubble'>
<div class='text'>{msg}</div>
<div class='time'>{t}</div>
</div>
</div>
"""
return html
# ================= UI =================
custom_css = """
body {background:#0f172a;}
.container {max-width:900px;margin:auto;}
.header {text-align:center;color:white;font-size:26px;font-weight:600;margin-bottom:10px;}
.msg {display:flex;margin:8px;}
.msg.me {justify-content:flex-end;}
.msg.other {justify-content:flex-start;}
.bubble {
padding:12px 16px;
border-radius:18px;
max-width:60%;
box-shadow:0 4px 14px rgba(0,0,0,0.2);
}
.me .bubble {background:#22c55e;color:white;}
.other .bubble {background:#1e293b;color:white;}
.time {font-size:10px;opacity:0.6;margin-top:4px;text-align:right;}
.chatbox {
height:420px;
overflow-y:auto;
padding:10px;
background:#020617;
border-radius:14px;
}
.system {color:#94a3b8;text-align:center;margin-top:20px;}
"""
with gr.Blocks(css=custom_css) as app:
user_state = gr.State()
room_state = gr.State()
gr.Markdown("<div class='header'>🔐 DuoSecure — Private 2-Person Chat</div>")
with gr.Tab("Login"):
email = gr.Textbox(label="Email")
password = gr.Textbox(type="password", label="Password")
btn = gr.Button("Enter")
status = gr.Markdown()
btn.click(register_or_login, [email, password], [user_state, status])
with gr.Tab("Invitation"):
create_btn = gr.Button("Create Private Room")
code_box = gr.Textbox(label="Room Code")
join_btn = gr.Button("Join Room")
invite_status = gr.Markdown()
create_btn.click(create_invite, user_state, [code_box, invite_status])
join_btn.click(join_invite, [user_state, code_box], [room_state, invite_status])
with gr.Tab("Chat"):
chat_html = gr.HTML(elem_classes="chatbox")
msg = gr.Textbox(placeholder="Type message and press Enter...", show_label=False)
msg.submit(send_message, [user_state, room_state, msg], None)
timer = gr.Timer(1)
timer.tick(fetch_messages, [room_state, user_state], chat_html)
with gr.Tab("Voice / Video"):
gr.Markdown("Allow camera & microphone to start secure conversation.")
gr.HTML("""
<video id="v" autoplay muted style="width:70%;border-radius:14px;"></video>
<script>
navigator.mediaDevices.getUserMedia({video:true,audio:true})
.then(s=>{document.getElementById("v").srcObject=s;});
</script>
""")
app.launch()
|