NayabShakeel commited on
Commit
66e9050
·
verified ·
1 Parent(s): cc96b50

Create dataset.py

Browse files
Files changed (1) hide show
  1. dataset.py +29 -0
dataset.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from datasets import load_dataset, Dataset
2
+ import pandas as pd
3
+
4
+ DATASET_NAME = "your-username/quotes-dataset"
5
+
6
+ def load_user_quotes():
7
+ """Load user-submitted quotes from Hugging Face Dataset"""
8
+ try:
9
+ dataset = load_dataset(DATASET_NAME)
10
+ return dataset["train"].to_pandas().to_dict(orient="records")
11
+ except:
12
+ return []
13
+
14
+ def save_user_quote(quote, author):
15
+ """Save a new user quote with upvotes"""
16
+ data = load_user_quotes()
17
+ data.append({"quote": quote, "author": author, "upvotes": 0}) # Default upvotes = 0
18
+
19
+ dataset = Dataset.from_pandas(pd.DataFrame(data))
20
+ dataset.push_to_hub(DATASET_NAME, split="train")
21
+
22
+ def upvote_quote(index):
23
+ """Increment upvote count for a quote"""
24
+ data = load_user_quotes()
25
+ if 0 <= index < len(data):
26
+ data[index]["upvotes"] += 1
27
+
28
+ dataset = Dataset.from_pandas(pd.DataFrame(data))
29
+ dataset.push_to_hub(DATASET_NAME, split="train")