start split html template to parts

supervisely/train/debug.env

@@ -2,6 +2,7 @@ PYTHONUNBUFFERED=1
 
 DEBUG_APP_DIR="/app_debug_data"
 DEBUG_CACHE_DIR="/app_cache"
+LOG_LEVEL="debug"
 
 TASK_ID=4212
 

supervisely/train/src/gui.html

@@ -29,14 +29,9 @@
         #yolov5-train .el-tabs.el-tabs-cards .el-tabs__item { flex: 1; margin-bottom: -3px; padding: 9px 16px 13px;
         height: auto; line-height: normal; border-radius: 4px; }
     </sly-style>
-    <sly-card title="1. Input Project" subtitle="This project will be used for training">
-        <sly-field title="" description="Project">
-            <a slot="title" target="_blank"
-               :href="`/projects/${data.projectId}/datasets`">{{data.projectName}} ({{data.projectImagesCount}}
-                images)</a>
-            <sly-icon slot="icon" :options="{ imageUrl: `${data.projectPreviewUrl}` }"/>
-        </sly-field>
-    </sly-card>
+
+    <div>input_project.html</div>
+
     <sly-card class="mt15"
               title="2. Training classes"
               subtitle="Select classes, that should be used for training. Training supports only classes of shape
@@ -192,7 +187,10 @@
                 <table class="beautiful-table">
                     <thead>
                         <tr>
-                            <th>Model</th>
+                            <th>
+                                <div>Model</div>
+                                <div>pixels</div>
+                            </th>
                             <th>Params (P)</th>
                             <th>Params (V)</th>
                         </tr>

supervisely/train/src/gui_parts/input_project.html

@@ -0,0 +1,8 @@
+<sly-card title="1. Input Project" subtitle="This project will be used for training">
+    <sly-field title="" description="Project">
+        <a slot="title" target="_blank"
+           :href="`/projects/${data.projectId}/datasets`">{{data.projectName}} ({{data.projectImagesCount}}
+            images)</a>
+        <sly-icon slot="icon" :options="{ imageUrl: `${data.projectPreviewUrl}` }"/>
+    </sly-field>
+</sly-card>

supervisely/train/src/sly_train.py

@@ -81,6 +81,23 @@ def train(api: sly.Api, task_id, context, state, app_logger):
     my_app.stop()
 
 
+def compile_template(main_file, parts_directory, res_path):
+    with open(main_file, 'r') as file:
+        template = file.read()
+
+    result = str(template)
+    files = sly.fs.list_files(parts_directory, [".html"])
+    for path in files:
+        name = sly.fs.get_file_name_with_ext(path)
+        with open(path, 'r') as file:
+            html = file.read()
+        result = result.replace(f"<div>{name}</div>", html)
+
+    with open(res_path, "w") as file:
+        file.write(result)
+    return res_path
+
+
 def main():
     sly.logger.info("Script arguments", extra={
         "context.teamId": team_id,
@@ -95,10 +112,16 @@ def main():
     # read project information and meta (classes + tags)
     init_project_info_and_meta()
 
+    # compile html template to a single file from multiple parts
+    template_path = os.path.join(g.root_source_path, "supervisely/train/src/gui.html")
+    parts_dir = os.path.join(g.root_source_path, "supervisely/train/src/gui_parts")
+    res_path = os.path.join(my_app.data_dir, "gui.html")
+    compile_template(template_path, parts_dir, res_path)
+
     # init data for UI widgets
     ui.init(data, state)
 
-    my_app.run(data=data, state=state)
+    my_app.run(template_path=res_path, data=data, state=state)
 
 
 # New features: