Update excel_chat.py

excel_chat.py

@@ -234,14 +234,17 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
             print(f"{cpt}/{len(filtred_df)}\nQUERY:\n{prompt[0]}\nCONTENT:\n{concatenated_content[:200]}...\n\nANSWER:\n{llm_answer}")
             df.at[index, dest_col] = llm_answer
 
-            if cpt == checkpoint:
-                df.to_excel("checkpointfile", index=False)
-                
-                checkpoint += 50
+            try:
                 
+                if cpt == checkpoint:
+                    df.to_excel("checkpointfile", index=False)
+                    
+                    checkpoint += 50
+            except Exception:
+                print("no checkpoint")
             
             cpt += 1
-            # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}')
+            # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}!')
 
     df.to_excel(file_name, index=False)
     
@@ -253,7 +256,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_file"
+    return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path, "checkpointfile"
 
 
 def get_columns(file,progress=gr.Progress()):