Merge branch 'main' of https://huggingface.co/spaces/fortuna-hackathon/positive-reframing into main

app.py

@@ -2,9 +2,20 @@ import streamlit as st
 import cohere
 import os
 
+
+# 'growth', 'remove_negations', 'self_affirmation'
+EXAMPLES = {
+    "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', 'detect_rhetoric']
+
 co = cohere.Client(os.getenv('COHERE_API_KEY')) 
 
 # Initialization
+def fill_example_caption(**kwargs):
+    print(f"Called with {kwargs}")
+    print(f"selectbox {prompt}")
 
 def generate_hashtags(input):
     if len(input) == 0:
@@ -36,11 +47,25 @@ def generate_hashtags(input):
 st.title('Positive Reframing Generator')
 st.write('''This is a simple **Streamlit** app that generates the input sentence with a positive spin to it''')
 
-input = st.text_area('Enter your post title caption here', height=100)
+prompts = ["Choose"] + list(EXAMPLES.keys())
+prompt = st.selectbox(
+    'Examples (select from this list)',
+    prompts,
+    index=0,
+)
+if prompt == "Choose":
+    input_val = ""
+    input_cat = categories[0]
+else:
+    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', 'detect_rhetoric'))
+    categories,
+    categories.index(input_cat),
+    )
 
 if st.button('Reframe'):
     output = generate_hashtags(input)