literary_style move

app.py

@@ -20,7 +20,7 @@ CHARACTER_DEFAULT = "The Old Lady Who Swallowed A Fly"
 # )
 PROMPT_TEMPLATE = PromptTemplate(
     input_variables=["num_words", "literary_style", "emotions", "character", "original_words"],
-    template="{num_words}{literary_style}{emotions}{character} express the following: \n{original_words}\n",
+    template="{num_words}{emotions}{character} express the following {literary_style}: \n{original_words}\n",
 )
 
 def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
@@ -95,16 +95,16 @@ def transform_text(desc, openai_api_key, temperature, llm_chain, num_words,
     # if translate_to != TRANSLATE_TO_DEFAULT:
     #     translate_to_str = "translated to " + translate_to + ", "
 
-    literary_style_str = "as a poem, "
+    literary_style_str = "as a poem"
     if literary_style != LITERARY_STYLE_DEFAULT:
         if literary_style == "Haiku":
-            literary_style_str = "as a haiku, "
+            literary_style_str = "as a haiku"
         elif literary_style == "Limerick":
-            literary_style_str = "as a limerick, "
+            literary_style_str = "as a limerick"
         # elif literary_style == "Joke":
         #     literary_style_str = "as a very funny joke with a setup and punchline, "
         elif literary_style == "Knock-knock":
-            literary_style_str = "as a very funny knock-knock joke, "
+            literary_style_str = "as a very funny knock-knock joke"
 
     character_str = "as if written by The Old Lady Who Swallowed A Fly, "
     if character != CHARACTER_DEFAULT: