Update app.py

app.py

@@ -52,7 +52,7 @@ def process_inputs(user_input_json, session_id_json, project_id, chat_url, updat
             unique_output.add(y)
 
     dict_output = {
-        uni: [[] * len(x) if isinstance(x, list) else [] for x in user_input] for uni in unique_output 
+        uni: [[] for _ in user_input] for uni in unique_output 
     }
 
     # --- Your function logic here ---
@@ -87,11 +87,12 @@ def run_chat(chat_url, project_id, session_id, user_input, update_variables, out
 
     print("[REQ CONTENT]", req.content)
 
-    out = req.json()["data"]["results"]
+    out = req.json()["data"]
 
-    print("[OUT]",out)
+    print("[OUT]",out.get("results"))
 
-    return out.encode('utf-8').decode('unicode_escape')
+    return out
+    #return {out_var:out[out_var] for out_var in output_variables}
 
 
 def run_process(df):
@@ -120,7 +121,15 @@ def run_process(df):
                 row["output_variables"],
                 row["token"]
             )
-        df.loc[index, "answer"] = answer
+
+        if isinstance(answer, list):
+            for ans in answer:
+                for key,val in ans.items():
+                    df.loc[index, key].append(val)
+        else:
+            for key,ans in answer.items():
+                df.loc[index, key] = ans 
+            
     csv_path = "output.csv"
     df.to_csv(csv_path, index=False)
     return csv_path, df