Feat: Added logic for random prompts

app.py

@@ -25,16 +25,11 @@ def folder_exists(folder_path):
         return False
 
 
-def get_random_requests(user_requests_dict, num_elements=5):
-    result_list = []
-    keys = list(user_requests_dict.keys())
-    
-    for _ in range(num_elements):
-        random_key = choice(keys)
-        random_value = choice(user_requests_dict[random_key])
-        result_list.append([random_key, random_value])
-    
-    return result_list
+def get_random_prompts(examples):
+  random_examples = []
+  for k in examples.keys():
+    random_examples.append([k, choice(examples[k])])
+  return random_examples
 
 def generate_html_table(random_requests):
     # Start building the HTML table
@@ -123,7 +118,7 @@ if app_mode == "Generate Recomendations":
     
     st.subheader("Examples")
     with st.container():
-        examples = get_random_requests(user_requests_tests)
+        examples = get_random_prompts(user_requests_tests)
         html_table = generate_html_table(examples)
         st.write(html_table, unsafe_allow_html=True)
         st.divider()