fixed checkbox

.dockerignore

@@ -1,4 +1,4 @@
-.github/
+# .github/
 __pycache__
 *.pyc
 *.pyo

app.py

@@ -26,7 +26,7 @@ with gr.Blocks(title="HTR Riksarkivet", theme=theme, css=css) as demo:
                             label="Image to run HTR on", type="numpy", tool="editor", elem_id="image_upload"
                         ).style(height=395)
 
-                    with gr.Row():
+                    with gr.Row(equal_height=False):
                         # with gr.Group():
                         # callback = gr.CSVLogger()
                         # # hf_writer = gr.HuggingFaceDatasetSaver(HF_API_TOKEN, "htr_pipelin_flags")
@@ -35,18 +35,23 @@ with gr.Blocks(title="HTR Riksarkivet", theme=theme, css=css) as demo:
                         #     variant="secondary",
                         #     visible=True,
                         # ).style(full_width=True)
-                        radio_file_input = gr.Radio(
-                            value="Text file", choices=["Text file ", "Page XML file "], label="What kind file output?"
+                        # radio_file_input = gr.Radio(
+                        #     value="Text file", choices=["Text file ", "Page XML file "], label="What kind file output?"
+                        # )
+
+                        radio_file_input = gr.CheckboxGroup(
+                            choices=["Txt", "Page XML"],
+                            value=["Txt"],
+                            label="Output file extension",
+                            # info="Only txt and page xml is supported for now!",
                         )
-                    with gr.Row():
-                        htr_clear_button = gr.Button("", variant="Secondary")
 
                         htr_pipeline_button = gr.Button(
                             "Run HTR",
                             variant="primary",
                             visible=True,
                             elem_id="run_pipeline_button",
-                        ).style(full_width=True)
+                        ).style(full_width=False)
 
                     with gr.Group():
                         with gr.Row():

src/htr_pipeline/gradio_backend.py

@@ -38,11 +38,13 @@ class FastTrack:
             f.write(rendered_xml)
 
         xml_img = self.visualize_xml_and_return_txt(image, xml_txt)
-
-        if radio_button_choices == "Text file":
-            returned_file_extension = xml_txt
+        if len(radio_button_choices) < 2:
+            if radio_button_choices[0] == "Txt":
+                returned_file_extension = xml_txt
+            else:
+                returned_file_extension = xml_xml
         else:
-            returned_file_extension = xml_xml
+            returned_file_extension = [xml_txt, xml_xml]
 
         return xml_img, returned_file_extension, gr.update(visible=True)