Utsav2001 commited on
Commit
757d134
·
verified ·
1 Parent(s): 1d6dedc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -61
app.py CHANGED
@@ -1,64 +1,46 @@
1
- import streamlit as st
2
- from transformers import pipeline
3
- from datasets import Dataset, load_dataset
4
- from huggingface_hub import HfApi, HfFolder
5
- import os
6
-
7
- # Set your Hugging Face API token
8
- HF_TOKEN = os.getenv("HF_token") # Ensure this environment variable is set securely
9
-
10
- # Initialize Hugging Face API
11
- api = HfApi()
12
- api.set_access_token(HF_TOKEN)
13
-
14
- # Initialize or load the dataset from Hugging Face
15
- dataset_repo = "utsav2001/Feedback"
16
- try:
17
- feedback_dataset = load_dataset(dataset_repo, split="train")
18
- except Exception as e:
19
- st.warning(f"Could not load dataset: {e}. Initializing a new dataset.")
20
- feedback_dataset = Dataset.from_dict({"input": [], "response": [], "rating": [], "feedback": []})
21
 
22
- # App title
23
- st.title("Sentiment Bot with Feedback")
24
-
25
- # Sentiment analysis pipeline
26
- sentiment_pipeline = pipeline("sentiment-analysis", framework="pt")
27
-
28
- # Input text area
29
- user_input = st.text_area("Enter the Review")
30
 
31
- if st.button("Analyze"):
32
- if user_input.strip():
33
- # Analyze sentiment
34
- result = sentiment_pipeline(user_input)
35
- sentiment = result[0]['label']
36
- confidence = result[0]['score']
37
-
38
- # Display sentiment and confidence
39
- st.write("Sentiment:", sentiment)
40
- st.write("Confidence Score:", confidence)
41
-
42
- # Collect user feedback
43
- st.subheader("Provide Feedback")
44
- rating = st.slider("Rate the response (out of 4):", 1, 4)
45
- feedback = st.text_area("Any additional feedback?")
46
-
47
- if st.button("Submit Feedback"):
48
- # Save feedback to the dataset
49
- new_data = {"input": user_input, "response": sentiment, "rating": rating, "feedback": feedback}
50
- feedback_dataset = feedback_dataset.add_item(new_data)
51
- st.success("Feedback submitted!")
52
-
53
- # Push the updated dataset to Hugging Face Hub
54
- try:
55
- feedback_dataset.push_to_hub(dataset_repo, token=HF_TOKEN)
56
- st.info("Feedback dataset updated on Hugging Face Hub.")
57
- except Exception as e:
58
- st.error(f"Failed to push dataset to Hub: {e}")
 
 
 
 
 
 
 
 
 
59
  else:
60
- st.error("Please enter a review before analyzing.")
61
-
62
- # Display feedback collected so far
63
- st.subheader("Feedback Collected")
64
- st.dataframe(feedback_dataset.to_pandas())
 
1
+ import json
2
+ from datetime import datetime
3
+ from pathlib import Path
4
+ from uuid import uuid4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
+ import streamlit as st
 
 
 
 
 
 
 
7
 
8
+ from huggingface_hub import CommitScheduler
9
+
10
+ # Directory for JSON dataset
11
+ JSON_DATASET_DIR = Path("json_dataset")
12
+ JSON_DATASET_DIR.mkdir(parents=True, exist_ok=True)
13
+
14
+ JSON_DATASET_PATH = JSON_DATASET_DIR / f"train-{uuid4()}.json"
15
+
16
+ scheduler = CommitScheduler(
17
+ repo_id="example-space-to-dataset-json",
18
+ repo_type="dataset",
19
+ folder_path=JSON_DATASET_DIR,
20
+ path_in_repo="data",
21
+ )
22
+
23
+ # Function to greet the user
24
+ def greet(name: str) -> str:
25
+ return "Hello " + name + "!"
26
+
27
+ # Function to save data as JSON
28
+ def save_json(name: str, greetings: str) -> None:
29
+ with scheduler.lock:
30
+ with JSON_DATASET_PATH.open("a") as f:
31
+ json.dump({"name": name, "greetings": greetings, "datetime": datetime.now().isoformat()}, f)
32
+ f.write("\n")
33
+
34
+ # Streamlit UI
35
+ st.title("Greeting App")
36
+
37
+ # Input and Output
38
+ name = st.text_input("Enter your name:")
39
+ if st.button("Greet"):
40
+ if name:
41
+ greeting = greet(name)
42
+ st.success(greeting)
43
+ save_json(name, greeting)
44
+ st.write("Greeting saved successfully!")
45
  else:
46
+ st.error("Please enter your name.")