Update app.py

app.py

@@ -600,7 +600,6 @@ def fn_analyze_task(project_context, task_description):
 
 # In[13]:
 
-
 # Recursive Task Executor
 def fn_process_task(project_desc_table, task_description, bucket_name='curify'):
     
@@ -614,13 +613,19 @@ def fn_process_task(project_desc_table, task_description, bucket_name='curify'):
 
         def traverse(node, previous_output=None):
             if not node:  # If the node is None or invalid
-                return {}, pd.DataFrame(), {}
+                return  # Exit if the node is invalid
 
             # Check if there is a condition to evaluate
             if "check" in node:
-                value = execution_results[node["check"]] # Evaluate the check condition
-                traverse(node.get(value, node.get("default")), previous_output)
-
+                # Safely attempt to retrieve the value from execution_results
+                if node["check"] in execution_results:
+                    value = execution_results[node["check"]]  # Evaluate the check condition
+                    traverse(node.get(value, node.get("default")), previous_output)
+                else:
+                    # Log an error and exit, but keep partial results
+                    logger.error(f"Key '{node['check']}' not found in execution_results.")
+                    return
+        
             # If the node contains an action
             elif "action" in node:
                 action_name = node["action"]
@@ -629,15 +634,21 @@ def fn_process_task(project_desc_table, task_description, bucket_name='curify'):
                 if input_key in execution_results.keys():
                     inputs = {input_key: execution_results[input_key]}
                 else:
+                    # Log an error and exit, but keep partial results
                     logger.error(f"Workflow action {action_name} input key {input_key} not in execution_results.")
-                    return {}, pd.DataFrame(), {}
+                    return
 
                 logger.debug(f"Executing: {action_name} with inputs: {inputs}")
                 
                 # Execute the action function
                 action_func = action_map.get(action_name, unsupported_task)
-                output = action_func(**inputs)
-                
+                try:
+                    output = action_func(**inputs)
+                except Exception as e:
+                    # Handle action function failure
+                    logger.error(f"Error executing action '{action_name}': {e}")
+                    return
+
                 # Store execution results or append to previous outputs
                 execution_status.append({"action": action_name, "output": output})
 
@@ -656,15 +667,15 @@ def fn_process_task(project_desc_table, task_description, bucket_name='curify'):
         try:
             traverse(TASK_WORKFLOW_TREE["start"])
             execution_results['doc_url'] = generate_document(task_description, execution_results)
-            return task_analysis, pd.DataFrame(execution_status), execution_results
         except Exception as e:
             logger.error(f"Traverse Error: {e}")
-            return task_analysis, pd.DataFrame(), {}        
+        finally:
+            # Always return partial results, even if an error occurs
+            return task_analysis, pd.DataFrame(execution_status), execution_results
     else:
         logger.error("Empty task analysis.")
         return {}, pd.DataFrame(), {}
 
-
 # In[14]: