Update app.py

app.py

@@ -79,17 +79,29 @@ class HFDatasetLogger:
             hf_token: HuggingFace authentication token
             private: Whether to create a private dataset
         """
-        self.dataset_name = dataset_name
         self.hf_token = hf_token
         self.private = private
         self.api = HfApi()
         self.dataset_exists = False
         self.parquet_filename = None
         
+        # If dataset_name doesn't have a username prefix, get it from the token
+        if "/" not in dataset_name:
+            try:
+                user_info = self.api.whoami(token=hf_token)
+                username = user_info["name"]
+                self.dataset_name = f"{username}/{dataset_name}"
+                print(f"  Resolved dataset name: {self.dataset_name}")
+            except Exception as e:
+                print(f"  Could not get username from token: {e}")
+                self.dataset_name = dataset_name
+        else:
+            self.dataset_name = dataset_name
+        
         # Check if dataset exists by listing files in the repo
         try:
             files = list_repo_files(
-                dataset_name,
+                self.dataset_name,
                 repo_type="dataset",
                 token=hf_token,
             )