Changed to use string instead of json inside the hugging face dataset upload function

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: JSON string or formatted text of conversation to save
+        conversation_data: String representation of conversation to save (plain text)
     
     Returns:
         URL of the created dataset or error message
@@ -72,25 +72,18 @@ def Dataset_Creator_Function(dataset_name: str, conversation_data: str) -> str:
         # Format the full repo name
         repo_id = f"{username}/{dataset_name}"
         
-        # If conversation_data is a string, try to parse it as JSON
-        if isinstance(conversation_data, str):
-            try:
-                conversation = json.loads(conversation_data)
-            except json.JSONDecodeError:
-                # If not valid JSON, treat as raw text
-                conversation = {"text": conversation_data}
-        else:
-            conversation = conversation_data
-            
-        # Add metadata
-        conversation["timestamp"] = datetime.datetime.now().isoformat()
-        conversation["dataset_id"] = str(uuid.uuid4())
+        # Create a conversation object
+        conversation = {
+            "text": conversation_data,
+            "timestamp": datetime.datetime.now().isoformat(),
+            "dataset_id": str(uuid.uuid4())
+        }
         
         # Create a Hugging Face dataset
         dataset = Dataset.from_dict({"conversations": [conversation]})
         
         # Push to the Hugging Face Hub
-        dataset.push_to_hub(repo_id, token=os.getenv("HF_API_KEY", ""))
+        dataset.push_to_hub(repo_id, token=os.getenv("HUGGINGFACE_API_KEY", ""))
         
         return f"Successfully created dataset at https://huggingface.co/datasets/{repo_id}"
     except Exception as e:
@@ -102,7 +95,7 @@ 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: Conversation history as JSON string or formatted text
+        conversation_data: String content to save to the dataset (no JSON conversion needed)
     """
     try:
         dataset_creator_response = Dataset_Creator_Function(dataset_name, conversation_data)