YchKhan commited on
Commit
378fa83
1 Parent(s): 6aa107f

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +7 -4
excel_chat.py CHANGED
@@ -88,10 +88,13 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
88
  print(f'xlsxfile = {excel_file}')
89
  df = pd.read_excel(excel_file)
90
  df[dest_col] = ""
91
- try:
92
- file_name = url.split("/")[-2] + ".xlsx"
93
- except:
94
- file_name = excel_file
 
 
 
95
  print(f"Keywords: {keywords}")
96
 
97
  filtred_df = filter_df(df, search_col, keywords)
 
88
  print(f'xlsxfile = {excel_file}')
89
  df = pd.read_excel(excel_file)
90
  df[dest_col] = ""
91
+ if file:
92
+ file_name = excel_file.split('.xlsx')[0] + "with_" + dest_col.replace(' ', '_') + ".xlsx"
93
+ elif url.endswith('Docs/', 'Docs'):
94
+ file_name = url.split("/Docs")[0].split("/")[-1] + ".xlsx"
95
+ else:
96
+ file_name = "meeting_recap_grid.xlsx"
97
+
98
  print(f"Keywords: {keywords}")
99
 
100
  filtred_df = filter_df(df, search_col, keywords)