Update routes.py

routes.py

@@ -760,3 +760,57 @@ def api_view_image(message_id):
     except Exception as e:
         app.logger.error(f"Error viewing image: {e}")
         return jsonify({'success': False, 'message': 'Failed to load image'}), 500
+
+from flask import Blueprint, request, jsonify, session, redirect, render_template
+import uuid
+
+routes = Blueprint("routes", __name__)
+
+# In-memory storage (use DB in production)
+users = {}
+
+# Home redirects to register
+@routes.route("/")
+def home():
+    return redirect("/register")
+
+# Serve register page
+@routes.route("/register")
+def register_page():
+    # If already logged in, redirect to chat
+    if session.get("user_id"):
+        return redirect("/chat")
+    return render_template("register.html")
+
+# Register API
+@routes.route("/api/register", methods=["POST"])
+def register_api():
+    data = request.get_json()
+    name = data.get("name")
+    email = data.get("email")
+
+    if not name or not email:
+        return jsonify({"success": False, "message": "Name and Email are required"}), 400
+
+    user_id = str(uuid.uuid4())
+    session["user_id"] = user_id
+    session["name"] = name
+
+    # Save user in memory (just for demo)
+    users[user_id] = {"name": name, "email": email}
+
+    return jsonify({"success": True, "message": "Registered successfully"})
+
+# Serve chat page
+@routes.route("/chat")
+def chat():
+    if not session.get("user_id"):
+        return redirect("/register")
+    return render_template("/chat.html", name=session.get("name"))
+
+# Logout route (optional)
+@routes.route("/logout")
+def logout():
+    session.clear()
+    return redirect("/register")
+