reverted dataset save funcs to last version

app.py

@@ -79,47 +79,56 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         
         print(f"Creating dataset: {repo_id}")
         
-        # Create a simple dataset from a dictionary with a train split
+        # Create a simple dataset from a dictionary
         data = {
             "text": [conversation_data],
             "timestamp": [datetime.datetime.now().isoformat()],
             "id": [str(uuid.uuid4())]
         }
         
-        # Explicitly ensure the repository 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 about repository creation: {str(repo_error)}")
-        
-        # Create the dataset and push to hub
+        # Create the dataset directly
         dataset = Dataset.from_dict(data)
         
-        # Push to Hugging Face Hub using the standard method
+        # Push to Hugging Face Hub using the simpler method from documentation
         dataset.push_to_hub(
-            repo_id=repo_id,
-            token=api_key,
-            private=False
+            repo_id=repo_id,  # Include username in repo_id
+            token=api_key,    # Pass token explicitly
+            private=False     # Make it public
         )
         
         # Generate the URL for the dataset
         dataset_url = f"https://huggingface.co/datasets/{repo_id}"
         print(f"Dataset successfully pushed to: {dataset_url}")
         
-        # Verify dataset exists by checking the API
-        verify_result = verify_dataset_exists(repo_id)
-        if verify_result["exists"]:
-            return f"Successfully created dataset at {dataset_url}"
-        else:
-            return f"Dataset was uploaded, but it may take a few minutes to appear at {dataset_url}. Error: {verify_result['message']}"
-            
+        return f"Successfully created dataset at {dataset_url}"
     except Exception as e:
         import traceback
         error_trace = traceback.format_exc()
         print(f"Dataset creation error: {str(e)}\n{error_trace}")
         return f"Error creating dataset: {str(e)}\n\nTroubleshooting tips:\n1. Verify your HF_API_KEY is valid\n2. Try a simpler dataset name with only letters and underscores\n3. Check your permissions for the Misfits-and-Machines organization"
 
+@tool
+def Dataset_Creator_Tool(dataset_name: str, conversation_data: str) -> str:
+    """A tool that posts a new dataset of the current conversation to Hugging Face.
+    
+    Args:
+        dataset_name: Name for the dataset (will be prefixed with 'Misfits-and-Machines/')
+        conversation_data: String content to save to the dataset
+    
+    Returns:
+        Link to the created dataset or error message with troubleshooting steps
+    """
+    try:
+        print(f"Creating dataset '{dataset_name}' with {len(conversation_data)} characters of data")
+        print(f"Dataset will be created at Misfits-and-Machines/{dataset_name.replace(' ', '_').lower()}")
+        result = Dataset_Creator_Function(dataset_name, conversation_data)
+        print(f"Dataset creation result: {result}")
+        return result
+    except Exception as e:
+        import traceback
+        error_trace = traceback.format_exc()
+        return f"Error using Dataset Creator tool: {str(e)}\n{error_trace}"
+    
 def verify_dataset_exists(repo_id: str) -> dict:
     """Verify that a dataset exists and is valid on the Hugging Face Hub.
     
@@ -147,28 +156,6 @@ def verify_dataset_exists(repo_id: str) -> dict:
     except Exception as e:
         return {"exists": False, "message": f"Error verifying dataset: {str(e)}"}
 
-@tool
-def Dataset_Creator_Tool(dataset_name: str, conversation_data: str) -> str:
-    """A tool that posts a new dataset of the current conversation to Hugging Face.
-    
-    Args:
-        dataset_name: Name for the dataset (will be prefixed with 'Misfits-and-Machines/')
-        conversation_data: String content to save to the dataset
-    
-    Returns:
-        Link to the created dataset or error message with troubleshooting steps
-    """
-    try:
-        print(f"Creating dataset '{dataset_name}' with {len(conversation_data)} characters of data")
-        print(f"Dataset will be created at Misfits-and-Machines/{dataset_name.replace(' ', '_').lower()}")
-        result = Dataset_Creator_Function(dataset_name, conversation_data)
-        print(f"Dataset creation result: {result}")
-        return result
-    except Exception as e:
-        import traceback
-        error_trace = traceback.format_exc()
-        return f"Error using Dataset Creator tool: {str(e)}\n{error_trace}"
-
 @tool
 def Check_Dataset_Validity(dataset_name: str) -> str:
     """A tool that checks if a dataset exists and is valid on Hugging Face.