Update app.py

app.py

@@ -14,10 +14,18 @@ from PIL import Image
 import requests
 import random
 import string
+from flask_socketio import SocketIO, emit, join_room, leave_room
+import threading
 
 app = Flask(__name__)
 CORS(app)
 
+socketio = SocketIO(app, cors_allowed_origins="*", async_mode='threading')
+
+# ADD THIS new storage for chat rooms:
+CHAT_ROOMS = {}  # { room_id: { admin_session, active_sessions, settings, created_at, expires_at } }
+CHAT_MESSAGES = {} 
+
 # In-memory storage
 SECRETS = {}  # { id: { data, file_data, file_type, expire_at, view_once, theme, analytics, etc. } }
 SHORT_LINKS = {}  # { short_id: full_id }
@@ -519,6 +527,168 @@ def cleanup_expired():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+# ADD THESE CHAT ENDPOINTS:
+
+@app.route("/api/chat/create", methods=["POST"])
+def create_chat_room():
+    try:
+        form = request.form
+        ttl = int(form.get("ttl", 3600))  # Default 1 hour
+        max_receivers = int(form.get("max_receivers", 5))
+        password = form.get("password", "")
+        allow_files = form.get("allow_files", "true").lower() == "true"
+        
+        room_id = str(uuid.uuid4())
+        admin_session = str(uuid.uuid4())  # Generate admin session
+        
+        CHAT_ROOMS[room_id] = {
+            "admin_session": admin_session,
+            "created_at": time.time(),
+            "expires_at": time.time() + ttl,
+            "settings": {
+                "max_receivers": max_receivers,
+                "password": password,
+                "allow_files": allow_files,
+                "burn_on_admin_exit": True
+            },
+            "active_sessions": {},
+            "receiver_counter": 0
+        }
+        
+        CHAT_MESSAGES[room_id] = []
+        
+        base_url = request.host_url.rstrip('/')
+        chat_url = f"{base_url}/tools/sharelock?chat={room_id}&admin={admin_session}"
+        
+        return jsonify({
+            "room_id": room_id,
+            "admin_session": admin_session,
+            "chat_url": chat_url,
+            "expires_at": CHAT_ROOMS[room_id]["expires_at"]
+        })
+        
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/api/chat/join/<room_id>")
+def join_chat_room(room_id):
+    try:
+        password = request.args.get("password", "")
+        admin_session = request.args.get("admin", "")
+        
+        if room_id not in CHAT_ROOMS:
+            return jsonify({"error": "Chat room not found"}), 404
+            
+        room = CHAT_ROOMS[room_id]
+        
+        # Check if expired
+        if time.time() > room["expires_at"]:
+            return jsonify({"error": "Chat room has expired"}), 410
+            
+        # Check password
+        if room["settings"]["password"] and password != room["settings"]["password"]:
+            return jsonify({"error": "Wrong password"}), 403
+            
+        # Determine role
+        if admin_session == room["admin_session"]:
+            role = "admin"
+            session_id = admin_session
+        else:
+            # Check max receivers
+            active_receivers = sum(1 for s in room["active_sessions"].values() if s["role"] == "receiver")
+            if active_receivers >= room["settings"]["max_receivers"]:
+                return jsonify({"error": "Chat room is full"}), 403
+                
+            role = "receiver"
+            session_id = str(uuid.uuid4())
+            room["receiver_counter"] += 1
+            
+        return jsonify({
+            "session_id": session_id,
+            "role": role,
+            "receiver_number": room["receiver_counter"] if role == "receiver" else None,
+            "room_settings": room["settings"],
+            "expires_at": room["expires_at"]
+        })
+        
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+# ADD WEBSOCKET EVENTS:
+@socketio.on('join_chat')
+def handle_join_chat(data):
+    room_id = data['room_id']
+    session_id = data['session_id']
+    role = data['role']
+    
+    if room_id not in CHAT_ROOMS:
+        return
+        
+    join_room(room_id)
+    
+    # Add to active sessions
+    CHAT_ROOMS[room_id]["active_sessions"][session_id] = {
+        "role": role,
+        "receiver_number": data.get('receiver_number'),
+        "joined_at": time.time(),
+        "last_seen": time.time()
+    }
+    
+    # Notify others
+    emit('user_joined', {
+        'role': role,
+        'receiver_number': data.get('receiver_number'),
+        'active_count': len(CHAT_ROOMS[room_id]["active_sessions"])
+    }, room=room_id, include_self=False)
+
+@socketio.on('send_chat_message')
+def handle_chat_message(data):
+    room_id = data['room_id']
+    session_id = data['session_id']
+    
+    if room_id not in CHAT_ROOMS or session_id not in CHAT_ROOMS[room_id]["active_sessions"]:
+        return
+        
+    session = CHAT_ROOMS[room_id]["active_sessions"][session_id]
+    
+    message = {
+        "id": str(uuid.uuid4()),
+        "sender_role": session["role"],
+        "sender_number": session.get("receiver_number"),
+        "content": data['content'],
+        "timestamp": time.time(),
+        "type": "text"
+    }
+    
+    CHAT_MESSAGES[room_id].append(message)
+    
+    # Broadcast to room
+    emit('new_message', message, room=room_id)
+
+@socketio.on('leave_chat')
+def handle_leave_chat(data):
+    room_id = data['room_id']
+    session_id = data['session_id']
+    
+    if room_id in CHAT_ROOMS and session_id in CHAT_ROOMS[room_id]["active_sessions"]:
+        session = CHAT_ROOMS[room_id]["active_sessions"][session_id]
+        del CHAT_ROOMS[room_id]["active_sessions"][session_id]
+        
+        # If admin left and burn_on_admin_exit is true
+        if session["role"] == "admin" and CHAT_ROOMS[room_id]["settings"]["burn_on_admin_exit"]:
+            emit('room_closing', {'reason': 'Admin left the room'}, room=room_id)
+            del CHAT_ROOMS[room_id]
+            if room_id in CHAT_MESSAGES:
+                del CHAT_MESSAGES[room_id]
+        else:
+            emit('user_left', {
+                'role': session["role"],
+                'receiver_number': session.get("receiver_number"),
+                'active_count': len(CHAT_ROOMS[room_id]["active_sessions"])
+            }, room=room_id)
+        
+        leave_room(room_id)
+
 # Error handlers
 @app.errorhandler(404)
 def not_found(error):
@@ -540,6 +710,7 @@ if __name__ == "__main__":
     print("   ✅ Multiple themes")
     print("   ✅ Password hints")
     print("   ✅ verify_only parameter support")
+    print("   ✅ Real-time chat rooms")  # ADD THIS LINE
     print("🚀 Server running on http://0.0.0.0:7860")
     
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    socketio.run(app, host="0.0.0.0", port=7860, debug=True)  # CHANGE THIS LINE