feat: remove unnecessary options for model selection

ui/constants.py

@@ -1,21 +1,19 @@
 """
 Constants used in the DuoSubs Hugging Face Web UI.
 
-This module defines file paths, supported subtitle formats, model precision options,
-and default values for use in the UI and other components.
+This module defines file paths, supported subtitle formats, and default values for use 
+in the UI and other components.
 """
 
 from pathlib import Path
 
-from duosubs import ModelPrecision, SubtitleFormat
+from duosubs import SubtitleFormat
 
 TITLE_HTML = Path(__file__).parent.parent / "assets" / "title.html"
 
 SUB_EXT_LIST: list[str] = [f.value for f in SubtitleFormat]
 SUB_EXT_LIST_WITH_DOT: list[str] = [f".{ext}" for ext in SUB_EXT_LIST]
-PRECISION_LIST: list[str] = [f.value for f in ModelPrecision]
 
-DEFAULT_PRECISION = ModelPrecision.FLOAT32.value
 DEFAULT_SUB_EXT = SubtitleFormat.ASS.value
 
 MODEL_NAME_LIST = [

ui/events.py

@@ -13,7 +13,6 @@ from duosubs import (
     LoadSubsError,
     MergeArgs,
     MergeSubsError,
-    ModelPrecision,
     OmitFile,
     SaveSubsError,
     SubtitleFormat,
@@ -30,7 +29,6 @@ def start_merging(
         secondary_subtitles: str,
         model_name: str,
         batch_size: int,
-        model_precision: str,
         ignore_non_overlap: bool,
         retain_newline: bool,
         secondary_above_primary: bool,
@@ -52,7 +50,6 @@ def start_merging(
         secondary_subtitles (str): Path to secondary subtitle file.
         model_name (str): Name of the model to use.
         batch_size (int): Batch size for inference.
-        model_precision (str): Precision mode for inference.
         ignore_non_overlap (bool): Whether to ignore non-overlapping subtitles.
         retain_newline (bool): Whether to retain newlines in output.
         secondary_above_primary (bool): Whether to place secondary subtitle above 
@@ -77,7 +74,6 @@ def start_merging(
     args = MergeArgs(
         primary=primary_subtitles,
         secondary=secondary_subtitles,
-        model_precision=ModelPrecision(model_precision),
         batch_size=int(batch_size),
         ignore_non_overlap_filter=ignore_non_overlap,
         retain_newline=retain_newline,

ui/layout.py

@@ -10,10 +10,8 @@ file handling.
 import gradio as gr
 
 from .constants import (
-    DEFAULT_PRECISION,
     DEFAULT_SUB_EXT,
     MODEL_NAME_LIST,
-    PRECISION_LIST,
     SUB_EXT_LIST,
     SUB_EXT_LIST_WITH_DOT,
     TITLE_HTML,
@@ -56,7 +54,7 @@ def create_main_gr_blocks_ui(
         title_content = open_html(TITLE_HTML)
         gr.HTML(title_content)
         with gr.Row():
-            with gr.Column(scale=9):
+            with gr.Column():
                 (
                     primary_file,
                     secondary_file,
@@ -64,28 +62,26 @@ def create_main_gr_blocks_ui(
                     merge_button,
                     cancel_button
                 ) = _create_subtitles_io_block()
-            with gr.Column(scale=11):
+            with gr.Column():
                 gr.Markdown("### 🔧 Configurations")
-                with gr.Tab("Model & Device"):
-                    (
-                        model_name,
-                        batch_size,
-                        model_precision
-                    ) =  _create_model_configurations_block()
-                with gr.Tab("Alignment Behavior"):
-                    ignore_non_overlap = _create_alignment_behaviour_block()
-                with gr.Tab("Output Styling"):
-                    (
-                        retain_newline,
-                        secondary_above_primary
-                    ) = _create_output_styling_block()
-                with gr.Tab("File Exports"):
-                    (
-                        omit_subtitles,
-                        combined_format,
-                        primary_format,
-                        secondary_format
-                    ) = _create_file_exports_block()
+                with gr.Accordion("Basic"):
+                    with gr.Tab("Model Selection"):
+                        model_name =  _create_model_configurations_block()
+                    with gr.Tab("Output Styling"):
+                        (
+                            retain_newline,
+                            secondary_above_primary
+                        ) = _create_output_styling_block()
+                    with gr.Tab("File Exports"):
+                        (
+                            omit_subtitles,
+                            combined_format,
+                            primary_format,
+                            secondary_format
+                        ) = _create_file_exports_block()
+                with gr.Accordion("Advanced", open=False):
+                    with gr.Tab("Alignment Behavior"):
+                        ignore_non_overlap = _create_alignment_behaviour_block()
 
         omit_subtitles.change(
             fn=validate_excluded_subtitle_file,
@@ -102,8 +98,6 @@ def create_main_gr_blocks_ui(
                 primary_file,
                 secondary_file,
                 model_name,
-                batch_size,
-                model_precision,
                 ignore_non_overlap, 
                 retain_newline,
                 secondary_above_primary,
@@ -166,43 +160,26 @@ def _create_subtitles_io_block(
     
     return primary_file, secondary_file, merged_file, merge_button, cancel_button
 
-def _create_model_configurations_block() -> tuple[gr.Dropdown, gr.Slider, gr.Dropdown]:
+def _create_model_configurations_block() -> gr.Dropdown:
     """
-    Creates model and device configuration UI components.
+    Creates model selection configuration UI components.
 
-    This function sets up the UI for selecting the model name, batch size, and model 
-    precision.
+    This function sets up the UI for selecting the model name.
 
     Returns:
-        tuple[gradio.Dropdown, gradio.Slider, gradio.Dropdown]:
-        - model_name
-        - batch_size
-        - model_precision
+        gradio.Dropdown: model name
     """
     with gr.Column():
         model_name = gr.Dropdown(
             choices=MODEL_NAME_LIST,
             label="Sentence Transformer Model",
             value=MODEL_NAME_LIST[0],
-            info="Model for computing subtitle sentence similarity."
-        )
-
-    with gr.Row():
-        batch_size = gr.Slider(
-            label="Batch Size",
-            minimum=8,
-            maximum=256,
-            value=256,
-            step=1,
-            info="Number of sentences to process in parallel"
-        )
-        model_precision = gr.Dropdown(
-            choices=PRECISION_LIST,
-            label="Model Precision",
-            value=DEFAULT_PRECISION,
-            info="Precision mode for inference"
+            info=(
+                "Model for computing subtitle sentence similarity, "
+                "with float32 precision"
+            )
         )
-    return model_name, batch_size, model_precision
+    return model_name
 
 def _create_alignment_behaviour_block() -> gr.Checkbox:
     """
@@ -296,8 +273,6 @@ def wrapped_start_merging(
         primary_subtitles: str,
         secondary_subtitles: str,
         model_name: str,
-        batch_size: int,
-        model_precision: str,
         ignore_non_overlap: bool,
         retain_newline: bool,
         secondary_above_primary: bool,
@@ -307,7 +282,7 @@ def wrapped_start_merging(
         secondary_format: str,
         cancel_state: list[bool],
         progress: gr.Progress | None = None
-) -> str | None:
+    ) -> str | None:
     """
     Wrapper for starting the merging process with all required arguments.
 
@@ -315,8 +290,6 @@ def wrapped_start_merging(
         primary_subtitles (str): Path to primary subtitle file.
         secondary_subtitles (str): Path to secondary subtitle file.
         model_name (str): Name of the model to use.
-        batch_size (int): Batch size for inference.
-        model_precision (str): Precision mode for inference.
         ignore_non_overlap (bool): Whether to ignore non-overlapping subtitles.
         retain_newline (bool): Whether to retain newlines in output.
         secondary_above_primary (bool): Whether to place secondary subtitle above 
@@ -341,8 +314,7 @@ def wrapped_start_merging(
         primary_subtitles,
         secondary_subtitles,
         model_name,
-        batch_size,
-        model_precision,
+        256,
         ignore_non_overlap,
         retain_newline,
         secondary_above_primary,