Added rhetoric classifier

app.py

@@ -16,6 +16,8 @@ def generate_hashtags(input):
         input_prompt = "This program takes an input and reframes it by removing negative words and rephrasing with with positivity while preserving the meaning of the original sentence. The word 'no' doesn't appear in the output\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())
     if category == "self_affirmation":
         input_prompt = "This program takes an input and reframes it with positive affirmations.\n-- \nInput: I am worthless\nOutput: I have value and my actions impact the world around me.\n--\nInput: I am a failure\nOutput: I never fail, only my attempts do. I will keep trying and improving\n--\nInput:  I don’t like myself\nOutput: I accept myself and improve what I can.\n--\nInput: {}\nOutput:".format(input.strip())
+    if category == "detect_rhetoric":
+        input_prompt = "This program takes an input and tells if it is rhetoric or not.\n--\nInput: Are you kidding me?\nOutput: rhetoric\n--\nInput: How should I know?\nOutput: rhetoric\n--\nInput:  How much does this cost?\nOutput: not rhetoric\n--\nInput: Do you know the answer to this problem?\nOutput: not rhetoric\n--\nInput: Is this for real?\nOutput: rhetoric\n--\nInput: Can you please finish this task by monday?\nOutput: not rhetoric\n--\nInput:{}\nOutput:".format(input.strip())
     response = co.generate( 
     model='xlarge', 
     prompt=input_prompt, 
@@ -38,7 +40,7 @@ input = st.text_area('Enter your post title caption here', height=100)
 
 category = st.radio(
     "Please select a category",
-    ('growth', 'remove_negations', 'self_affirmation'))
+    ('growth', 'remove_negations', 'self_affirmation', 'detect_rhetoric'))
 
 if st.button('Reframe'):
     output = generate_hashtags(input)