Update app.py

app.py

@@ -940,14 +940,23 @@ if query:
         dataset = Dataset.from_dict({'id': [], 'text': []})
 
     def save_comment(comment):
-        # Add the comment to the dataset
-        dataset['id'].append(len(dataset))
-        dataset['text'].append(comment)
+    # Check if the comments directory 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)
+
+    # Save the dataset to disk
+    dataset.save_to_disk('comments')
 
-        # Save the dataset to disk
-        dataset.save_to_disk('comments')
+    print('Comment saved to dataset.')
 
-        print('Comment saved to dataset.')
 
     st.title("Abstractalytics Web App")
     st.write("We appreciate your feedback!")