From b003cdb29a8a3d73b3073d6f2761ffbbc650a045 Mon Sep 17 00:00:00 2001
From: Dennis Schiese <schiesedennis@googlemail.com>
Date: Mon, 10 Jun 2024 14:07:37 +0200
Subject: [PATCH] Adjusted button positioning, added feedback

---
 explanation_frontend.py | 36 ++++++++++++++++++++++++++---------
 util.py                 | 42 ++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 68 insertions(+), 10 deletions(-)

diff --git a/explanation_frontend.py b/explanation_frontend.py
index ba0d348..70a8cfc 100644
--- a/explanation_frontend.py
+++ b/explanation_frontend.py
@@ -3,7 +3,7 @@
 import requests
 import json
 from rdflib import Namespace
-from util import include_css
+from util import include_css, get_random_element, feedback_messages, feedback_icons
 from code_editor import code_editor
 import pandas as pd
 from decouple import config
@@ -226,14 +226,24 @@ def showExplanationContainer(component, lang, plainKey): # use the current compo
     with st.container(border=False):
         col1, col2 = st.columns([0.4,0.6])
         with col1:
-            st.markdown(f"""<div style="margin-bottom: 25px;"><h3>Template</h3><div>{template}</div></div>""", unsafe_allow_html=True)
-            st.markdown(f"""<div style="margin-bottom: 25px;"><h3>Generative</h3><div>{generative}</div></div>""", unsafe_allow_html=True)
-            st.warning("Which explanation do you think is better?")
-            templateButton, generativeButton, noneButton, placeholder = st.columns((1,1,1,2))
-            templateButton.button("Template", key=plainKey+"template", type="primary")
-            generativeButton.button("Generative", key=plainKey+"generative", type="primary")
-            noneButton.button("None", key=plainKey+"none")
-
+            template_based_container = st.container()
+            with template_based_container:
+                st.markdown(f"""<h3>Template</h3>""", unsafe_allow_html=True)
+                text_template, buttons_template = st.columns([0.85,0.15])
+                with text_template:
+                    st.markdown(f"""<div style="margin-bottom: 25px;">{template}</div>""", unsafe_allow_html=True)
+                with buttons_template:
+                    feedback_button(plainKey+"template"+"correct",":white_check_mark:")
+                    feedback_button(plainKey+"template"+"wrong",":x:")
+            generative_based_container = st.container()
+            with generative_based_container:
+                st.markdown(f"""<h3>Generative</h3>""", unsafe_allow_html=True)
+                text_template, buttons_template = st.columns([0.85,0.15])
+                with text_template:
+                    st.markdown(f"""<div style="margin-bottom: 25px;">{generative}</div>""", unsafe_allow_html=True)
+                with buttons_template:
+                    feedback_button(plainKey+"generative"+"correct",":white_check_mark:")
+                    feedback_button(plainKey+"generative"+"wrong",":x:")
         with col2:
             st.markdown("""<h3>Dataset</h3>""", unsafe_allow_html=True)
             code_editor(component["dataset"],lang="text", theme="default", options={"wrap": True})
@@ -241,6 +251,14 @@ def showExplanationContainer(component, lang, plainKey): # use the current compo
     with st.expander("Show used prompt"):   
         code_editor(component["prompt"], lang="text", theme="default", options={"wrap": True})
 
+def feedback_button(key, icon):
+    if st.button(icon, key=key, type="secondary"):
+        send_feedback()
+        st.toast(get_random_element(feedback_messages), icon=get_random_element(feedback_icons))
+
+def send_feedback():
+    return None
+
 def show_meta_data():
     if st.session_state.pipeline_finished:
         containerPipelineAndComponentsRadio = st.container(border=False)
diff --git a/util.py b/util.py
index 2b1572b..01263dd 100644
--- a/util.py
+++ b/util.py
@@ -1,6 +1,46 @@
+import random
+
 def include_css(st, filenames):
     content = ""
     for filename in filenames:
         with open(filename) as f:
             content += f.read()
-    st.markdown(f"<style>{content}</style>", unsafe_allow_html=True)
\ No newline at end of file
+    st.markdown(f"<style>{content}</style>", unsafe_allow_html=True)
+
+def get_random_element(elements):
+    return elements[random.randint(0, len(elements) - 1)]
+
+feedback_messages = [
+    "Hey, thanks a bunch for your help!",
+    "You rock! Thanks for your feedback.",
+    "You're the best; thanks for your help!",
+    "Much appreciated. Thanks!",
+    "You're a lifesaver; thank you!",
+    "I can't thank you enough for your support.",
+    "Big thanks for all your help!",
+    "I owe you one, thanks!",
+    "You're a \u2605; thanks for your help!",
+    "Thanks a million for your support!",
+    "Thanks for being such a great supporter!",
+    "Your help meant the world to me. Here's a hug-filled thank you!",
+    "Super grateful for your help.",
+    "Thanks a ton for your support!",
+    "Couldn't have done it without you. Thanks!",
+    "You're awesome; thanks for everything!",
+    "Thanks for being so supportive!",
+    "I really appreciate it; you're so kind!",
+    "Your help was right on time. Thanks!",
+    "I appreciate your help more than you'll ever know.",
+    "Kudos to you, and thanks a million!",
+    "Thank you for brightening my day!",
+    "You've been incredible! Thanks a ton.",
+    "Thank you. Let's go for a drink at the next opportunity!"
+]
+
+feedback_icons = [
+    "🙏",
+    "🤗",
+    "👍",
+    "👏",
+    "👌"
+]
\ No newline at end of file