Fix filename in preview section of WebUI

src/autogluon_assistant/ui/pages/preview.py

@@ -1,3 +1,5 @@
+import os
+
 import streamlit as st
 from st_aggrid import AgGrid, GridOptionsBuilder
 
@@ -44,7 +46,7 @@ def preview_dataset():
         elif st.session_state.selected_dataset == "Sample Dataset":
             file_options = get_sample_files()
         if st.session_state.output_file is not None:
-            file_options.append(st.session_state.output_filename)
+            file_options.append(os.path.basename(st.session_state.output_filename))
         selected_file = st.selectbox(
             "Preview File",
             options=file_options,
@@ -64,7 +66,9 @@ def preview_dataset():
             """,
                 unsafe_allow_html=True,
             )
-            if selected_file == st.session_state.output_filename:
+            if st.session_state.output_filename and selected_file == os.path.basename(
+                st.session_state.output_filename
+            ):
                 output_file = st.session_state.output_file
                 gb = GridOptionsBuilder.from_dataframe(output_file)
                 gb.configure_pagination()

src/autogluon_assistant/ui/pages/task.py

@@ -239,7 +239,7 @@ def download_log_button():
         st.download_button(
             label="📥  Download Logs",
             data=st.session_state.logs,
-            file_name=f"autogluon_logs_{st.session_state.user_session_id}.txt",
+            file_name="aga_logs.txt",
             mime="text/plain",
         )
 
@@ -262,7 +262,6 @@ def toggle_running_state():
     st.session_state.stage_container = copy.deepcopy(INITIAL_STAGE)
     st.session_state.stage_status = {}
     st.session_state.increment_time = 0
-    st.session_state.progress_bar = None
 
 
 def toggle_cancel_state():
@@ -409,6 +408,8 @@ def run_section():
 def setup_local_dataset():
     """Download all files from GitHub directory to local directory"""
     dataset_dir = Path("sample_dataset/knot_theory")
+    if dataset_dir.exists():
+        return
     dataset_dir.mkdir(parents=True, exist_ok=True)
 
     api_url = "https://api.github.com/repos/mli/ag-docs/contents/knot_theory"

src/autogluon_assistant/ui/utils/utils.py

@@ -41,15 +41,11 @@ def generate_output_filename():
     Returns:
         str: A unique filename for the output CSV file.
     """
-    user_session_id = get_user_session_id()
-    unique_id = user_session_id[:8]
+    user_data_dir = get_user_data_dir()
     timestamp = pd.Timestamp.now().strftime("%Y%m%d_%H%M%S")
-    directory_path = os.path.join(unique_id, timestamp)
-    # Create directories if they don't exist
+    directory_path = os.path.join(user_data_dir, timestamp)
     os.makedirs(directory_path, exist_ok=True)
-    # Generate full file path
     output_filepath = os.path.join(directory_path, "output.csv")
-
     return output_filepath