Update app.py

app.py

@@ -62,31 +62,29 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         URL of the created dataset or error message
     """
     try:
-        # Get API key from environment variables - try both possible names
+        # Get API key from environment variables
         api_key = os.getenv("HF_API_KEY") or os.getenv("HUGGINGFACE_API_KEY", "")
         if not api_key:
             return "Error: No Hugging Face API key found in environment variables"
             
+        # Force the username to be the known value
+        username = "Misfits-and-Machines"
+        
         # Initialize Hugging Face API
         hf_api = HfApi(token=api_key)
         
-        # Retrieve username for the repository name
-        try:
-            user_info = hf_api.whoami()
-            username = user_info.get("name", "Misfits-and-Machines")  # fallback
-        except Exception as e:
-            return f"Error authenticating with Hugging Face: {str(e)}"
-        
         # Sanitize dataset name and create repo_id
         safe_dataset_name = dataset_name.replace(" ", "-").lower()
         repo_id = f"{username}/{safe_dataset_name}"
         
-        print(f"Creating or confirming repo: {repo_id}")
-        # Create or confirm the repository exists
+        print(f"Creating dataset repository: {repo_id}")
+        
+        # Create the repository explicitly
         try:
             hf_api.create_repo(repo_id=repo_id, repo_type="dataset", exist_ok=True)
+            print(f"Repository {repo_id} created or confirmed")
         except Exception as repo_error:
-            print(f"Note: Repository creation returned: {str(repo_error)}")
+            print(f"Repository creation note: {str(repo_error)}")
         
         # Build conversation object
         conversation = {
@@ -98,16 +96,15 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         # Create a Hugging Face dataset
         dataset = Dataset.from_dict({"conversations": [conversation]})
         
-        # Push to the Hub without the unsupported arguments
+        # Push dataset to the Hub with minimal parameters
         dataset.push_to_hub(
             repo_id=repo_id,
-            token=api_key,
-            commit_message="Add conversation dataset"
+            token=api_key
         )
         
         dataset_url = f"https://huggingface.co/datasets/{repo_id}"
-        print(f"Dataset successfully pushed to {dataset_url}")
-        return dataset_url
+        print(f"Dataset successfully pushed to: {dataset_url}")
+        return f"Successfully created dataset: {dataset_url}"
     except Exception as e:
         import traceback
         error_trace = traceback.format_exc()
@@ -121,13 +118,16 @@ 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:
+        print(f"Creating dataset '{dataset_name}' with {len(conversation_data)} characters of data")
         result = Dataset_Creator_Function(dataset_name, conversation_data)
-        print(f"Dataset_Creator_Function returned: {result}")
+        print(f"Result: {result}")
         return result
     except Exception as e:
-        return f"Error using Dataset Creator tool: {str(e)}"
+        import traceback
+        error_trace = traceback.format_exc()
+        return f"Error using Dataset Creator tool: {str(e)}\n{error_trace}"
 
 
 @tool