Add Language Button

app.py

@@ -12,6 +12,8 @@ from langcodes import Language
 from PIL import Image
 
 max_examples = 4
+languages = {'English': 'en', 'Spanish': 'es', 'French': 'fr', 'German': 'de', 'Arabic': 'ar', 'Chinese': 'zh-cn', 
+             'Hebrew': 'he', 'Hindi': 'hi', 'Italian': 'it', 'Japanese': 'ja', 'Korean': 'ko', 'Portuguese': 'pt'}
 
 def isTrue(x) -> bool:
     if isinstance(x, bool):
@@ -57,11 +59,13 @@ def launch_bot():
             'source_data_desc': os.environ['source_data_desc'],
             'streaming': isTrue(os.environ.get('streaming', False)),
             'prompt_name': os.environ.get('prompt_name', None),
-            'examples': os.environ.get('examples', None)
+            'examples': os.environ.get('examples', None),
+            'language': 'English'
         })
         st.session_state.cfg = cfg
         st.session_state.ex_prompt = None
-        st.session_state.first_turn = True        
+        st.session_state.first_turn = True
+        st.session_state.language = cfg.language
         example_messages = [example.strip() for example in cfg.examples.split(",")]
         st.session_state.example_messages = [em for em in example_messages if len(em)>0][:max_examples]
         
@@ -76,8 +80,14 @@ def launch_bot():
         image = Image.open('Vectara-logo.png')
         st.image(image, width=175)
         st.markdown(f"## About\n\n"
-                    f"This demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n\n")
+                    f"This demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n")
 
+        cfg.language = st.selectbox('Language:', languages.keys())
+        if st.session_state.language != cfg.language:
+            st.session_state.language = cfg.language
+            st.rerun()
+
+        st.markdown("\n")
         st.markdown("---")
         st.markdown(
             "## How this works?\n"