Update app.py

app.py

@@ -5,10 +5,20 @@ from typing import Generator
 from groq import Groq
 import datetime
 import json
+from googletrans import Translator
 
 _ = load_dotenv(find_dotenv())
 st.set_page_config(page_icon="💬", layout="wide", page_title="...")
 
+# Language options
+languages = {
+    "English": "en",
+    "Spanish": "es",
+    # Add more languages as needed
+}
+
+# Default language
+default_language = "English"
 
 def icon(emoji: str):
     """Shows an emoji as a Notion-style page icon."""
@@ -22,6 +32,39 @@ icon("⚡")
 
 st.subheader("Chatbot", divider="rainbow", anchor=False)
 
+# Language selection dropdown
+language_option = st.selectbox(
+    "Select Language:",
+    options=list(languages.keys()),
+    index=list(languages.keys()).index(default_language),
+)
+
+# Function to get language code
+def get_language_code(language):
+    return languages.get(language, "en")
+
+translator = Translator()
+
+# Translate function
+def translate_text(text, target_lang):
+    translation = translator.translate(text, dest=target_lang)
+    return translation.text
+
+# Translate all text on the page
+def translate_page_content(language):
+    elements = st.container._roots
+    for element in elements:
+        if hasattr(element, "body"):
+            element.body = translate_text(element.body, get_language_code(language))
+    st.experimental_rerun()
+
+# Translate the page content initially
+translate_page_content(default_language)
+
+# Listen for language changes and re-translate the page content
+if language_option != default_language:
+    translate_page_content(language_option)
+
 client = Groq(
     api_key=os.environ['GROQ_API_KEY'],
 )