Update excel_chat.py

excel_chat.py

@@ -7,6 +7,7 @@ import numpy as np
 from groq import Groq
 import anthropic
 from users_management import update_json, users
+from code_df_custom import load_excel
 
 #users = ['maksG', 'AlmaA', 'YchK']
 
@@ -114,8 +115,9 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
 
 def get_columns(file,progress=gr.Progress()):
     if file is not None:
-        df = pd.read_excel(file)
+        #df = pd.read_excel(file)
+        filename, df = load_excel(file)
         columns = list(df.columns)
-        return gr.update(choices=columns), gr.update(choices=columns), gr.update(choices=columns), gr.update(choices=columns + [""]), gr.update(choices=columns + ['[ALL]']), df.head(5)
+        return gr.update(choices=columns), gr.update(choices=columns), gr.update(choices=columns), gr.update(choices=columns + [""]), gr.update(choices=columns + ['[ALL]']), df.head(5), filename, df
     else:
-        return gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), pd.DataFrame()
+        return gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), gr.update(choices=[]), pd.DataFrame(), '', pd.DataFrame()