import gradio as gr from ui import UIComponents def create_app(): ui = UIComponents() with gr.Blocks(css=ui.custom_css, title="AI Helmet Detection System") as demo: ui.create_header() with gr.Tabs(): with gr.TabItem("Detection"): with gr.Row(): settings_components = ui.create_settings_panel() results_components = ui.create_results_panel() ui.create_examples_tab( settings_components['input_image'], results_components ) ui.create_info_tab() ui.setup_event_handlers(settings_components, results_components) return demo if __name__ == "__main__": app = create_app() app.launch( debug=True, server_name="0.0.0.0", server_port=7860, show_api=False )