Update main_app.py

main_app.py

@@ -229,7 +229,7 @@ def _():
 def _(cloudant, cloudant_db):
     variant_selectors = cloudant_retrieve_documents(
         client=cloudant,
-        db_name=cloudant_db.value,
+        db_name=cloudant_db,
         selectors={"filename": {"$exists": True}},
         fields=["category"],
     )
@@ -246,6 +246,21 @@ def _(mo):
     template_variant = mo.ui.dropdown(template_variants, label="Code Template:", value="Base")
     return (template_variant,)
 
+@app.cell
+def _(cloudant, cloudant_db, template_variant, refresh):
+    refresh.value
+    retrieved_template = (
+        cloudant_retrieve_documents(
+            client=cloudant,
+            db_name=cloudant_db,
+            selectors={"category": {"$eq": template_variant.value}},
+            fields=json.loads(os.getenv("FIELDS_TO_RETRIEVE")) or [],
+        )
+        if template_variant.value
+        else None
+    )
+    template = retrieved_template["docs"][0] if retrieved_template else None
+    return (template,)
 
 @app.cell
 def _(client_callout_kind, client_instantiation_form, mo, template_variant):
@@ -511,34 +526,6 @@ def _(mo, purging_section):
 
 @app.cell
 def _(mo, template_variant):
-    # Template for WatsonX.ai deployable function
-    if template_variant.value == "Stream Files to IBM COS [Example]":
-        with open("stream_files_to_cos.py", "r") as file:
-            template = file.read()
-    else:
-        template = '''def your_function_name():
-
-        import subprocess
-        subprocess.check_output('pip install gensim', shell=True)
-        import gensim
-
-        def score(input_data):
-            message_from_input_payload = payload.get("input_data")[0].get("values")[0][0]
-            response_message = "Received message - {0}".format(message_from_input_payload)
-
-            # Score using the pre-defined model
-            score_response = {
-                'predictions': [{'fields': ['Response_message_field', 'installed_lib_version'],
-                                'values': [[response_message, gensim.__version__]]
-                                }]
-            }
-            return score_response
-
-        return score
-
-    score = your_function_name()
-            '''
-
     function_editor = (
         mo.md('''
         #### **Create your function by editing the template:**
@@ -547,7 +534,7 @@ def _(mo, template_variant):
 
         ''')
         .batch(
-            editor = mo.ui.code_editor(value=template, language="python", min_height=200, theme="dark")
+            editor = mo.ui.code_editor(value=template["function_code"], language="python", min_height=200, theme="dark")
         )
         .form(show_clear_button=True, bordered=False)
         )
@@ -593,54 +580,31 @@ def _(ast, function_editor, mo, os):
 
 @app.cell
 def _():
+    dependencies = template["dependencies"]
+    yaml_template = f"""dependencies:
+          - pip
+          - pip:
+    {'\n'.join(f'        - {dep}' for dep in dependencies)}
+    """
+    
     yaml_templates = {
         "empty": """dependencies:
       - pip
       - pip:
         - <library_name>
         - ...
-    """,
-        "llama_index_and_scikit": """dependencies:
-      - pip
-      - pip:
-        - scikit-learn==1.6.1
-        - scikit-llm==1.4.1
-        - plotly==6.0.1
-        - altair==5.5.0
-        - PyMuPDF==1.25.1
-        - llama-index==0.12.24
-        - llama-index-readers-file==0.4.6
-        - llama-index-readers-web==0.3.8
-    """,
-        "data_product_hub": """dependencies:
-      - pip
-      - pip:
-        - PyMuPDF==1.25.1
-        - llama-index==0.12.24
-        - llama-index-readers-file==0.4.6
-        - llama-index-readers-web==0.3.8
-        - plotly==6.0.1
-        - altair==5.5.0
-        - dask==2025.2.0
-        - dph-services==0.4.0
-    """,
-        "watsonx_data": """dependencies:
-      - pip
-      - pip:
-        - prestodb
-        - PyMuPDF==1.25.1
-        - llama-index==0.12.24
-        - llama-index-readers-file==0.4.6
-        - llama-index-readers-web==0.3.8
-        - ibm-watsonxdata==0.4.0
     """
     }
+
+    dependency_pack_name = template["function_name"] + "_dependencies"
+    # After the yaml_template definition, add:
+    yaml_templates[dependency_pack_name] = yaml_template
     return (yaml_templates,)
 
 
 @app.cell
 def _(mo, yaml_templates):
-    yaml_template = mo.ui.dropdown(yaml_templates, searchable=True, label="**Select a template:**", value="empty")
+    yaml_template = mo.ui.dropdown(yaml_templates, searchable=True, label="**Select a template:**", value=dependency_pack_name)
     return (yaml_template,)
 
 
@@ -677,10 +641,10 @@ def _(mo, yaml_template):
             package_name = mo.ui.text(placeholder="Python Package for...", 
                                       label="Package Extension Name:", 
                                       kind="text", 
-                                      value="Custom Python Package"),
+                                      value=f"Custom Python Package: {dependency_pack_name}"),
             software_spec_name = mo.ui.text(placeholder="Software Spec Name", 
                                             label="Custom Software Spec Name:", 
-                                            kind="text", value="Extended Python Function Software Spec"),
+                                            kind="text", value=f"Extended Python Function Software Spec - {dependency_pack_name}"),
             package_description = mo.ui.text_area(placeholder="Write a description for your package.",
                                                   label="Package Description:", 
                                                   value=" "),
@@ -1152,7 +1116,6 @@ def _(hw_selection_table, set_selected_hw_spec):
 def _(mo):
     input_schema_checkbox = mo.ui.checkbox(label="Add input schema (optional)")
     output_schema_checkbox = mo.ui.checkbox(label="Add output schema (optional)")
-    # sample_input_checkbox = mo.ui.checkbox(label="Add sample input example (optional)")
     return input_schema_checkbox, output_schema_checkbox
 
 
@@ -1171,15 +1134,13 @@ def _(
         if function_name is not None:
             fnc_nm = function_name
         else:
-            if template_variant.value == "Stream Files to IBM COS [Example]":
-                fnc_nm = "stream_file_to_cos"
-            else:
-                fnc_nm = "your_function_name"
+            fnc_nm = "custom_pythoN_functin"
 
         uploaded_function_name = mo.ui.text(placeholder="<Must be the same as the name in editor>", label="Function Name:", kind="text", value=f"{fnc_nm}", full_width=False)
         tags_editor = mo.ui.array(
-            [mo.ui.text(placeholder="Metadata Tags..."), mo.ui.text(), mo.ui.text()],
-            label="Optional Metadata Tags"
+           [mo.ui.text(placeholder="Metadata Tags...", value=use_case) for use_case in template["use_cases"]] +
+           [mo.ui.text(placeholder="Metadata Tags...") for _ in range(max(0, 4 - len(template["use_cases"])))],
+           label="Optional Metadata Tags"
         )
         software_spec = get_selected_sw_spec()
 
@@ -1190,8 +1151,17 @@ def _(
             rows=5,
             full_width=True
         )
+        
+    return (
+        description_input,
+        software_spec,
+        tags_editor,
+        uploaded_function_name,
+    )
 
-
+@app.cell
+def _(mo, selection_table, description_input, uploaded_function_name, tags_editor):
+    if selection_table.value is not None:
         func_metadata=mo.hstack([
             description_input,
             mo.hstack([
@@ -1202,42 +1172,47 @@ def _(
             widths=[0.6,0.4],
             gap=2.75
         )
+    return func_metadata
 
+@app.cell
+def _(mo, selection_table, input_schema_checkbox, output_schema_checkbox):
+    if selection_table.value is not None:
         schema_metadata=mo.hstack([
             input_schema_checkbox,
             output_schema_checkbox,
-            # sample_input_checkbox
         ],
             justify="center", gap=1, align="center", wrap=True
         )
+    return schema_metadata
 
-    fm = mo.vstack([
-        func_metadata,
-        ],
-        align="center",
-        gap=2
-    )
-    sc_m = mo.vstack([
-        schema_metadata,
-        mo.md("**Make sure to select the checkbox options before filling in descriptions and tags or they will reset.**")
-        ],
-        align="center",
-        gap=2
-    )
-    return (
-        description_input,
-        fm,
-        sc_m,
-        software_spec,
-        tags_editor,
-        uploaded_function_name,
-    )
+@app.cell
+def _(mo, selection_table, func_metadata):
+    if selection_table.value is not None:
+        fm = mo.vstack([
+            func_metadata,
+            ],
+            align="center",
+            gap=2
+        )
+    return fm
+
+@app.cell
+def _(mo, selection_table, schema_metadata):
+    if selection_table.value is not None:
+        sc_m = mo.vstack([
+            schema_metadata,
+            mo.md("**Make sure to select the checkbox options before filling in descriptions and tags or they will reset.**")
+            ],
+            align="center",
+            gap=2
+        )
+    return sc_m
 
 
 @app.cell
 def _(json, mo, template_variant):
-    if template_variant.value == "Stream Files to IBM COS [Example]":
-        from cos_stream_schema_examples import input_schema, output_schema
+    if template["input_schema"]:
+        input_schema = template["input_schema"]
     else:
         input_schema = [
             {
@@ -1259,7 +1234,10 @@ def _(json, mo, template_variant):
                 ]
             }
         ]
-
+        
+    if template["output_schema"]:
+        output_schema = template["output_schema"]
+    else:
         output_schema = [
             {
                 'id': '1',
@@ -1275,21 +1253,30 @@ def _(json, mo, template_variant):
             }
         ]
 
+    return input_schema, output_schema
+
+    input_schema_editor = mo.ui.code_editor(value=json.dumps(input_schema, indent=4), language="python", min_height=100,theme="dark")
+    output_schema_editor = mo.ui.code_editor(value=json.dumps(output_schema, indent=4), language="python", min_height=100,theme="dark")
 
 
+    # schema_editors
+    return input_schema_editor, output_schema_editor
+
+@app.cell
+def _(mo, input_schema, output_schema):
     input_schema_editor = mo.ui.code_editor(value=json.dumps(input_schema, indent=4), language="python", min_height=100,theme="dark")
     output_schema_editor = mo.ui.code_editor(value=json.dumps(output_schema, indent=4), language="python", min_height=100,theme="dark")
+    return input_schema_editor, output_schema_editor
 
+@app.cell
+def _(mo, input_schema_editor, output_schema_editor):
     schema_editors = mo.accordion(
         {
             """**Input Schema Metadata Editor**""": input_schema_editor,
             """**Output Schema Metadata Editor**""": output_schema_editor,
         }, multiple=True
     )
-
-    # schema_editors
-    return input_schema_editor, output_schema_editor, schema_editors
-
+    return schema_editors
 
 @app.cell
 def _(