Add Language Setting

app.py

@@ -9,6 +9,9 @@ import streamlit as st
 from streamlit_pills import pills
 from streamlit_feedback import streamlit_feedback
 
+from langdetect import detect_langs
+from langcodes import Language
+
 import sqlite3
 import pandas as pd
 from datasets import load_dataset
@@ -29,6 +32,10 @@ headers = {
 }
 amp_api_key = os.getenv('AMPLITUDE_TOKEN')
 
+def identifyLanguage(response):
+    lang_code = detect_langs(response)[0].lang
+    return Language.make(language=lang_code).display_name()
+
 def thumbs_feedback(feedback, **kwargs):
     """
     Sends feedback to Amplitude Analytics
@@ -40,9 +47,11 @@ def thumbs_feedback(feedback, **kwargs):
                 "event_type": "provided_feedback",
                 "event_properties": {
                     "Space Name": kwargs.get("demo_name", "Unknown"),
+                    "Demo Type": "Agent",
                     "query": kwargs.get("prompt", "No user input"),
                     "response": kwargs.get("response", "No chat response"),
-                    "feedback": feedback["score"]
+                    "feedback": feedback["score"],
+                    "Response Language": kwargs.get("language", "Unknown language")
                 }
             }]
         }
@@ -163,8 +172,10 @@ def launch_bot():
                 "event_type": "submitted_query",
                 "event_properties": {
                     "Space Name": cfg['demo_name'],
+                    "Demo Type": "Agent",
                     "query": st.session_state.messages[-2]["content"],
-                    "response": st.session_state.messages[-1]["content"]
+                    "response": st.session_state.messages[-1]["content"],
+                    "Response Language": identifyLanguage(st.session_state.messages[-1])
                 }
             }]
         }
@@ -182,7 +193,8 @@ def launch_bot():
         streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
                                       kwargs = {"prompt": st.session_state.messages[-2]["content"],
                                                 "response": st.session_state.messages[-1]["content"],
-                                                "demo_name": cfg["demo_name"]})
+                                                "demo_name": cfg["demo_name"],
+                                                "language": identifyLanguage(st.session_state.messages[-1]["content"])})
         
     log_placeholder = st.empty()
     with log_placeholder.container():

requirements.txt

@@ -4,6 +4,8 @@ python-dotenv==1.0.1
 streamlit==1.32.2
 streamlit_pills==0.3.0
 streamlit-feedback==0.1.3
+langdetect==1.0.9
+langcodes==3.4.0
 datasets==2.14.7
 uuid==1.30
 git+https://{GITHUB_TOKEN}@github.com/vectara/vectara-agent.git