Update app.py

app.py

@@ -8,7 +8,6 @@ import os
 import re
 
 import nltk
-import yaml
 
 import pandas as pd
 import streamlit as st
@@ -180,65 +179,6 @@ 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 == "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":
-                upsert = component.pop("upsert")
-
-                data["embeddings"] = component
-                data["writable"] = True
-
-                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.
@@ -339,19 +279,12 @@ class Application:
         st.sidebar.markdown("---")
 
         with st.sidebar:
-            col1, col2 = st.columns(2)
-
             # Build or re-build workflow when build button clicked
-            build = col1.button("Build", help="Build the workflow and run within this application")
+            build = st.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)