update

app.py

@@ -48,36 +48,56 @@ def get_method_dir(method):
 
 METHODS = ["No-XAI", "Dater", "Chain-of-Table", "Plan-of-SQLs"]
 
-def save_session_data(username, data):
+
+def save_session_data(username, data, is_update=False):
     try:
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        file_name = f'{username}_{timestamp}_session.json'
+        seed = data.get('seed', 'unknown')
+        start_time = data.get('start_time', datetime.now().isoformat())
+        file_name = f'{username}_seed{seed}_{start_time}_session.json'
+        file_name = "".join(c for c in file_name if c.isalnum() or c in ['_', '-', '.'])
+
         json_data = json.dumps(data, indent=4)
         temp_file_path = f"/tmp/{file_name}"
         with open(temp_file_path, 'w') as f:
             f.write(json_data)
+
         api = HfApi()
+        repo_path = "session_data_foward_simulation"
+
+        if is_update:
+            # If updating, we need to delete the old file first
+            existing_files = api.list_repo_files(repo_id="luulinh90s/Tabular-LLM-Study-Data", repo_type="space")
+            existing_file = next(
+                (f for f in existing_files if f.startswith(f'{repo_path}/{username}_seed{seed}_{start_time}')), None)
+            if existing_file:
+                api.delete_file(repo_id="luulinh90s/Tabular-LLM-Study-Data", repo_type="space",
+                                path_in_repo=existing_file)
+
         api.upload_file(
             path_or_fileobj=temp_file_path,
-            path_in_repo=f"session_data_foward_simulation/{file_name}",
+            path_in_repo=f"{repo_path}/{file_name}",
             repo_id="luulinh90s/Tabular-LLM-Study-Data",
             repo_type="space",
         )
         os.remove(temp_file_path)
-        logger.info(f"Session data saved for user {username} in Hugging Face Data Space")
+        logger.info(
+            f"Session data {'updated' if is_update else 'saved'} for user {username} in Hugging Face Data Space")
     except Exception as e:
-        logger.exception(f"Error saving session data for user {username}: {e}")
+        logger.exception(f"Error {'updating' if is_update else 'saving'} session data for user {username}: {e}")
+
 
 def load_session_data(username):
     try:
         api = HfApi()
         files = api.list_repo_files(repo_id="luulinh90s/Tabular-LLM-Study-Data", repo_type="space")
-        user_files = [f for f in files if f.startswith(f'session_data_foward_simulation/{username}_') and f.endswith('_session.json')]
+        user_files = [f for f in files if
+                      f.startswith(f'session_data_foward_simulation/{username}_') and f.endswith('_session.json')]
         if not user_files:
             logger.warning(f"No session data found for user {username}")
             return None
         latest_file = sorted(user_files, reverse=True)[0]
-        file_path = hf_hub_download(repo_id="luulinh90s/Tabular-LLM-Study-Data", repo_type="space", filename=latest_file)
+        file_path = hf_hub_download(repo_id="luulinh90s/Tabular-LLM-Study-Data", repo_type="space",
+                                    filename=latest_file)
         with open(file_path, 'r') as f:
             data = json.load(f)
         logger.info(f"Session data loaded for user {username} from Hugging Face Data Space")
@@ -130,14 +150,15 @@ def index():
             selected_samples = select_balanced_samples(all_samples)
             if len(selected_samples) == 0:
                 return "No common samples were found", 500
+            start_time = datetime.now().isoformat()
             session_data = {
                 'username': username,
-                'seed': seed,
+                'seed': str(seed),  # Store as string
                 'method': method,
                 'selected_samples': selected_samples,
                 'current_index': 0,
                 'responses': [],
-                'start_time': datetime.now().isoformat()
+                'start_time': start_time
             }
             save_session_data(username, session_data)
             return redirect(url_for('experiment', username=username))
@@ -273,7 +294,7 @@ def completed(username):
         session_data['true_percentage'] = true_percentage
         session_data['false_percentage'] = false_percentage
 
-        save_session_data(username, session_data)
+        save_session_data(username, session_data, is_update=True)  # Update the existing file
 
         return render_template('completed.html',
                                accuracy=accuracy,