Character radio add

app.py

@@ -12,10 +12,11 @@ TEMPERATURE_DEFAULT = 0.5
 EMOTION_DEFAULT = "N/A"
 TRANSLATE_TO_DEFAULT = "Don't translate"
 LITERARY_STYLE_DEFAULT = "Prose"
+CHARACTER_DEFAULT = "The Old Lady Who Swallowed A Fly"
 PROMPT_TEMPLATE = PromptTemplate(
-    input_variables=["original_words", "num_words", "formality", "emotions", "translate_to", "literary_style"],
-    template="Express {num_words}in a {formality} manner, "
-             "{emotions}{translate_to}{literary_style}the following: \n{original_words}\n",
+    input_variables=["original_words", "num_words", "formality", "emotions", "translate_to", "literary_style", "character"],
+    template="Express {num_words} in a {formality} manner, "
+             "{emotions}{translate_to}{literary_style}{character} the following: \n{original_words}\n",
 )
 
 
@@ -35,7 +36,7 @@ def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
 def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, formality,
                    anticipation_level, joy_level, trust_level,
                    fear_level, surprise_level, sadness_level, disgust_level, anger_level,
-                   translate_to, literary_style):
+                       translate_to, literary_style, character):
     if not openai_api_key or openai_api_key == "":
         return "<pre>Please paste your OpenAI API key (see https://beta.openai.com)</pre>"
 
@@ -99,18 +100,28 @@ def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, form
         elif literary_style == "Knock-knock":
             literary_style_str = "as a very funny knock-knock joke, "
 
+    character_str = ""
+    if character != CHARACTER_DEFAULT:
+        if character == "The Old Woman Who Lived In A Shoe":
+             character_str = "by the Old Woman Who Lived In A Shoe, "
+        elif character == "Mary Had A Little Lamb":
+             character_str = "by Mary from Mary Had A Little Lamb, "
+        elif character == "Humpty-Dumpty":
+             character_str = "by Humpty Dumpty, "
+    
     formatted_prompt = PROMPT_TEMPLATE.format(
         original_words=desc,
         num_words=num_words_prompt,
         formality=formality,
         emotions=emotions_str,
         translate_to=translate_to_str,
-        literary_style=literary_style_str
+        literary_style=literary_style_str,
+        character=character_str,
     )
 
     generated_text = llm_chain.run({'original_words': desc, 'num_words': num_words_prompt, 'formality': formality,
                                     'emotions': emotions_str, 'translate_to': translate_to_str,
-                                    'literary_style': literary_style_str}).strip()
+                                    'literary_style': literary_style_str, 'character': character_str}).strip()
 
     # replace all newlines with <br> in generated_text
     generated_text = generated_text.replace("\n", "<br>")
@@ -148,6 +159,7 @@ with block:
     anger_level_state = gr.State(EMOTION_DEFAULT)
     translate_to_state = gr.State(TRANSLATE_TO_DEFAULT)
     literary_style_state = gr.State(LITERARY_STYLE_DEFAULT)
+    character_state = gr.State(CHARACTER_DEFAULT)
 
     with gr.Row():
         temperature_slider = gr.Slider(label="GPT Temperature", value=TEMPERATURE_DEFAULT, minimum=0.0, maximum=1.0,
@@ -178,6 +190,12 @@ with block:
                                    inputs=[formality_radio, formality_state],
                                    outputs=[formality_state])
 
+            character_radio = gr.Radio(label="From the perspective of:", choices=["The Old Lady Who Swallowed A Fly", "The Old Woman Who Lived In A Shoe", "Mary Had A Little Lamb", "Humpty-Dumpty"],
+                                       value=CHARACTER_DEFAULT)
+            formality_radio.change(update_foo,
+                                   inputs=[character_radio, character_state],
+                                   outputs=[character_state])
+
             translate_to_radio = gr.Radio(label="Translate to:", choices=[
                 TRANSLATE_TO_DEFAULT, "Arabic", "British English", "Chinese (Simplified)", "Chinese (Traditional)",
                 "Czech", "Danish", "Dutch", "emojis", "English", "Finnish", "French", "Gen Z slang", "German", "Greek",
@@ -277,7 +295,7 @@ with block:
                      formality_state,
                      anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                      surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
-                     translate_to_state, literary_style_state],
+                     translate_to_state, literary_style_state, character_state],
                  outputs=[output])
 
     request.submit(transform_text,
@@ -286,7 +304,7 @@ with block:
                        formality_state,
                        anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                        surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
-                       translate_to_state, literary_style_state],
+                       translate_to_state, literary_style_state, character_state],
                    outputs=[output])
 
     openai_api_key_textbox.change(set_openai_api_key,