Update app.py

app.py

@@ -8,8 +8,7 @@ import marimo
 __generated_with = "0.13.4"
 app = marimo.App(
     width="full",
-    app_title="ExcelSheetProcessor-9000",
-    html_head_file="ExcelSheetProcessor-9000.html",
+    app_title="watsonx-SheetProcessor-9000",
 )
 
 
@@ -181,15 +180,6 @@ def _(mo, prompt_stack):
     return
 
 
-@app.cell
-def _(mo, results_stack):
-    ui_accordion_section_7 = mo.accordion(
-        {"Section 7: **Run and View Results**": results_stack}
-    ) 
-    ui_accordion_section_7
-    return
-
-
 @app.cell
 def _(client_instantiation_form, os):
     if client_instantiation_form.value:
@@ -338,8 +328,6 @@ def _(file, io, mo, os, pd):
 
 @app.cell
 def _(mo, table_dataframe_raw):
-    _ = table_dataframe_raw
-
     if not table_dataframe_raw.empty:
         apply_header_readjustment = mo.ui.checkbox(label="Activate Header Adjustment")
     else:
@@ -348,9 +336,13 @@ def _(mo, table_dataframe_raw):
 
 
 @app.cell
-def _(apply_header_readjustment, mo, sheet_selector, table_dataframe_raw):
-    _ = table_dataframe_raw
+def _(mo):
+    show_variable_sidebar = mo.ui.checkbox(label="Show Sidebar with Input Variables", value=False)
+    return (show_variable_sidebar,)
 
+
+@app.cell
+def _(apply_header_readjustment, mo, sheet_selector, table_dataframe_raw):
     if not table_dataframe_raw.empty:
         if apply_header_readjustment.value:
             header_row = mo.ui.number(
@@ -439,11 +431,10 @@ def _(
 
 @app.cell
 def _(mo, pd, table_column_names, table_dataframe):
-    _ = table_dataframe
-
     if not table_dataframe.empty:
         center_column_text = {col: "center" for col in table_column_names}
         table = mo.ui.table(table_dataframe, 
+                                      show_column_summaries=False,
                                       initial_selection=[0],
                                       wrapped_columns=table_column_names, 
                                       text_justify_columns=center_column_text, 
@@ -462,8 +453,6 @@ def _(mo, pd, table_column_names, table_dataframe):
 
 @app.cell
 def _(create_parameter_table, table_column_names, table_dataframe):
-    _ = table_dataframe
-
     if not table_dataframe.empty:
         column_selector = create_parameter_table(
                                                label="Select the Columns to Process",
@@ -590,8 +579,15 @@ def _(GenParams, ModelInference, ast, client, llm_parameters, selected_model):
 
 
 @app.cell
-def _(load_templates):
-    template_folder = "./prompt_templates"
+def _(mo):
+    prompt_template_folders = get_subfolder_paths("watsonx Sheet Processor MNB/prompt_templates", depth=1)
+    prompt_template_model_paths = mo.ui.dropdown(options=prompt_template_folders, label="Prompt Template Variants *(Select Based on your Model)*")
+    return (prompt_template_model_paths,)
+
+
+@app.cell
+def _(load_templates, prompt_template_model_paths):
+    template_folder = str(prompt_template_model_paths.value)
     templates = load_templates(template_folder)
     return (templates,)
 
@@ -603,8 +599,14 @@ def _(mo):
 
 
 @app.cell
-def _(mo, prompt_number_slider, template_selector_stack):
-    prompt_setup_stack = mo.hstack([prompt_number_slider, template_selector_stack], justify="space-around")
+def _(mo, prompt_number_slider, prompt_template_model_paths):
+    prompt_mechanics_stack = mo.vstack([prompt_number_slider, prompt_template_model_paths], align="start")
+    return (prompt_mechanics_stack,)
+
+
+@app.cell
+def _(mo, prompt_mechanics_stack, template_selector_stack):
+    prompt_setup_stack = mo.hstack([prompt_mechanics_stack, template_selector_stack], justify="space-around")
     return (prompt_setup_stack,)
 
 
@@ -614,6 +616,35 @@ def _(mo):
     return get_pt_state, set_pt_state
 
 
+@app.function
+def get_subfolder_paths(base_path, depth=1):
+    """Lists subfolder paths up to specified depth.
+
+    Args:
+        base_path: Directory to search in
+        depth: Subdirectory levels to traverse (default: 1)
+
+    Returns:
+        Dict with folder names as keys and paths as values
+    """
+    import os
+
+    result = {}
+    if depth <= 0 or not os.path.isdir(base_path):
+        return result
+
+    for entry in os.scandir(base_path):
+        if entry.is_dir():
+            result[entry.name] = entry.path
+            if depth > 1:
+                # Recursively get subdirectories
+                subfolders = get_subfolder_paths(entry.path, depth - 1)
+                # Update result with subfolders
+                result.update(subfolders)
+
+    return result
+
+
 @app.cell
 def _(mo, prompt_number_slider, set_pt_state, templates):
     def update_state_from_templates(value):
@@ -635,18 +666,29 @@ def _(mo, prompt_number_slider, set_pt_state, templates):
         )
 
     template_selectors = [create_template_dropdowns(i) for i in range(1, prompt_number_slider.value+1)]
-    template_selector_stack = mo.hstack(template_selectors, wrap=True, justify="start")
+    template_selector_stack = mo.vstack(template_selectors, align="start")
     return (template_selector_stack,)
 
 
 @app.cell
-def _(fields_to_process, mo, prompt_number_slider):
+def _(fields_to_process):
     if fields_to_process:
         variable_names = [key for key in fields_to_process[0].keys() 
                          if key != "_marimo_row_id" and key != "upload_a_file"]
     else:
         variable_names = []
-    
+    return (variable_names,)
+
+
+@app.cell
+def _(prompt_number_slider):
+    def is_disabled(button_num):
+        return prompt_number_slider.value < button_num
+    return (is_disabled,)
+
+
+@app.cell
+def _(mo):
     def create_stats_from_variables(variable_names):
         """
         Creates mo.stat objects for each variable name in the list.
@@ -654,7 +696,7 @@ def _(fields_to_process, mo, prompt_number_slider):
         Values display the variable name itself.
         """
         stats = []
-    
+
         for i, var_name in enumerate(variable_names):
             stat = mo.stat(
                 value=f"{{{var_name}}}",
@@ -662,23 +704,18 @@ def _(fields_to_process, mo, prompt_number_slider):
                 bordered=True
             )
             stats.append(stat)
-        
-        return mo.sidebar(stats, width="525")
 
-    def is_disabled(button_num):
-        return prompt_number_slider.value < button_num
-    return create_stats_from_variables, is_disabled, variable_names
+        return mo.sidebar(stats, width="375")
+    return (create_stats_from_variables,)
 
 
 @app.cell
 def _(create_stats_from_variables, variable_names):
     if variable_names:
-        prompt_input_variables = create_stats_from_variables(variable_names)
+        prompt_input_variables_sidebar = create_stats_from_variables(variable_names)
     else:
-        prompt_input_variables = None
-    
-    prompt_input_variables
-    return
+        prompt_input_variables_sidebar = None
+    return (prompt_input_variables_sidebar,)
 
 
 @app.cell
@@ -691,12 +728,12 @@ def _(get_pt_state, mo, prompt_number_slider):
 
         prompt_column_label = mo.ui.text(label=f"**Add output column name for Prompt {num}:**", value=f"Added Column {num}")
         prompt_editor = mo.ui.code_editor(value=get_template_content(num), language="python", min_height=300, theme="dark")
-    
+
         return [
             prompt_column_label,
             prompt_editor
         ]
-    
+
     prompts_raw = [create_prompt(i) for i in range(1, prompt_number_slider.value+1)]
     return (prompts_raw,)
 
@@ -842,9 +879,28 @@ def _(
 
 
 @app.cell
-def _(table_dataframe):
-    _ = table_dataframe
+def _(mo, results_stack):
+    ui_accordion_section_7 = mo.accordion(
+        {"Section 7: **Run and View Results**": results_stack}
+    ) 
+    ui_accordion_section_7
+    return
+
+
+@app.cell
+def _(show_variable_sidebar):
+    show_variable_sidebar.right()
+    return
 
+
+@app.cell
+def _(prompt_input_variables_sidebar, show_variable_sidebar):
+    show_variable_sidebar.value is not False and prompt_input_variables_sidebar
+    return
+
+
+@app.cell
+def _(table_dataframe):
     if not table_dataframe.empty:
         results_table = table_dataframe.copy()
     else:
@@ -868,6 +924,7 @@ def _(
         results_table_view = mo.ui.table(results_table)
     return (results_table_view,)
 
+
 @app.cell
 def _(mo, prompts_raw):
     prompts =  [mo.vstack([prompt[0], prompt[1]]) for prompt in prompts_raw]