import streamlit as st
import sys
import os
sys.path.insert(1, os.path.join(sys.path[0], '..'))
from utils import load_model, image_capture_cb, load_ocr_model

if __name__ == "__main__":
    # set page configurations and display/annotation options
    st.set_page_config(
        page_title="Circuit Sketch Recognizer",
        layout="wide"
    )
    
    with st.sidebar:
        font_size = st.slider(label="Font Size", min_value=6, max_value=64, step=1, value=24)
        line_width = st.slider(label="Bounding Box Line Thickness", min_value=1, max_value=8, step=1, value=3)

    model = load_model()
    ocr_model, ocr_processor = load_ocr_model()
    
    # Camera Input allows user to take a picture
    col1, col2 = st.columns(2)
    with col1:
        capture = st.camera_input("Take a picture with Camera")
    if capture is not None:
        image_capture_cb(model, ocr_model, ocr_processor, capture, font_size, line_width, col2)