Update excel_chat.py

excel_chat.py

@@ -219,7 +219,6 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
 
     filtred_df = filter_df(df, search_col, keywords)
 
-    checkpoints_files = []
     cpt = 1
     checkpoint = 50
     for index, row in filtred_df.iterrows():
@@ -236,8 +235,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
             df.at[index, dest_col] = llm_answer
 
             if cpt == checkpoint:
-                df.to_excel(str(cpt))
-                checkpoints_files.append(str(cpt))
+                df.to_excel("checkpointfile")
                 
                 checkpoint += 50
                 
@@ -255,7 +253,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
     with zipfile.ZipFile(zip_file_path, 'w') as zipf:
         zipf.write(conf_file_path, os.path.basename(conf_file_path))
         
-    return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path, checkpoints_files
+    return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path, "checkpoints_file"
 
 
 def get_columns(file,progress=gr.Progress()):