Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 import re
 
 import nltk
+import yaml
 
 import pandas as pd
 import streamlit as st
@@ -171,6 +172,73 @@ class Application:
 
         self.workflow = Workflow(tasks)
 
+    def yaml(self, components):
+        """
+        Builds a yaml string for components.
+
+        Args:
+            components: list of components to export to YAML
+
+        Returns:
+            YAML string
+        """
+
+        # pylint: disable=W0108
+        data = {}
+        tasks = []
+        name = None
+
+        for component in components:
+            component = dict(component)
+            name = wtype = component.pop("type")
+
+            if wtype == "summary":
+                data["summary"] = {"path": component.pop("path")}
+                tasks.append({"action": "summary"})
+
+            elif wtype == "segment":
+                data["segmentation"] = component
+                tasks.append({"action": "segmentation"})
+
+            elif wtype == "service":
+                config = dict(**component)
+                config["task"] = "service"
+                tasks.append(config)
+
+            elif wtype == "tabular":
+                data["tabular"] = component
+                tasks.append({"action": "tabular"})
+
+            elif wtype == "textract":
+                data["textractor"] = component
+                tasks.append({"action": "textractor", "task": "url"})
+
+            elif wtype == "transcribe":
+                data["transcription"] = {"path": component.pop("path")}
+                tasks.append({"action": "transcription", "task": "url"})
+
+            elif wtype == "translate":
+                data["translation"] = {}
+                tasks.append({"action": "translation", "args": list(component.values())})
+
+            elif wtype == "embeddings":
+                index = component.pop("index")
+                upsert = component.pop("upsert")
+
+                data["embeddings"] = component
+                data["writable"] = True
+
+                if index:
+                    data["path"] = index
+
+                name = "index"
+                tasks.append({"action": "upsert" if upsert else "index"})
+
+        # Add in workflow
+        data["workflow"] = {name: {"tasks": tasks}}
+
+        return (name, yaml.dump(data))
+
     def find(self, key):
         """
         Lookup record from cached data by uid key.
@@ -271,12 +339,19 @@ class Application:
         st.sidebar.markdown("---")
 
         with st.sidebar:
+            col1, col2 = st.columns(2)
+        
             # Build or re-build workflow when build button clicked
-            build = st.button("Build", help="Build the workflow and run within this application")
+            build = col1.button("Build", help="Build the workflow and run within this application")
             if build:
                 with st.spinner("Building workflow...."):
                     self.build(components)
 
+            # Generate API configuration
+            _, config = self.yaml(components)
+
+            col2.download_button("Export", config, file_name="workflow.yml", mime="text/yaml", help="Export the API workflow as YAML")
+
         with st.expander("Data", expanded=not self.data):
             data = st.text_area("Input", height=10)