Update main_app.py

main_app.py

@@ -502,6 +502,8 @@ def _(
 
         3. If it isn't, you can create a custom software spec that adds a package_extension add-on to it. Use the template selector to see some examples, but when you are ready add your packages in the code editor after the "pip:" section.
 
+        {base_software_spec_id_selector}
+
         {yaml_template}
 
         4. Give your package extension and software spec names and descriptions and submit.
@@ -637,25 +639,46 @@ def _(ast, template, function_editor, mo, os):
             mo.md(f"Syntax error in function code: {str(e)}")
     return (function_name,)
 
+@app.cell
+def _():
+    base_spec_ids = {
+        "runtime-25.1-py3.12": "f47ae1c3-198e-5718-b59d-2ea471561e9e",
+        "runtime-24.1-py3.11": "45f12dfe-aa78-5b8d-9f38-0ee223c47309",
+    }
+    base_spec_id_names = list(base_spec_ids.keys())
+    return base_spec_id_names, base_spec_ids
 
 @app.cell
 def _(template):
     dependencies = template["dependencies"]
     dependencies_string = ', '.join(dependencies)
-    added_yaml_template = f"""dependencies:
+
+    if base_software_spec_id_selector.value == "runtime-25.1-py3.12":
+        # Format for Python 3.12 - simple list without pip:
+        added_yaml_template = f"""dependencies:
+    {'\n'.join(f'  - {dep}' for dep in dependencies)}
+    """
+        yaml_templates = {
+            "empty": """dependencies:
+      - <library_name>
+      - ...
+    """
+        }
+    else:
+        # Format for Python 3.11 - with pip:
+        added_yaml_template = f"""dependencies:
           - pip
           - pip:
     {'\n'.join(f'        - {dep}' for dep in dependencies)}
     """
-
-    yaml_templates = {
-        "empty": """dependencies:
+        yaml_templates = {
+            "empty": """dependencies:
       - pip
       - pip:
         - <library_name>
         - ...
     """
-    }
+        }
 
     dependency_pack_name = template["function_name"] + "_dependencies"
     # After the yaml_template definition, add:
@@ -811,11 +834,20 @@ def _(check_packages, mo, package_df):
     package_analysis_stack = mo.vstack([check_packages, package_df], justify="center")
     return (package_analysis_stack,)
 
+@app.cell
+def _(base_spec_id_names, mo):
+    base_software_spec_id_selector = mo.ui.dropdown(
+        options=base_spec_id_names,
+        label="**Select your Base Software Specification** *(defaults to Python 3.12)*:",
+        allow_select_none=False,
+        value=base_spec_id_names[0],
+    )
+    return (base_software_spec_id_selector,)
 
 @app.cell
 def _(client):
-    if client is not None:
-        base_sw_spec_id = "45f12dfe-aa78-5b8d-9f38-0ee223c47309"
+    if client is not None and base_software_spec_id_selector.value:
+        base_sw_spec_id = base_spec_ids[base_software_spec_id_selector.value]
         base_software_spec = client.software_specifications.get_details(base_sw_spec_id)
     else:
         base_sw_spec_id = None
@@ -1145,7 +1177,7 @@ def _(client, mo, pd, specs_df):
             (specs_df["STATE"] == "supported")
             & (
                 specs_df["NAME"].isin(
-                    ["runtime-24.1-py3.11", "runtime-24.1-py3.11-cuda"]
+                    ["runtime-25.1-py3.12","runtime-25.1-py3.12-cuda","runtime-24.1-py3.11", "runtime-24.1-py3.11-cuda"]
                 )
             )
         ]
@@ -1157,13 +1189,17 @@ def _(client, mo, pd, specs_df):
 
         # Create a mapping dictionary for framework names based on software specifications
         framework_mapping = {
-            "tensorflow_rt24.1-py3.11": "TensorFlow",
-            "pytorch-onnx_rt24.1-py3.11": "PyTorch",
+            "tensorflow_rt24.1-py3.11": "TensorFlow 3.11",
+            "tensorflow_rt25.1-py3.12": "TensorFlow 3.12",
+            "pytorch-onnx_rt24.1-py3.11": "PyTorch 3.11",
+            "pytorch-onnx_rt25.1-py3.12": "PyTorch 3.12",
             "onnxruntime_opset_19": "ONNX or ONNXRuntime",
-            "runtime-24.1-py3.11": "AI Services/Python Functions/Python Scripts",
             "autoai-ts_rt24.1-py3.11": "AutoAI",
             "autoai-kb_rt24.1-py3.11": "AutoAI",
-            "runtime-24.1-py3.11-cuda": "CUDA-enabled (GPU) Python Runtime",
+            "runtime-25.1-py3.12": "AI Services/Python Functions/Python Scripts Python 3.12 Runtime",
+            "runtime-24.1-py3.11": "AI Services/Python Functions/Python Scripts Python 3.11 Runtime",
+            "runtime-25.1-py3.12-cuda": "CUDA-enabled (GPU) Python 3.12 Runtime",
+            "runtime-24.1-py3.11-cuda": "CUDA-enabled (GPU) Python 3.11 Runtime",
             "runtime-24.1-r4.3": "R Runtime 4.3",
             "spark-mllib_3.4": "Apache Spark 3.4",
             "autoai-rag_rt24.1-py3.11": "AutoAI RAG",
@@ -1171,15 +1207,19 @@ def _(client, mo, pd, specs_df):
 
         # Define the preferred order for items to appear at the top
         preferred_order = [
+            "runtime-25.1-py3.12",
+            "runtime-25.1-py3.12-cuda",
+            "runtime-25.1-py3.12-xc",
             "runtime-24.1-py3.11",
             "runtime-24.1-py3.11-cuda",
+            "runtime-24.1-py3.11-xc",
             "runtime-24.1-r4.3",
-            "ai-service-v5-software-specification",
             "autoai-rag_rt24.1-py3.11",
             "autoai-ts_rt24.1-py3.11",
             "autoai-kb_rt24.1-py3.11",
             "tensorflow_rt24.1-py3.11",
             "pytorch-onnx_rt24.1-py3.11",
+            "pytorch-onnx_rt25.1-py3.12",
             "onnxruntime_opset_19",
             "spark-mllib_3.4",
         ]
@@ -1214,7 +1254,7 @@ def _(client, mo, pd, specs_df):
         selection_table = mo.ui.table(
             supported_specs,
             selection="single",  # Only allow selecting one row
-            label="#### **Select a supported software_spec runtime for your function asset** (For Python Functions select - *'runtime-24.1-py3.11'* ):",
+            label="#### **Select a supported software_spec runtime for your function asset** (For Python Functions select - *'runtime-25.1-py3.12'* ):",
             initial_selection=[
                 0
             ],  # Now selecting the first row, which should be runtime-24.1-py3.11