Update app.py

app.py

@@ -24,18 +24,19 @@ class Process:
 
     @staticmethod
     @st.cache(ttl=30 * 60, max_entries=3, allow_output_mutation=True, show_spinner=False)
-    def get(components):
+    def get(components, data):
         """
         Lookup or creates a new workflow process instance.
 
         Args:
             components: input components
+            data: initial data, only passed when indexing
 
         Returns:
             Process
         """
 
-        process = Process()
+        process = Process(data)
 
         # Build workflow
         with st.spinner("Building workflow...."):
@@ -43,9 +44,12 @@ class Process:
 
         return process
 
-    def __init__(self):
+    def __init__(self, data):
         """
         Creates a new Process.
+
+        Args:
+            data: initial data, only passed when indexing
         """
 
         # Component options
@@ -60,7 +64,7 @@ class Process:
         # Embeddings index params
         self.embeddings = None
         self.documents = None
-        self.data = None
+        self.data = data
 
     def build(self, components):
         """
@@ -615,12 +619,13 @@ class Application:
         # Wrap data as list for workflow processing
         return [data]
 
-    def query(self, workflow):
+    def query(self, workflow, index):
         """
         Gets input query.
 
         Args:
             workflow: workflow configuration
+            index: True if this is an indexing workflow
 
         Returns:
             input query
@@ -629,35 +634,37 @@ class Application:
         default = self.appsetting(workflow, "query")
         default = default if default else ""
 
-        # Set query and limit
-        query = st.text_input("Query", value=default)
+        # Get query if this is an indexing workflow
+        query = st.text_input("Query", value=default) if index else None
 
         # Save query state
         st.session_state["query"] = query
 
         return query
 
-    def process(self, workflow, components):
+    def process(self, workflow, components, index):
         """
         Processes the current application action.
 
         Args:
             workflow: workflow configuration
             components: workflow components
+            index: True if this is an indexing workflow
         """
 
+        # Get input data and initialize query
+        data = self.data(workflow)
+        query = self.query(workflow, index)
+
         # Get workflow process
-        process = Process.get(components)
+        process = Process.get(components, data if index else None)
 
         # Run workflow process
-        process.run(self.data(workflow))
+        process.run(data)
 
         # Run search
-        if process.embeddings:
-            process.search(self.query(workflow))
-        else:
-            # Clear session query
-            st.session_state["query"] = None
+        if index:
+            process.search(query)
 
     def run(self):
         """
@@ -680,7 +687,7 @@ class Application:
 
         if selected:
             # Process current action
-            self.process(workflow, components)
+            self.process(workflow, components, "embeddings" in selected)
 
             with st.sidebar:
                 # Generate export button after workflow is complete