Update main_app.py

main_app.py

@@ -498,7 +498,7 @@ def _(
     package_meta,
     ss_asset_response,
     yaml_template,
-    base_software_spec_id_selector
+    base_software_spec_id_selector,
 ):
     packages_section = mo.md(
         f"""
@@ -665,32 +665,29 @@ def _():
 
 
 @app.cell
-def _(template,base_software_spec_id_selector):
+def _(template, base_software_spec_id_selector):
     dependencies = template["dependencies"]
     dependencies_string = ", ".join(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)}
+        added_yaml_template = f"""{"\n".join(f"{dep}" for dep in dependencies)}
     """
         yaml_templates = {
-            "empty": """dependencies:
-      - <library_name>
-      - ...
+            "empty": f"""<library_name>\n\n<library_name>
     """
         }
     else:
         # Format for Python 3.11 - with pip:
         added_yaml_template = f"""dependencies:
-          - pip
-          - pip:
+        - pip
+        - pip:
     {"\n".join(f"        - {dep}" for dep in dependencies)}
     """
         yaml_templates = {
             "empty": """dependencies:
-      - pip
-      - pip:
+    - pip
+    - pip:
         - <library_name>
         - ...
     """
@@ -728,11 +725,17 @@ def _(tempfile):
 
 @app.cell
 def _(mo, yaml_template, dependency_pack_name):
-    pkg_types = {"Conda Yaml": "conda_yml", "Custom user library": "custom_library"}
+    pkg_types = {
+        "requirements.txt": "requirements_txt",
+        "Conda Yaml": "conda_yml",
+        "Custom user library": "custom_library",
+    }
 
     package_meta = (
         mo.md(
-            """**Create your Conda YAML by editing the template:**
+            """**Create your Package Extension by editing the template:**
+
+        {package_type}
 
         {yml_editor}
 
@@ -776,7 +779,7 @@ def _(mo, yaml_template, dependency_pack_name):
                 max_length=int(256),
             ),
             package_type=mo.ui.dropdown(
-                pkg_types, label="Select your package type:", value="Conda Yaml"
+                pkg_types, label="Select your package type:", value="requirements.txt"
             ),
         )
         .form(show_clear_button=True, bordered=False)
@@ -785,7 +788,7 @@ def _(mo, yaml_template, dependency_pack_name):
 
 
 @app.cell
-def _(mo,dependencies_string):
+def _(mo, dependencies_string):
     check_packages = (
         mo.md(
             """
@@ -863,7 +866,7 @@ def _(base_spec_id_names, mo):
 
 
 @app.cell
-def _(client,base_software_spec_id_selector,base_spec_ids):
+def _(client, base_software_spec_id_selector, base_spec_ids):
     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)
@@ -873,6 +876,19 @@ def _(client,base_software_spec_id_selector,base_spec_ids):
     return base_software_spec, base_sw_spec_id
 
 
+@app.cell
+def _(
+    client,
+):
+    if client:
+        base_hw_spec_id = "b128f957-581d-46d0-95b6-8af5cd5be580"
+        base_hardware_spec = client.hardware_specifications.get_details(base_hw_spec_id)
+    else:
+        base_hw_spec_id = None
+        base_hardware_spec = None
+    return base_hardware_spec, base_hw_spec_id
+
+
 @app.cell
 def _(client, create_yaml_tempfile, package_meta, uuid):
     if package_meta.value is not None and client is not None:
@@ -1309,9 +1325,11 @@ def _(mo):
 
 
 @app.cell
-def _(selection_table, set_selected_sw_spec):
+def _(selection_table, set_selected_sw_spec, base_sw_spec_id):
     if selection_table.value is not None:
         set_selected_sw_spec(selection_table.value["ID"].iloc[0])
+    else:
+        set_selected_sw_spec(base_sw_spec_id)
     return
 
 
@@ -1322,9 +1340,11 @@ def _(mo):
 
 
 @app.cell
-def _(hw_selection_table, set_selected_hw_spec):
+def _(hw_selection_table, set_selected_hw_spec, base_hw_spec_id):
     if hw_selection_table.value is not None:
         set_selected_hw_spec(hw_selection_table.value["ID"].iloc[0])
+    else:
+        set_selected_hw_spec(base_hw_spec_id)
     return