Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
63af44b
1
Parent(s):
f2514c2
lfg
Browse files
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)
|