formality and translation removal

app.py

@@ -7,16 +7,21 @@ from langchain.chains import LLMChain
 import datetime
 
 NUM_WORDS_DEFAULT = 0
-FORMALITY_DEFAULT = "Casual"
+# FORMALITY_DEFAULT = "Casual"
 TEMPERATURE_DEFAULT = 0.5
 EMOTION_DEFAULT = "N/A"
-TRANSLATE_TO_DEFAULT = "Don't translate"
+# TRANSLATE_TO_DEFAULT = "Don't translate"
 LITERARY_STYLE_DEFAULT = "Poetry"
 CHARACTER_DEFAULT = "The Old Lady Who Swallowed A Fly"
+# PROMPT_TEMPLATE = PromptTemplate(
+#     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",
+# )
 PROMPT_TEMPLATE = PromptTemplate(
-    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",
+    input_variables=["original_words", "num_words", "emotions", "literary_style", "character"],
+    template="Express {num_words}, "
+             "{emotions}{literary_style}{character} the following: \n{original_words}\n",
 )
 
 
@@ -33,10 +38,15 @@ def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
         return openai_api_key, 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, character):
+# 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, character):
+
+def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, 
+                   anticipation_level, joy_level, trust_level, fear_level, 
+                   surprise_level, sadness_level, disgust_level, anger_level, 
+                   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>"
 
@@ -45,7 +55,7 @@ def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, form
         num_words_prompt = "using up to " + str(num_words) + " words, "
 
     # Change some arguments to lower case
-    formality = formality.lower()
+    # formality = formality.lower()
     anticipation_level = anticipation_level.lower()
     joy_level = joy_level.lower()
     trust_level = trust_level.lower()
@@ -83,9 +93,9 @@ def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, form
         else:
             emotions_str = "with emotions of " + ", ".join(emotions[:-1]) + " and " + emotions[-1] + ", "
 
-    translate_to_str = ""
-    if translate_to != TRANSLATE_TO_DEFAULT:
-        translate_to_str = "translated to " + translate_to + ", "
+    # translate_to_str = ""
+    # if translate_to != TRANSLATE_TO_DEFAULT:
+    #     translate_to_str = "translated to " + translate_to + ", "
 
     literary_style_str = ""
     if literary_style != LITERARY_STYLE_DEFAULT:
@@ -109,19 +119,32 @@ def transform_text(desc, openai_api_key, temperature, llm_chain, num_words, form
         elif character == "Humpty-Dumpty":
              character_str = "as if written 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,
+    #     character=character_str,
+    # )
+
     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,
         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, 'character': character_str}).strip()
 
-    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, 'character': character_str}).strip()
+                           
+    generated_text = llm_chain.run({'original_words': desc, 'num_words': num_words_prompt,
+                                    'emotions': emotions_str, '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,7 +171,7 @@ with block:
     temperature_state = gr.State(TEMPERATURE_DEFAULT)
     llm_chain_state = gr.State()
     num_words_state = gr.State(NUM_WORDS_DEFAULT)
-    formality_state = gr.State(FORMALITY_DEFAULT)
+    # formality_state = gr.State(FORMALITY_DEFAULT)
     anticipation_level_state = gr.State(EMOTION_DEFAULT)
     joy_level_state = gr.State(EMOTION_DEFAULT)
     trust_level_state = gr.State(EMOTION_DEFAULT)
@@ -157,7 +180,7 @@ with block:
     sadness_level_state = gr.State(EMOTION_DEFAULT)
     disgust_level_state = gr.State(EMOTION_DEFAULT)
     anger_level_state = gr.State(EMOTION_DEFAULT)
-    translate_to_state = gr.State(TRANSLATE_TO_DEFAULT)
+    # translate_to_state = gr.State(TRANSLATE_TO_DEFAULT)
     literary_style_state = gr.State(LITERARY_STYLE_DEFAULT)
     character_state = gr.State(CHARACTER_DEFAULT)
 
@@ -184,15 +207,29 @@ with block:
                                     inputs=[num_words_slider, num_words_state],
                                     outputs=[num_words_state])
 
+            character_radio = gr.Radio(label="As if written by:", 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)
+            character_radio.change(update_foo,
+                                   inputs=[character_radio, character_state],
+                                   outputs=[character_state])
+
+            literary_style_radio = gr.Radio(label="Literary style:", choices=[
+                LITERARY_STYLE_DEFAULT, "Poetry", "Haiku", "Limerick", "Joke", "Knock-knock"],
+                                            value=LITERARY_STYLE_DEFAULT)
+
+            literary_style_radio.change(update_foo,
+                                        inputs=[literary_style_radio, literary_style_state],
+                                        outputs=[literary_style_state])
+
             # formality_radio = gr.Radio(label="Formality:", choices=["Casual", "Polite", "Honorific"],
             #                            value=FORMALITY_DEFAULT)
+            
             # formality_radio.change(update_foo,
             #                        inputs=[formality_radio, formality_state],
             #                        outputs=[formality_state])
 
             # translate_to_radio = gr.Radio(label="Translate to:", choices=[TRANSLATE_TO_DEFAULT, "emojis", "Gen Z slang", "how the stereotypical Karen would say it", "Pirate"], value=TRANSLATE_TO_DEFAULT)
 
-            
             # 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",
@@ -201,25 +238,10 @@ with block:
             #     "Klingon", "Korean", "Norwegian", "Old English", "Pirate", "Polish", "Portuguese", "Romanian",
             #     "Russian", "Spanish", "Strange Planet expospeak technical talk", "Swedish", "Thai", "Turkish", "Vietnamese", "Yoda"], value=TRANSLATE_TO_DEFAULT)
 
-            character_radio = gr.Radio(label="As if written by:", 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)
-            character_radio.change(update_foo,
-                                   inputs=[character_radio, character_state],
-                                   outputs=[character_state])
-            
             # translate_to_radio.change(update_foo,
             #                           inputs=[translate_to_radio, translate_to_state],
             #                           outputs=[translate_to_state])
-
-            literary_style_radio = gr.Radio(label="Literary style:", choices=[
-                LITERARY_STYLE_DEFAULT, "Poetry", "Haiku", "Limerick", "Joke", "Knock-knock"],
-                                            value=LITERARY_STYLE_DEFAULT)
-
-            literary_style_radio.change(update_foo,
-                                        inputs=[literary_style_radio, literary_style_state],
-                                        outputs=[literary_style_state])
-
-        
+                    
             gr.Examples(
                 examples=[
                     "My armpits are sweating gravy",
@@ -293,23 +315,38 @@ with block:
 
     gr.HTML("<center>Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗</a></center>")
 
+    # submit.click(transform_text,
+    #              inputs=[
+    #                  request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state,
+    #                  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, character_state],
+    #              outputs=[output])
+
     submit.click(transform_text,
                  inputs=[
                      request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state,
-                     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, character_state],
+                     surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state, literary_style_state, character_state],
                  outputs=[output])
 
+    # request.submit(transform_text,
+    #                inputs=[
+    #                    request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state,
+    #                    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, character_state],
+    #                outputs=[output])
+
     request.submit(transform_text,
                    inputs=[
                        request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state,
-                       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, character_state],
-                   outputs=[output])
+                       literary_style_state, character_state],
+                   outputs=[output])    
 
     openai_api_key_textbox.change(set_openai_api_key,
                                   inputs=[openai_api_key_textbox, openai_api_key_state, temperature_state,