Update app.py

app.py

@@ -59,14 +59,21 @@ if uploaded_file is not None:
         st.session_state.answer = answer
 
     # Language selection and translation
-    translation_language = st.selectbox("Select translation language:", ["es", "fr", "de", "hi"])  # Add more languages as needed
-
-    if st.button("Translate Answer"):
-        translated_answer = GoogleTranslator(source='auto', target=translation_language).translate(st.session_state.answer)
-        st.session_state.translated_answer = translated_answer
-
-    if st.session_state.translated_answer:
-        st.write("Translated Answer:", st.session_state.translated_answer)
+    translation_language = st.text_input("Enter translation language:")  # Corrected to use st.text_input
+
+if st.button("Translate Answer"):
+    if translation_language:  # Check if a language was entered
+        try:
+            translated_answer = GoogleTranslator(source='auto', target=translation_language).translate(st.session_state.answer)
+            st.session_state.translated_answer = translated_answer
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+    else:
+        st.error("Please enter a valid language.")
+
+# Display translated answer if available
+if st.session_state.get('translated_answer'):
+    st.write("Translated Answer:", st.session_state.translated_answer)
 
     if st.button("Generate PDF"):
         # Generate PDF