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 = 50
+    checkpoint = 2
     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):
@@ -235,11 +235,11 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
             df.at[index, dest_col] = llm_answer
 
             try:
-                
                 if cpt == checkpoint:
-                    df.to_excel("checkpointfile", index=False)
+                    df2 = df
+                    df2.to_excel("checkpointfile", index=False)
                     
-                    checkpoint += 50
+                    checkpoint += 1
             except Exception:
                 print("no checkpoint")