Update excel_chat.py

excel_chat.py

@@ -94,7 +94,7 @@ def filter_df(df, column_name, keywords):
     return filtered_df
 
 def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col, keywords, client, user):
-    update_json(user, prompt, keywords)
+    new_prompts, new_keywords = update_json(user, prompt, keywords)
     print(f'xlsxfile = {excel_file}')
     df = pd.read_excel(excel_file)
     df[dest_col] = ""
@@ -116,7 +116,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
             df.at[index, dest_col] = llm_answer
 
     df.to_excel(file_name, index=False)
-    return file_name, df.head(5)
+    return file_name, df.head(5), new_prompts, new_keywords
 
 
 def get_columns(file,progress=gr.Progress()):