Implement emotions

app.py

@@ -7,17 +7,22 @@ from langchain.chains import LLMChain
 import datetime
 
 NUM_WORDS_DEFAULT = 20
+FORMALITY_DEFAULT = "Casual"
+TEMPERATURE_DEFAULT = 0.5
+EMOTION_DEFAULT = "N/A"
 PROMPT_TEMPLATE = PromptTemplate(
-    input_variables=["original_words", "num_words"],
-    template="Restate, into an HTML div, in about {num_words} words, the following, with emotions of admiration, "
-             "in a formal manner, in a passive voice: \n{original_words}\n",
+    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",
 )
 
+
 # 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)
         openai_api_key = api_key
 
         os.environ["OPENAI_API_KEY"] = api_key
@@ -28,29 +33,38 @@ def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
         return openai_api_key, llm_chain
 
 
-def openai_create(prompt, openai_api_key, temperature):
-    print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
-    print("prompt: " + prompt)
-    print("temperature: ", temperature)
-
-    openai.api_key = openai_api_key
-    response = openai.Completion.create(
-        model="text-davinci-003",
-        prompt=prompt,
-        temperature=temperature,
-        max_tokens=2000,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0
-    )
-    return response.choices[0].text
-
-
-def desc2sheet(desc, openai_api_key, temperature, llm_chain, num_words):
+def desc2sheet(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):
     if not openai_api_key or openai_api_key == "":
         return "<pre>Please paste your OpenAI API key (see https://beta.openai.com)</pre>"
 
-    html = llm_chain.run({'original_words': desc, 'num_words': num_words})
+    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
 
 
@@ -66,38 +80,132 @@ def update_num_words(slider, state):
         return state
 
 
+def update_formality(radio, state):
+    if radio:
+        state = radio
+        return state
+
+
+def update_foo(widget, state):
+    if widget:
+        state = widget
+        return state
+
+
 block = gr.Blocks(css=".gradio-container {background-color: lightgray}")
 
 with block:
     openai_api_key_state = gr.State()
-    temperature_state = gr.State(0.0)
+    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)
+    anticipation_level_state = gr.State(EMOTION_DEFAULT)
+    joy_level_state = gr.State(EMOTION_DEFAULT)
+    trust_level_state = gr.State(EMOTION_DEFAULT)
+    fear_level_state = gr.State(EMOTION_DEFAULT)
+    surprise_level_state = gr.State(EMOTION_DEFAULT)
+    sadness_level_state = gr.State(EMOTION_DEFAULT)
+    disgust_level_state = gr.State(EMOTION_DEFAULT)
+    anger_level_state = gr.State(EMOTION_DEFAULT)
 
     with gr.Row():
-        temperature_slider = gr.Slider(label="GPT Temperature", value=0.0, minimum=0.0, maximum=1.0, step=0.1)
-        title = gr.Markdown("""<h3><center>GPT-3.5 Table-inator</center></h3>""")
+        temperature_slider = gr.Slider(label="GPT Temperature", value=TEMPERATURE_DEFAULT, minimum=0.0, maximum=1.0,
+                                       step=0.1)
+        title = gr.Markdown("""<h3><center>GPT-3.5 Express-inator</center></h3>""")
         openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key",
                                             show_label=False, lines=1, type='password')
 
     table = gr.Markdown("")
     with gr.Row():
         request = gr.Textbox(label="GPT prompt: " + PROMPT_TEMPLATE.template_format,
-                             value="Paris in the spring",
-                             placeholder="Ex: Computer languages")
+                             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.change(update_num_words,
+                            inputs=[num_words_slider, num_words_state],
+                            outputs=[num_words_state])
+
+    formality_radio = gr.Radio(label="Formality", choices=["Casual", "Polite", "Honorific"], value=FORMALITY_DEFAULT)
+    formality_radio.change(update_formality,
+                           inputs=[formality_radio, formality_state],
+                           outputs=[formality_state])
+
+    with gr.Accordion("Emotions", open=False):
+        anticipation_level_radio = gr.Radio(label="Anticipation level",
+                                            choices=[EMOTION_DEFAULT, "Interest", "Anticipation", "Vigilance"],
+                                            value=EMOTION_DEFAULT)
+        anticipation_level_radio.change(update_foo,
+                                        inputs=[anticipation_level_radio, anticipation_level_state],
+                                        outputs=[anticipation_level_state])
+
+        joy_level_radio = gr.Radio(label="Joy level",
+                                   choices=[EMOTION_DEFAULT, "Serenity", "Joy", "Ecstasy"],
+                                   value=EMOTION_DEFAULT)
+        joy_level_radio.change(update_foo,
+                               inputs=[joy_level_radio, joy_level_state],
+                               outputs=[joy_level_state])
+
+        trust_level_radio = gr.Radio(label="Trust level",
+                                   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])
+
+        fear_level_radio = gr.Radio(label="Fear level",
+                                      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])
+
+        surprise_level_radio = gr.Radio(label="Surprise level",
+                                        choices=[EMOTION_DEFAULT, "Distraction", "Surprise", "Amazement"],
+                                        value=EMOTION_DEFAULT)
+        surprise_level_radio.change(update_foo,
+                                    inputs=[surprise_level_radio, surprise_level_state],
+                                    outputs=[surprise_level_state])
+
+        sadness_level_radio = gr.Radio(label="Sadness level",
+                                        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])
+
+        disgust_level_radio = gr.Radio(label="Disgust level",
+                                        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])
+
+        anger_level_radio = gr.Radio(label="Anger level",
+                                        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])
+
 
     submit.click(desc2sheet,
                  inputs=[
-                     request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state],
+                     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],
                  outputs=[table])
 
     request.submit(desc2sheet,
                    inputs=[
-                     request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state],
+                       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],
                    outputs=[table])
 
     openai_api_key_textbox.change(set_openai_api_key,
@@ -109,9 +217,4 @@ with block:
                               inputs=[temperature_slider, temperature_state],
                               outputs=[temperature_state])
 
-    num_words_slider.change(update_num_words,
-                            inputs=[num_words_slider, num_words_state],
-                            outputs=[num_words_state])
-
-
 block.launch()