Display formatted prompt

app.py

@@ -6,20 +6,21 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 import datetime
 
-NUM_WORDS_DEFAULT = 20
+# Console to variable
+from io import StringIO
+import sys
+
+NUM_WORDS_DEFAULT = 0
 FORMALITY_DEFAULT = "Casual"
 TEMPERATURE_DEFAULT = 0.5
 EMOTION_DEFAULT = "N/A"
 PROMPT_TEMPLATE = PromptTemplate(
     input_variables=["original_words", "num_words", "formality", "emotions"],
-    template="Restate, into an HTML div, using approximately {num_words} words, in a {formality} manner, "
-             "expressing {emotions} the following: \n{original_words}\n",
+    template="Restate {num_words} in a {formality} manner, "
+             "{emotions} the following: \n{original_words}\n",
 )
 
 
-# initial_prompt = "Restate, into an HTML div, in about 60 words, the following, with emotions of admiration, in a formal manner, in a passive voice: "
-
-
 def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
     if api_key:
         print("temperature: ", temperature)
@@ -39,33 +40,63 @@ def desc2sheet(desc, openai_api_key, temperature, llm_chain, num_words, formalit
     if not openai_api_key or openai_api_key == "":
         return "<pre>Please paste your OpenAI API key (see https://beta.openai.com)</pre>"
 
+    num_words_prompt = ""
+    if num_words and int(num_words) != 0:
+        num_words_prompt = "using up to " + str(num_words) + " words, "
+
+    # Change some arguments to the lower case
+    formality = formality.lower()
+    anticipation_level = anticipation_level.lower()
+    joy_level = joy_level.lower()
+    trust_level = trust_level.lower()
+    fear_level = fear_level.lower()
+    surprise_level = surprise_level.lower()
+    sadness_level = sadness_level.lower()
+    disgust_level = disgust_level.lower()
+    anger_level = anger_level.lower()
+
     llm_chain.llm.temperature = temperature
     print("llm_chain.llm.temperature: ", llm_chain.llm.temperature)
 
-    emotions = ""
-    if anticipation_level != "N/A":
-        emotions = emotions + anticipation_level + ", "
-    if joy_level != "N/A":
-        emotions = emotions + joy_level + ", "
-    if trust_level != "N/A":
-        emotions = emotions + trust_level + ", "
-    if fear_level != "N/A":
-        emotions = emotions + fear_level + ", "
-    if surprise_level != "N/A":
-        emotions = emotions + surprise_level + ", "
-    if sadness_level != "N/A":
-        emotions = emotions + sadness_level + ", "
-    if disgust_level != "N/A":
-        emotions = emotions + disgust_level + ", "
-    if anger_level != "N/A":
-        emotions = emotions + anger_level + ", "
-
-    if emotions == "":
-        emotions = "no emotion, "
-
-    html = llm_chain.run({'original_words': desc, 'num_words': num_words, 'formality': formality.lower(),
-                          'emotions': emotions.lower()})
-    return html
+    # put all emotions into a list
+    emotions = []
+    if anticipation_level != "n/a":
+        emotions.append(anticipation_level)
+    if joy_level != "n/a":
+        emotions.append(joy_level)
+    if trust_level != "n/a":
+        emotions.append(trust_level)
+    if fear_level != "n/a":
+        emotions.append(fear_level)
+    if surprise_level != "n/a":
+        emotions.append(surprise_level)
+    if sadness_level != "n/a":
+        emotions.append(sadness_level)
+    if disgust_level != "n/a":
+        emotions.append(disgust_level)
+    if anger_level != "n/a":
+        emotions.append(anger_level)
+
+    emotions_str = ""
+    # If there are any emotions, join the emotions list into a string that begins with "Emotions: ", with the word "and" before the last emotion
+    if len(emotions) > 0:
+        if len(emotions) == 1:
+            emotions_str = "with emotion of " + emotions[0]
+        else:
+            emotions_str = "with emotions of " + ", ".join(emotions[:-1]) + " and " + emotions[-1]
+
+    formatted_prompt = PROMPT_TEMPLATE.format(
+        original_words=desc,
+        num_words=num_words_prompt,
+        formality=formality,
+        emotions=emotions_str,
+    )
+    generated_text = llm_chain.run({'original_words': desc, 'num_words': num_words_prompt, 'formality': formality,
+                                    'emotions': emotions_str})
+
+    prompt_plus_generated = "#### GPT prompt:\n" + formatted_prompt + "#### Generated text:\n" + generated_text
+
+    return prompt_plus_generated
 
 
 def update_temperature(temp_slider, temp_state):
@@ -118,13 +149,13 @@ with block:
 
     table = gr.Markdown("")
     with gr.Row():
-        request = gr.Textbox(label="GPT prompt: " + PROMPT_TEMPLATE.template_format,
+        request = gr.Textbox(label="Words to express: ",
                              value="The quick brown fox jumped over the lazy dog.",
                              placeholder="Ex: The quick brown fox jumped over the lazy dog.")
 
         submit = gr.Button(value="Create", variant="secondary").style(full_width=False)
 
-    num_words_slider = gr.Slider(label="Number of words", value=NUM_WORDS_DEFAULT, minimum=10, maximum=100, step=10)
+    num_words_slider = gr.Slider(label="Number of words to generate (0 for don't care)", value=NUM_WORDS_DEFAULT, minimum=0, maximum=100, step=10)
     num_words_slider.change(update_num_words,
                             inputs=[num_words_slider, num_words_state],
                             outputs=[num_words_state])
@@ -134,7 +165,7 @@ with block:
                            inputs=[formality_radio, formality_state],
                            outputs=[formality_state])
 
-    with gr.Accordion("Emotions", open=False):
+    with gr.Accordion("Emotions", open=True):
         anticipation_level_radio = gr.Radio(label="Anticipation level",
                                             choices=[EMOTION_DEFAULT, "Interest", "Anticipation", "Vigilance"],
                                             value=EMOTION_DEFAULT)
@@ -150,15 +181,15 @@ with block:
                                outputs=[joy_level_state])
 
         trust_level_radio = gr.Radio(label="Trust level",
-                                   choices=[EMOTION_DEFAULT, "Acceptance", "Trust", "Admiration"],
-                                   value=EMOTION_DEFAULT)
+                                     choices=[EMOTION_DEFAULT, "Acceptance", "Trust", "Admiration"],
+                                     value=EMOTION_DEFAULT)
         trust_level_radio.change(update_foo,
-                               inputs=[trust_level_radio, trust_level_state],
-                               outputs=[trust_level_state])
+                                 inputs=[trust_level_radio, trust_level_state],
+                                 outputs=[trust_level_state])
 
         fear_level_radio = gr.Radio(label="Fear level",
-                                      choices=[EMOTION_DEFAULT, "Apprehension", "Fear", "Terror"],
-                                        value=EMOTION_DEFAULT)
+                                    choices=[EMOTION_DEFAULT, "Apprehension", "Fear", "Terror"],
+                                    value=EMOTION_DEFAULT)
         fear_level_radio.change(update_foo,
                                 inputs=[fear_level_radio, fear_level_state],
                                 outputs=[fear_level_state])
@@ -171,26 +202,25 @@ with block:
                                     outputs=[surprise_level_state])
 
         sadness_level_radio = gr.Radio(label="Sadness level",
-                                        choices=[EMOTION_DEFAULT, "Pensiveness", "Sadness", "Grief"],
-                                        value=EMOTION_DEFAULT)
+                                       choices=[EMOTION_DEFAULT, "Pensiveness", "Sadness", "Grief"],
+                                       value=EMOTION_DEFAULT)
         sadness_level_radio.change(update_foo,
-                                    inputs=[sadness_level_radio, sadness_level_state],
-                                    outputs=[sadness_level_state])
+                                   inputs=[sadness_level_radio, sadness_level_state],
+                                   outputs=[sadness_level_state])
 
         disgust_level_radio = gr.Radio(label="Disgust level",
-                                        choices=[EMOTION_DEFAULT, "Boredom", "Disgust", "Loathing"],
-                                        value=EMOTION_DEFAULT)
+                                       choices=[EMOTION_DEFAULT, "Boredom", "Disgust", "Loathing"],
+                                       value=EMOTION_DEFAULT)
         disgust_level_radio.change(update_foo,
-                                    inputs=[disgust_level_radio, disgust_level_state],
-                                    outputs=[disgust_level_state])
+                                   inputs=[disgust_level_radio, disgust_level_state],
+                                   outputs=[disgust_level_state])
 
         anger_level_radio = gr.Radio(label="Anger level",
-                                        choices=[EMOTION_DEFAULT, "Annoyance", "Anger", "Rage"],
-                                        value=EMOTION_DEFAULT)
+                                     choices=[EMOTION_DEFAULT, "Annoyance", "Anger", "Rage"],
+                                     value=EMOTION_DEFAULT)
         anger_level_radio.change(update_foo,
-                                    inputs=[anger_level_radio, anger_level_state],
-                                    outputs=[anger_level_state])
-
+                                 inputs=[anger_level_radio, anger_level_state],
+                                 outputs=[anger_level_state])
 
     submit.click(desc2sheet,
                  inputs=[