[Amir] having prompt and category filled on selection

app.py

@@ -3,10 +3,13 @@ import cohere
 import os
 
 
+# 'growth', 'remove_negations', 'self_affirmation'
 EXAMPLES = {
-    "I lost my Job": ("I lost my Job."),
-    "I lost my Dog": ("I lost my Dog."),
+    "I lost my Job": ("I lost my Job.", 'growth'),
+    "I lost my Dog": ("I lost my Dog.", 'self_affirmation'),
 }
+categories = ['growth', 'remove_negations', 'self_affirmation']
+
 co = cohere.Client(os.getenv('COHERE_API_KEY')) 
 
 # Initialization
@@ -50,13 +53,17 @@ prompt = st.selectbox(
 )
 if prompt == "Choose":
     input_val = ""
+    input_cat = categories[0]
 else:
-    input_val = EXAMPLES[prompt]
+    input_val = EXAMPLES[prompt][0]
+    input_cat = EXAMPLES[prompt][1]
 input = st.text_area('Enter your post title caption or select from examples', input_val, height=100)
 
 category = st.radio(
     "Please select a category",
-    ('growth', 'remove_negations', 'self_affirmation'))
+    categories,
+    categories.index(input_cat),
+    )
 
 if st.button('Reframe'):
     output = generate_hashtags(input)