Arslan1997 commited on
Commit
63af44b
·
1 Parent(s): f2514c2
src/managers/session_manager.py CHANGED
@@ -187,7 +187,7 @@ This dataset appears clean with consistent formatting and no missing values, mak
187
 
188
 
189
 
190
- def update_session_dataset(self, session_id: str, datasets, names, desc: str):
191
  """
192
  Update session with new dataset and optionally auto-generate description
193
  """
@@ -206,7 +206,7 @@ This dataset appears clean with consistent formatting and no missing values, mak
206
  # Register the new dataset in DuckDB
207
 
208
  # Auto-generate description if we have datasets
209
- if datasets:
210
  try:
211
  generated_desc = generate_dataset_description(datasets, desc, names)
212
  desc = generated_desc # No need to format again since it's already formatted
 
187
 
188
 
189
 
190
+ def update_session_dataset(self, session_id: str, pre_generated=False, datasets, names, desc: str):
191
  """
192
  Update session with new dataset and optionally auto-generate description
193
  """
 
206
  # Register the new dataset in DuckDB
207
 
208
  # Auto-generate description if we have datasets
209
+ if datasets and pre_generated==False:
210
  try:
211
  generated_desc = generate_dataset_description(datasets, desc, names)
212
  desc = generated_desc # No need to format again since it's already formatted
src/routes/session_routes.py CHANGED
@@ -307,8 +307,8 @@ async def upload_dataframe(
307
  # Create datasets dictionary with the new dataset
308
  datasets = {name: new_df}
309
 
310
- # Update the session with the new dataset (this will replace any existing datasets)
311
- app_state.update_session_dataset(session_id, datasets, [name], desc)
312
 
313
  # Log session state AFTER upload
314
  session_state_after_upload = app_state.get_session_state(session_id)
 
307
  # Create datasets dictionary with the new dataset
308
  datasets = {name: new_df}
309
 
310
+ # Update the session with the new dataset (this will replace any existing datasets) but not update desc, as that is passed already
311
+ app_state.update_session_dataset(session_id, datasets,pre_generated=True, [name], desc)
312
 
313
  # Log session state AFTER upload
314
  session_state_after_upload = app_state.get_session_state(session_id)