Update app.py

app.py

@@ -98,39 +98,53 @@ token = os.getenv("HF_TOKEN")  # Replace "YOUR_AUTHENTICATION_TOKEN" with your a
 login(token=token)
 
 # Function to push feedback data to Hugging Face Hub dataset
-def push_to_dataset(feedback, comments):
+def push_to_dataset(feedback, comments, image):
     # Load existing dataset or create a new one if it doesn't exist
     try:
-        ds = load_dataset("YashB1/Feedbacks", split="evaluation")
+        ds = load_dataset("YashB1/Feedbacks_Version2", split="evaluation")
     except FileNotFoundError:
         # If dataset doesn't exist, create a new one
-        ds = Dataset.from_dict({"feedback": [], "comments": []})
+        ds = Dataset.from_dict({"feedback": [], "comments": [], "image": []})
 
     # Add new feedback to the dataset
-    new_data = {"feedback": [feedback], "comments": [comments]}  # Convert feedback and comments to lists
+    new_data = {"feedback": [feedback], "comments": [comments], "image": [image]}  # Convert feedback and comments to lists
     new_data = Dataset.from_dict(new_data)  
-    ds = concatenate_datasets([ds,new_data])
 
-    # Save dataset back to Hugging Face Hub
-    ds.push_to_hub("YashB1/Feedbacks", split="evaluation")
+    ds = concatenate_datasets([ds, new_data])
 
-def collect_feedback():
-    st.title("Feedback Collection App")
+    # Push the updated dataset to Hugging Face Hub
+    ds.push_to_hub("YashB1/Feedbacks_Version2", split="evaluation")
 
-    # Display thumbs up/down buttons for feedback
-    feedback = st.radio("Was this helpful?", ('👍', '👎'))
+def main():
+    st.header("Feedback and Image Upload Example")
+    st.write("Provide feedback using thumbs up or thumbs down, then upload an image, and add comments.")
 
-    # Text input for comments
-    comments = st.text_area("Please provide any comments or suggestions:")
+    # Feedback section
+    feedback = st.radio("Feedback:", options=["Thumbs Up 👍", "Thumbs Down 👎"])
 
+    # Image upload section
+    st.write("Click the button below to upload an image.")
+    image_data = paste(key="image_clipboard", label="Upload Image")
+    if image_data is not None:
+        header, encoded = image_data.split(",", 1)
+        binary_data = base64.b64decode(encoded)
+        bytes_data = BytesIO(binary_data)
+        st.image(bytes_data, caption="Uploaded Image", use_column_width=True)
+    else:
+        st.write("No image uploaded yet.")
+
+    # Comments section
+    comments = st.text_area("Comments:")
+
+    # Button to submit feedback, image, and comments
     if st.button("Submit"):
-        if feedback == '👍':
-            st.success("Thank you for your positive feedback!")
-        elif feedback == '👎':
-            st.error("We're sorry to hear that. Please let us know how we can improve.")
-        
-        # Push feedback data to Hugging Face Hub dataset
-        push_to_dataset(feedback, comments)
+        if feedback and comments.strip() and image_data:
+            # Convert image_data to string
+            image_data_str = str(image_data)
+            push_to_dataset(feedback, comments, image_data_str)
+            st.success("Feedback submitted successfully!")
+        else:
+            st.error("Please provide feedback, comments, and upload an image.")
 
 if __name__ == "__main__":
-    collect_feedback()
+    main()