Update app.py

app.py

@@ -18,6 +18,20 @@ def convert_inputs(user_input, var):
     
     return var
 
+
+def convert_output_vars(user_input, output_vars):
+    if isinstance(var, str):
+        var = [[var]]
+    elif not isinstance(var[0], list):
+        var = [var]
+        
+    var += var*(len(user_input)//len(var) + 1)
+
+    diff = len(var) - len(user_input) 
+    var = var[:-diff]
+    
+    return var
+
 def process_inputs(user_input_json, session_id_json, project_id, chat_url, update_vars_json, output_vars_json, token):
     # Convert JSON strings into Python lists
     user_input = json.loads(user_input_json)