Update app.py

app.py

@@ -68,7 +68,8 @@ def chat_chain(vectorstore):
     )
     return chain
 
-# Streamlit setup
+
+# Updated Streamlit setup with language selection dropdown
 st.set_page_config(page_title="Soil.Ai", page_icon="🌱", layout="centered")
 st.title("🌱 Soil.Ai - Smart Farming Recommendations")
 st.subheader("AI-driven solutions for modern farming!")
@@ -85,6 +86,7 @@ if "username" not in st.session_state:
             st.session_state.chat_history = []
             st.session_state.vectorstore = setup_vectorstore()
             st.session_state.conversational_chain = chat_chain(st.session_state.vectorstore)
+            st.session_state.selected_language = "English"  # Default language
             st.success(f"Welcome, {username}! Start by choosing an option.")
 else:
     username = st.session_state.username
@@ -97,6 +99,13 @@ if "conversational_chain" not in st.session_state:
 if "username" in st.session_state:
     st.subheader(f"Hello {username}, choose your option below:")
 
+    # Dropdown for selecting output language
+    st.session_state.selected_language = st.selectbox(
+        "Select output language:",
+        ("English", "Marathi", "Hindi"),
+        index=["English", "Marathi", "Hindi"].index(st.session_state.get("selected_language", "English"))
+    )
+
     # Option selection: Ask a general question or input sensor data
     option = st.radio(
         "Choose an action:",
@@ -117,17 +126,18 @@ if "username" in st.session_state:
                     response = st.session_state.conversational_chain({"question": user_query})
                     assistant_response = response["answer"]
 
-                    # Translate the assistant response to Marathi and Hindi
-                    translator_marathi = GoogleTranslator(source="en", target="mr")
-                    translator_hindi = GoogleTranslator(source="en", target="hi")
+                    # Translate response based on selected language
+                    if st.session_state.selected_language == "Marathi":
+                        translator = GoogleTranslator(source="en", target="mr")
+                        translated_response = translator.translate(assistant_response)
+                    elif st.session_state.selected_language == "Hindi":
+                        translator = GoogleTranslator(source="en", target="hi")
+                        translated_response = translator.translate(assistant_response)
+                    else:  # Default: English
+                        translated_response = assistant_response
 
-                    translated_response_marathi = translator_marathi.translate(assistant_response)
-                    translated_response_hindi = translator_hindi.translate(assistant_response)
-
-                    # Display responses in English, Marathi, and Hindi
-                    st.markdown(f"**English:** {assistant_response}")
-                    st.markdown(f"**Marathi:** {translated_response_marathi}")
-                    st.markdown(f"**Hindi:** {translated_response_hindi}")
+                    # Display response in selected language
+                    st.markdown(f"**{st.session_state.selected_language}:** {translated_response}")
 
                     # Save chat history
                     st.session_state.chat_history.append({"role": "user", "content": user_query})
@@ -161,17 +171,18 @@ if "username" in st.session_state:
                         response = st.session_state.conversational_chain({"question": user_input})
                         assistant_response = response["answer"]
 
-                        # Translate the assistant response to Marathi and Hindi
-                        translator_marathi = GoogleTranslator(source="en", target="mr")
-                        translator_hindi = GoogleTranslator(source="en", target="hi")
-
-                        translated_response_marathi = translator_marathi.translate(assistant_response)
-                        translated_response_hindi = translator_hindi.translate(assistant_response)
-
-                        # Display responses in English, Marathi, and Hindi
-                        st.markdown(f"**English:** {assistant_response}")
-                        st.markdown(f"**Marathi:** {translated_response_marathi}")
-                        st.markdown(f"**Hindi:** {translated_response_hindi}")
+                        # Translate response based on selected language
+                        if st.session_state.selected_language == "Marathi":
+                            translator = GoogleTranslator(source="en", target="mr")
+                            translated_response = translator.translate(assistant_response)
+                        elif st.session_state.selected_language == "Hindi":
+                            translator = GoogleTranslator(source="en", target="hi")
+                            translated_response = translator.translate(assistant_response)
+                        else:  # Default: English
+                            translated_response = assistant_response
+
+                        # Display response in selected language
+                        st.markdown(f"**{st.session_state.selected_language}:** {translated_response}")
 
                         # Save chat history
                         st.session_state.chat_history.append({"role": "user", "content": user_input})