Change in HF upload URL display

app.py

@@ -56,7 +56,7 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
     
     Args:
         dataset_name: Name for the dataset (will be prefixed with username)
-        conversation_data: String representation of conversation to save (plain text)
+        conversation_data: String representing the conversation data
     
     Returns:
         URL of the created dataset or error message
@@ -73,7 +73,7 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         # Get username for the repository name
         try:
             user_info = hf_api.whoami()
-            username = user_info.get("name", "Misfits-and-Machines")  # Use the provided username as default
+            username = user_info.get("name", "Misfits-and-Machines")  # fallback
         except Exception as e:
             return f"Error authenticating with Hugging Face: {str(e)}"
         
@@ -83,7 +83,7 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         
         print(f"Creating dataset repository: {repo_id}")
         
-        # Create a conversation object
+        # Build conversation object
         conversation = {
             "text": conversation_data,
             "timestamp": datetime.datetime.now().isoformat(),
@@ -93,23 +93,25 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         # Create a Hugging Face dataset
         dataset = Dataset.from_dict({"conversations": [conversation]})
         
-        # First try to create the repository explicitly
+        # Create or confirm the repo exists
         try:
             hf_api.create_repo(repo_id=repo_id, repo_type="dataset", exist_ok=True)
             print(f"Repository {repo_id} created or already exists")
         except Exception as repo_error:
             print(f"Note: Repository creation returned: {str(repo_error)}")
         
-        # Push to the Hugging Face Hub with explicit parameters
+        # Push the dataset
         dataset.push_to_hub(
             repo_id=repo_id,
             token=api_key,
-            create_repo=True,  # Create the repo if it doesn't exist
-            repo_type="dataset",  # Explicitly specify this is a dataset
+            create_repo=True, 
+            repo_type="dataset",
             commit_message="Add conversation dataset"
         )
         
-        return f"Successfully created dataset at https://huggingface.co/datasets/{repo_id}"
+        dataset_url = f"https://huggingface.co/datasets/{repo_id}"
+        print(f"Dataset successfully pushed to {dataset_url}")
+        return dataset_url
     except Exception as e:
         import traceback
         error_trace = traceback.format_exc()
@@ -123,10 +125,11 @@ def Dataset_Creator_Tool(dataset_name: str, conversation_data: str) -> str:
     Args:
         dataset_name: Name for the dataset (will be prefixed with username)
         conversation_data: String content to save to the dataset (no JSON conversion needed)
-    """
+"""
     try:
-        dataset_creator_response = Dataset_Creator_Function(dataset_name, conversation_data)
-        return dataset_creator_response
+        result = Dataset_Creator_Function(dataset_name, conversation_data)
+        print(f"Dataset_Creator_Function returned: {result}")
+        return result
     except Exception as e:
         return f"Error using Dataset Creator tool: {str(e)}"