Update app.py

app.py

@@ -940,17 +940,14 @@ if query:
         dataset = Dataset.from_dict({'id': [], 'text': []})
 
     def save_comment(comment):
-        # Check if the comments directory exists
+        # Check if the dataset exists
         if os.path.exists('comments'):
-            # Load the dataset from disk
             dataset = Dataset.load_from_disk('comments')
         else:
-            # Create a new dataset
             dataset = Dataset.from_dict({'id': [], 'text': []})
 
-        # Add the comment to the dataset
-        dataset['id'].append(len(dataset))
-        dataset['text'].append(comment)
+        # Append the new comment to the dataset
+        dataset = dataset.append({'id': len(dataset), 'text': comment})
 
         # Save the dataset to disk
         dataset.save_to_disk('comments')