Update app.py

app.py

@@ -39,12 +39,22 @@ if st.button("Run Workflow"):
 
     # Export workflow configuration as YAML or JSON
     export_format = st.radio("Export Format", ["YAML", "JSON"])
-    if st.button("Export Workflow"):
-        if export_format == "YAML":
-            with open("workflow_config.yaml", "w") as yaml_file:
-                yaml.dump(workflow_config, yaml_file)
-            st.success("Workflow configuration exported as YAML.")
-        elif export_format == "JSON":
-            with open("workflow_config.json", "w") as json_file:
-                json.dump(workflow_config, json_file, indent=4)
-            st.success("Workflow configuration exported as JSON.")
+
+    with open("workflow_config.yaml", "w") as yaml_file:
+        yaml.dump(workflow_config, yaml_file)
+        st.download_button(
+        label="Export Config",
+        data=yaml_file,
+        file_name='Workflow_Config.yaml',
+        mime='yaml',
+        )
+    
+    # if st.download_button("Export Workflow"):
+    #     if export_format == "YAML":
+    #         with open("workflow_config.yaml", "w") as yaml_file:
+    #             yaml.dump(workflow_config, yaml_file)
+    #         st.success("Workflow configuration exported as YAML.")
+    #     elif export_format == "JSON":
+    #         with open("workflow_config.json", "w") as json_file:
+    #             json.dump(workflow_config, json_file, indent=4)
+    #         st.success("Workflow configuration exported as JSON.")