mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-23 21:06:39 +00:00
8df4fc3b19
Integrates real-time document collaboration with a new tool and UI. Co-authored-by: nicsins <nicsins@gmail.com>
32 lines
940 B
Python
32 lines
940 B
Python
from flask_socketio import SocketIO, emit, join_room, leave_room
|
|
from flask import request
|
|
|
|
socketio = SocketIO(cors_allowed_origins="*")
|
|
|
|
# In-memory store for documents: {doc_id: content}
|
|
documents = {}
|
|
|
|
def init_collaboration(app):
|
|
socketio.init_app(app)
|
|
|
|
@socketio.on('connect')
|
|
def handle_connect():
|
|
pass
|
|
|
|
@socketio.on('join_document')
|
|
def on_join(data):
|
|
room = data.get('doc_id', 'default')
|
|
join_room(room)
|
|
# Send current document state
|
|
content = documents.get(room, "")
|
|
emit('document_state', {'content': content}, room=request.sid)
|
|
|
|
@socketio.on('update_document')
|
|
def on_update(data):
|
|
room = data.get('doc_id', 'default')
|
|
content = data.get('content', '')
|
|
documents[room] = content
|
|
# Broadcast to others in the room
|
|
emit('document_updated', {'content': content}, room=room, include_self=False)
|
|
|
|
return socketio
|