Added radio button selection for different prompts

app.py

@@ -9,11 +9,16 @@ co = cohere.Client(os.getenv('COHERE_API_KEY'))
 def generate_hashtags(input):
     if len(input) == 0:
         return None
+    
+    if category == "positive_affirmation":
+        input_prompt = "This program will take a sentence as input and reframe it with positive and growth mindset.\n--\nInput: I have a lot of work to do today.\nOutput: I have a lot of work to do today. It's better for me to make a list and break it down into smaller chunks and finish them off one by one.\n--\nInput: I am barely able to lift 10 pound weights.\nOutput: I should exercise more consistently and gradually increase my weight limit. I should also try eating healthy foods.\n--\nInput: {}\nOutput:".format(input.strip())
+    if category == "remove_negations":
+        input_prompt = "This program will take a sentence as input and reframe it positively without negations..\n--\nInput: I wouldn’t say I don’t want to go.\nOutput: I would like to go.\n--\nInput: That’s not a bad idea.\nOutput: That is a good idea.\n--\nInput: {}\nOutput:".format(input.strip())
     response = co.generate( 
     model='xlarge', 
-    prompt="This program will take a sentence as input and reframe it with positive and growth mindset.\n\n--\nInput: I have a lot of work to do today.\nOutput: I have a lot of work to do today. It's better for me to make a list and break it down into smaller chunks and finish them off one by one.\n--\nInput: I am barely able to lift 10 pound weights.\nOutput: I should exercise more consistently and gradually increase my weight limit. I should also try eating healthy foods.\n--\nInput: {}\nOutput:".format(input.strip()), 
-    max_tokens=80, 
-    temperature=0.8, 
+    prompt=input_prompt, 
+    max_tokens=100, 
+    temperature=0.9, 
     k=0, 
     p=1, 
     frequency_penalty=0, 
@@ -28,6 +33,11 @@ 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)
+
+category = st.radio(
+    "Please select a category",
+    ('positive_affirmation', 'remove_negations'))
+
 if st.button('Reframe'):
     output = generate_hashtags(input)
     st.write(output)