socket_client.py

import socketio
import logging
import coloredlogs
import asyncio
import streamlit as st

coloredlogs.install(level="INFO",
                    fmt="%(asctime)s %(name)s %(funcName)s() %(filename)s %(lineno)d %(levelname)-8s %(message)s")
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


END_OF_SENTENCE = "<EOS>"
sio = socketio.Client()

chunk_list = []

@sio.event
def connect():
    logger.info("[CONNECTED TO SERVER]")
    

@sio.event
def disconnect():
    logger.info("[DISCONNECTED FROM SERVER]")
    

@sio.event
def chunk_retrieve(chunk):
    chunk_list.append(chunk)