Update app.py

app.py

@@ -1,10 +1,77 @@
 import streamlit as st
-from transformers import pipeline
-
-st.title("Sentiment Bot")
-sentiment_pipeline = pipeline("sentiment-analysis", framework="pt")
-user_input = st.text_area("Enter the Review")
-if st.button("Analyze"):
-    result = sentiment_pipeline(user_input)
-    st.write("Sentiment:", result[0]['label'])
-    st.write("Confidence Score:", result[0]['score'])
+import json
+import os
+
+# Initialize results file
+filename = "results.json"
+
+if not os.path.exists(filename):
+    with open(filename, "w") as file:
+        json.dump([], file)
+
+# Function to load existing data
+def load_existing_data():
+    with open(filename, "r") as file:
+        try:
+            return json.load(file)
+        except json.JSONDecodeError:
+            return []
+
+# 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)
+
+# Streamlit UI
+st.title("Feedback-enabled Chatbot")
+
+if "prompts" not in st.session_state:
+    st.session_state.prompts = []
+if "responses" not in st.session_state:
+    st.session_state.responses = []
+
+# 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.write("AI Response:", ai_response)
+
+# Feedback section
+if len(st.session_state.prompts) > 0 and len(st.session_state.responses) > 0:
+    st.write("Provide feedback for the last interaction:")
+    feedback_comment = st.text_area("Your Feedback:")
+    feedback_value = st.radio("Feedback Type:", options=["Positive", "Negative"], index=0)
+    if st.button("Submit Feedback"):
+        feedback_type = 1 if feedback_value == "Positive" else 0
+        save_feedback(
+            st.session_state.prompts[-1],
+            st.session_state.responses[-1],
+            feedback_comment,
+            feedback_type,
+        )
+        st.success("Feedback saved successfully!")