Update app.py

app.py

@@ -1,46 +1,92 @@
+import streamlit as st
 import json
-from datetime import datetime
-from pathlib import Path
-from uuid import uuid4
+import os
 
-import streamlit as st
+# Initialize results file
+filename = "results.json"
 
-from huggingface_hub import CommitScheduler
+if not os.path.exists(filename):
+    with open(filename, "w") as file:
+        json.dump([], file)
 
-# Directory for JSON dataset
-JSON_DATASET_DIR = Path("json_dataset")
-JSON_DATASET_DIR.mkdir(parents=True, exist_ok=True)
+# Function to load existing data
+def load_existing_data():
+    with open(filename, "r") as file:
+        try:
+            return json.load(file)
+        except json.JSONDecodeError:
+            return []
 
-JSON_DATASET_PATH = JSON_DATASET_DIR / f"train-{uuid4()}.json"
+# Function to save feedback
+def save_feedback(prompt, response, feedback_comment, feedback_value):
+    existing_data = load_existing_data()
+    
+    # Check for existing feedback for the same prompt-response pair
+    found = False
+    for data in existing_data:
+        if data.get("prompt") == prompt and data.get("response") == response:
+            data["feedback"] = feedback_comment
+            data["positiveORnegative"] = "Positive" if feedback_value == 1 else "Negative"
+            found = True
+            break
+    
+    if not found:
+        # Add new feedback entry
+        new_entry = {
+            "prompt": prompt,
+            "response": response,
+            "feedback": feedback_comment,
+            "positiveORnegative": "Positive" if feedback_value == 1 else "Negative"
+        }
+        existing_data.append(new_entry)
+    
+    # Save updated data
+    with open(filename, "w") as file:
+        json.dump(existing_data, file, indent=4)
 
-scheduler = CommitScheduler(
-    repo_id="Feedback",
-    repo_type="dataset",
-    folder_path=JSON_DATASET_DIR,
-    path_in_repo="data",
-)
+# Streamlit UI
+st.title("Feedback-enabled Chatbot")
 
-# Function to greet the user
-def greet(name: str) -> str:
-    return "Hello " + name + "!"
+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
 
-# Function to save data as JSON
-def save_json(name: str, greetings: str) -> None:
-    with scheduler.lock:
-        with JSON_DATASET_PATH.open("a") as f:
-            json.dump({"name": name, "greetings": greetings, "datetime": datetime.now().isoformat()}, f)
-            f.write("\n")
+# User input section
+user_input = st.text_input("Enter your message:")
+if st.button("Get Response"):
+    # Generate AI response (replace with your model's logic)
+    ai_response = f"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)
 
-# Streamlit UI
-st.title("Greeting App")
-
-# Input and Output
-name = st.text_input("Enter your name:")
-if st.button("Greet"):
-    if name:
-        greeting = greet(name)
-        st.success(greeting)
-        save_json(name, greeting)
-        st.write("Greeting saved successfully!")
-    else:
-        st.error("Please enter your name.")
+# 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_value = 1
+            st.session_state.show_feedback_box = True
+    with col2:
+        if st.button("👎 Thumbs Down"):
+            st.session_state.feedback_value = 0
+            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_feedback(
+                st.session_state.prompts[-1],
+                st.session_state.responses[-1],
+                feedback_comment,
+                st.session_state.feedback_value,
+            )
+            st.success("Feedback saved successfully!")
+            st.session_state.show_feedback_box = False
+            st.session_state.show_feedback = False