Spaces:
Sleeping
Sleeping
File size: 3,600 Bytes
9e7b3d7 918f0a4 757d134 9e7b3d7 260127a 757d134 9e7b3d7 918f0a4 260127a 918f0a4 757d134 9e7b3d7 918f0a4 9e7b3d7 757d134 9e7b3d7 918f0a4 9e7b3d7 260127a 9e7b3d7 260127a 9e7b3d7 918f0a4 9e7b3d7 918f0a4 260127a 918f0a4 9e7b3d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import streamlit as st
import json
from datetime import datetime
from pathlib import Path
from uuid import uuid4
import os
from huggingface_hub import CommitScheduler
# Setup the directory for saving data
JSON_DATASET_DIR = Path("feedback_dataset")
JSON_DATASET_DIR.mkdir(parents=True, exist_ok=True)
# Define a unique file name
JSON_DATASET_PATH = JSON_DATASET_DIR / f"feedback-{uuid4()}.json"
# Scheduler configuration for your dataset repo
scheduler = CommitScheduler(
repo_id="Utsav2001/Feedback", # Replace with your dataset repo
repo_type="dataset",
folder_path=JSON_DATASET_DIR, # Local directory to sync
path_in_repo="data", # Path in the dataset repository
token=os.getenv('hf_write') # Ensure your HF write token is set
)
# Streamlit UI
st.title("OpenROAD-Assistant")
if "prompts" not in st.session_state:
st.session_state.prompts = []
if "responses" not in st.session_state:
st.session_state.responses = []
if "show_feedback" not in st.session_state:
st.session_state.show_feedback = False
if "feedbacks" not in st.session_state:
st.session_state.feedbacks = []
if "feedback_type" not in st.session_state:
st.session_state.feedback_type = None
# Function to save data using Hugging Face CommitScheduler
def save_feedback_to_hub(prompt, ai_response, feedback_type, feedback_comment):
try:
with scheduler.lock:
with JSON_DATASET_PATH.open("a") as f:
json.dump(
{
"prompt": prompt,
"ai_response": ai_response,
"feedback_type": feedback_type,
"feedback_comment": feedback_comment,
"datetime": datetime.now().isoformat(),
},
f,
)
f.write("\n")
# Push the changes to Hugging Face Hub
scheduler.push_to_hub()
return "Feedback saved successfully to Hugging Face Hub!"
except Exception as e:
return f"Error saving feedback: {e}"
# User input section
user_input = st.text_input("Enter your message:")
if st.button("Get Response"):
# Simulate an AI response (replace with actual AI model logic)
ai_response = f"This is a response to: {user_input}"
st.session_state.prompts.append(user_input)
st.session_state.responses.append(ai_response)
st.session_state.show_feedback = True
st.write("AI Response:", ai_response)
# Feedback section
if st.session_state.show_feedback and len(st.session_state.prompts) > 0 and len(st.session_state.responses) > 0:
st.write("Provide feedback for the last interaction:")
col1, col2 = st.columns(2)
with col1:
if st.button("👍 Thumbs Up"):
st.session_state.feedback_type = "Positive"
st.session_state.show_feedback_box = True
with col2:
if st.button("👎 Thumbs Down"):
st.session_state.feedback_type = "Negative"
st.session_state.show_feedback_box = True
if "show_feedback_box" in st.session_state and st.session_state.show_feedback_box:
feedback_comment = st.text_area("Your Feedback:")
if st.button("Submit Feedback"):
save_status = save_feedback_to_hub(
st.session_state.prompts[-1],
st.session_state.responses[-1],
st.session_state.feedback_type,
feedback_comment
)
st.success(save_status)
st.session_state.show_feedback_box = False
st.session_state.show_feedback = False
|