Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 import json
 import uuid
+import requests
 
 def convert_inputs(user_input, var):
     if var == "__random__":
@@ -39,11 +40,60 @@ def process_inputs(user_input_json, session_id_json, project_id, chat_url, updat
         "update_variables": update_vars,
         "output_variables": output_vars,
         "answer": [[] * len(x) if isinstance(x, list) else [] for x in user_input]
-    })
+    })  
 
     return df
 
+
+def run_chat(chat_url, project_id, session_id, user_input, update_variables, output_variables, token):
+     req = requests.post(
+        chat_url,
+        json={
+            "project_id": project_id,
+            "session_id": session_id,
+            "user_input": user_input,
+            "update_variables": update_variables,
+            "output_variables": output_variables
+        },
+        headers={"Authorization" : f"Bearer {token}"}
+    )
+
+    print("[REQ CONTENT]", req.content)
+
+    out = req.json()["data"]["results"]
+
+    print("[OUT]",out)
+
+    return out.encode('utf-8').decode('unicode_escape')
+
+
 def run_process(df):
+    for index, row in df.iterrows():
+        if isinstance(row["user_input"], list):
+            answer = []
+            for user_input in row["user_input"]:
+                answer.append(
+                    run_chat(
+                        row["chat_url"],
+                        row["project_id"],#: project_id,
+                        row["session_id"],#: session_id,
+                        user_input,#: user_input,
+                        row["update_variables"],#: update_variables,
+                        row["output_variables"],
+                        row["token"]
+                    )
+                )
+        else:
+            answer = run_chat(
+                row["chat_url"],
+                row["project_id"],#: project_id,
+                row["session_id"],#: session_id,
+                row["user_input"],#: user_input,
+                row["update_variables"],#: update_variables,
+                row["output_variables"],
+                row["token"]
+            )
+        df.loc[index, "answer"] = answer
     csv_path = "output.csv"
     df.to_csv(csv_path, index=False)
     return csv_path