YchKhan commited on
Commit
13ff5b8
1 Parent(s): 3252717

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +7 -1
excel_chat.py CHANGED
@@ -110,7 +110,13 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
110
  df.at[index, dest_col] = llm_answer
111
 
112
  df.to_excel(file_name, index=False)
113
- return file_name, df.head(5), new_prompts, new_keywords, new_user, conf_file_path
 
 
 
 
 
 
114
 
115
 
116
  def get_columns(file,progress=gr.Progress()):
 
110
  df.at[index, dest_col] = llm_answer
111
 
112
  df.to_excel(file_name, index=False)
113
+
114
+ zip_file_path = 'config_file.zip'
115
+
116
+ with zipfile.ZipFile(zip_file_path, 'w') as zipf:
117
+ zipf.write(conf_file_path, os.path.basename(conf_file_path))
118
+
119
+ return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path
120
 
121
 
122
  def get_columns(file,progress=gr.Progress()):