Update app.py

app.py

@@ -31,6 +31,15 @@ CREATE TABLE IF NOT EXISTS messages(
 ''')
 conn.commit()
 
+# ---------------- Default Admin ----------------
+DEFAULT_ADMIN = "admin"
+DEFAULT_PASSWORD = "1234"
+cursor.execute("SELECT * FROM users WHERE username=?", (DEFAULT_ADMIN,))
+if not cursor.fetchone():
+    cursor.execute("INSERT INTO users(username,password_hash) VALUES (?,?)",
+                   (DEFAULT_ADMIN, hashlib.sha256(DEFAULT_PASSWORD.encode()).hexdigest()))
+    conn.commit()
+
 # ---------------- Utility ----------------
 def hash_password(password):
     return hashlib.sha256(password.encode()).hexdigest()
@@ -53,42 +62,43 @@ def login(username, password):
     else:
         return "Login failed!", gr.update(visible=False), None
 
-def send_message(sender_id, receiver_username, message):
-    cursor.execute("SELECT id FROM users WHERE username=?", (receiver_username,))
-    receiver = cursor.fetchone()
-    if not receiver:
-        return "Receiver not found!"
-    receiver_id = receiver[0]
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+def send_message(sender_id, receiver_id, message):
+    timestamp = datetime.now().strftime("%H:%M:%S")
     cursor.execute("INSERT INTO messages(sender_id,receiver_id,message,timestamp) VALUES (?,?,?,?)",
                    (sender_id, receiver_id, message, timestamp))
     conn.commit()
-    return "Message sent!"
-
-def get_messages(user_id, other_username):
-    cursor.execute("SELECT id FROM users WHERE username=?", (other_username,))
-    other = cursor.fetchone()
-    if not other:
-        return "No such user!"
-    other_id = other[0]
+    return ""
+
+def get_messages_html(user_id, receiver_id):
     cursor.execute("""
         SELECT sender_id, message, timestamp FROM messages
         WHERE (sender_id=? AND receiver_id=?) OR (sender_id=? AND receiver_id=?)
         ORDER BY id ASC
-    """, (user_id, other_id, other_id, user_id))
+    """, (user_id, receiver_id, receiver_id, user_id))
     msgs = cursor.fetchall()
     result = ""
     for s_id, msg, ts in msgs:
-        sender = "You" if s_id == user_id else other_username
-        result += f"{sender} ({ts}): {msg}\n"
+        if s_id == user_id:
+            # Your messages on the right
+            result += f'<div style="text-align:right; margin:5px;"><span style="background:#DCF8C6; padding:8px; border-radius:10px; display:inline-block;">{msg} <br><small>{ts}</small></span></div>'
+        else:
+            # Friend messages on the left
+            result += f'<div style="text-align:left; margin:5px;"><span style="background:#FFF; padding:8px; border-radius:10px; display:inline-block;">{msg} <br><small>{ts}</small></span></div>'
     if not result:
         return "No messages yet."
     return result
 
+def get_all_users(exclude_id):
+    cursor.execute("SELECT username FROM users WHERE id!=?", (exclude_id,))
+    users = [u[0] for u in cursor.fetchall()]
+    if not users:
+        users = ["No other users"]
+    return gr.Dropdown.update(choices=users)
+
 # ---------------- Gradio ----------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 💬 Mini WhatsApp Prototype")
-    
+    gr.Markdown("## 💬 WhatsApp-Style Prototype")
+
     # Login/Register
     with gr.Row():
         username = gr.Textbox(label="Username")
@@ -96,22 +106,45 @@ with gr.Blocks() as demo:
         login_btn = gr.Button("Login")
         register_btn = gr.Button("Register")
         login_status = gr.Label("")
-    
+
     chat_ui = gr.Column(visible=False)
     current_user_id = gr.State(value=None)
-    
+
     login_btn.click(login, [username, password], [login_status, chat_ui, current_user_id])
     register_btn.click(register, [username, password], login_status)
-    
-    # Chat UI
+
+    # ---------------- Chat UI ----------------
     with chat_ui:
-        receiver = gr.Textbox(label="Send to (username)")
-        message = gr.Textbox(label="Message", lines=2)
+        with gr.Row():
+            receiver_dropdown = gr.Dropdown(label="Select User")
+        message_input = gr.Textbox(label="Message", lines=2)
         send_btn = gr.Button("Send")
-        refresh_btn = gr.Button("Refresh Chat")
-        chat_display = gr.Textbox(label="Chat", lines=15)
-        
-        send_btn.click(send_message, [current_user_id, receiver, message], chat_display)
-        refresh_btn.click(get_messages, [current_user_id, receiver], chat_display)
+        chat_display = gr.HTML(label="Chat")
+
+        # Update dropdown dynamically
+        receiver_dropdown.change(get_all_users, current_user_id, receiver_dropdown)
+
+        # Send message
+        def send_and_refresh(sender_id, receiver, msg):
+            if not receiver or not msg:
+                return ""
+            cursor.execute("SELECT id FROM users WHERE username=?", (receiver,))
+            r = cursor.fetchone()
+            if not r:
+                return "Receiver not found."
+            send_message(sender_id, r[0], msg)
+            return get_messages_html(sender_id, r[0])
+
+        send_btn.click(send_and_refresh, [current_user_id, receiver_dropdown, message_input], chat_display)
+
+        # Real-time refresh
+        def refresh(user_id, receiver):
+            if not receiver:
+                return ""
+            cursor.execute("SELECT id FROM users WHERE username=?", (receiver,))
+            r = cursor.fetchone()
+            if not r:
+                return "Receiver not found."
+            return get_messages_html(user_id, r[0])
 
-demo.launch()
+        chat_display.render(refresh, [current_user_id, receiver_dropdown], chat_display, every=2)