Update excel_chat.py

excel_chat.py

@@ -220,7 +220,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
     filtred_df = filter_df(df, search_col, keywords)
 
     cpt = 1
-    checkpoint = 2
+    checkpoint = 50
     for index, row in filtred_df.iterrows():
         concatenated_content = "\n\n".join(f"{column_name}: {str(row[column_name])}" for column_name in source_cols)
         if not concatenated_content == "\n\n".join(f"{column_name}: nan" for column_name in source_cols):
@@ -236,20 +236,17 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
 
             try:
                 if cpt == checkpoint:
-                    df2 = df
-                    df2.to_excel("checkpointfile.xlsx", index=False)
-                    
+                    df.to_excel("checkpointfile.xlsx", index=False)
                     checkpoint += 1
+                    
             except Exception as e:
-                print(e)
-                print("no checkpoint")
+                print(f"no checkpoint : {e}")
             
             cpt += 1
             # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}!')
 
     df.to_excel(file_name, index=False)
     
-    checkpoints_files.append(file_name)
     print(checkpoints_files)
     
     zip_file_path = 'config_file.zip'