Create app.py

app.py

@@ -0,0 +1,154 @@
+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()