Random sentence for serbian

app.py

@@ -10,10 +10,6 @@ from googletrans import Translator
 import string
 import re
 
-# import re
-# from nltk.tokenize import word_tokenize
-
-
 def random_sentence(lang):
     gen = DocumentGenerator()
     translator = Translator()
@@ -109,19 +105,31 @@ with tab4:
                        "Bosna je najbolja zemlja na svijetu"]
 
     
-    selected_example_sr = st.selectbox('Choose an example as demo', ['None']+serbian_examples)
+    selected_example_sr = st.selectbox('Choose an example as demo', ['None',"Generate a random sentence (Serbian)", "Generate a random sentence (Bosnian)", "Generate a random sentence (Croat)"]+serbian_examples)
 
     if selected_example_sr != 'None':
         input_string_serbian = selected_example_sr
-        
+
+    if selected_example_sr == "Generate a random sentence (Serbian)":
+        input_string_serbian = random_sentence("sr")
+
+    if selected_example_sr == "Generate a random sentence (Bosnian)":
+        input_string_serbian = random_sentence("bos")
+
+    if selected_example_sr == "Generate a random sentence (Croat)":
+        input_string_serbian = random_sentence("hr")
+    
     if st.button("Transliterate Serbo-Croatian-Bosnian"):
         if input_string_serbian:
             output_string = serbian_sentence_to_latin(input_string_serbian)
-            st.subheader("Transliterated Output")
-            st.write(output_string)
-            
+            st.subheader("Transliterated Output:")
+            if selected_example_sr == "Generate a random sentence" :
+                st.write(input_string_serbian)
+                st.write(output_string)
+            else:
+                st.write(output_string)
         else:
-            st.warning("Please enter a string")
+            st.warning("Please enter a string.")
 
 
 with tab5: