Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-
+import yaml  # Import the pyyaml package
 from backend_services import ServiceOne, ServiceTwo
 
 # List of available services
@@ -22,11 +22,29 @@ for service_name in selected_services:
 # User-defined workflow execution
 if st.button("Run Workflow"):
     input_data = ...  # Define your input data
+    workflow_config = []  # List to store the workflow configuration
     for service_name in selected_services:
         service_instance = services[service_name]
         service_instance.parameters = parameters[service_name]
         output_data = service_instance.execute(input_data)
         input_data = output_data
+        # Store the service name and parameters in the workflow configuration
+        workflow_config.append({
+            "service": service_name,
+            "parameters": service_instance.parameters
+        })
 
     # Display the final output
     st.success("Workflow executed successfully. Output: {}".format(input_data))
+
+    # 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.")