File size: 833 Bytes
3bc9036 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import queue
import threading
from streamlit.runtime.scriptrunner import add_script_run_ctx
from src.ocr import ocr_thread
from src.chatbot import run_chatbot
class SystemInitializer:
def initialize_system(self):
"""Initializes queues and starts OCR thread."""
queues = {
'frame_queue': queue.Queue(maxsize=1),
'text_queue': queue.Queue(maxsize=1),
'annotation_queue': queue.Queue(maxsize=1),
'prompt_queue': queue.Queue(maxsize=1),
'ppx_queue': queue.Queue(maxsize=1)
}
ocr_thread_with_ctx = threading.Thread(target=ocr_thread, args=(queues['frame_queue'], queues['text_queue']))
add_script_run_ctx(ocr_thread_with_ctx)
ocr_thread_with_ctx.start()
return queues
def run_chatbot(self):
run_chatbot() |