Implement lots of state

app.py

@@ -1,14 +1,31 @@
+import os
 import openai
 import gradio as gr
+from langchain import OpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
 import datetime
 
-initial_prompt = "List the following in a HTML table with an appropriate number of rows in an appropriate order, giving the table an appropriate caption. Each non-numeric cell in the table, regardless of column, contains a link whenever possible to an appropriate Wikipedia page that opens in a new tab. No other links besides Wikipedia pages are permitted. The table contains around 5 columns with appropriate attributes: "
+NUM_WORDS_DEFAULT = 20
+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",
+)
 
+# 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):
+
+def set_openai_api_key(api_key, openai_api_key, temperature, llm_chain):
     if api_key:
         openai_api_key = api_key
-        return openai_api_key
+
+        os.environ["OPENAI_API_KEY"] = api_key
+        llm = OpenAI(model_name='text-davinci-003', temperature=temperature, max_tokens=512)
+        os.environ["OPENAI_API_KEY"] = ""
+
+        llm_chain = LLMChain(llm=llm, prompt=PROMPT_TEMPLATE, verbose=True)
+        return openai_api_key, llm_chain
 
 
 def openai_create(prompt, openai_api_key, temperature):
@@ -29,11 +46,11 @@ def openai_create(prompt, openai_api_key, temperature):
     return response.choices[0].text
 
 
-def desc2sheet(desc, openai_api_key, temperature):
+def desc2sheet(desc, openai_api_key, temperature, llm_chain, num_words):
     if not openai_api_key or openai_api_key == "":
         return "<pre>Please paste your OpenAI API key (see https://beta.openai.com)</pre>"
 
-    html = openai_create(initial_prompt + desc + '\n', openai_api_key, temperature)
+    html = llm_chain.run({'original_words': desc, 'num_words': num_words})
     return html
 
 
@@ -42,9 +59,21 @@ def update_temperature(temp_slider, temp_state):
         temp_state = temp_slider
         return temp_state
 
+
+def update_num_words(slider, state):
+    if slider:
+        state = slider
+        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)
+    llm_chain_state = gr.State()
+    num_words_state = gr.State(NUM_WORDS_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>""")
@@ -53,23 +82,36 @@ with block:
 
     table = gr.Markdown("")
     with gr.Row():
-        request = gr.Textbox(label="GPT prompt: " + initial_prompt,
+        request = gr.Textbox(label="GPT prompt: " + PROMPT_TEMPLATE.template_format,
+                             value="Paris in the spring",
                              placeholder="Ex: Computer languages")
 
         submit = gr.Button(value="Create", variant="secondary").style(full_width=False)
 
-    openai_api_key_state = gr.State()
-    temperature_state = gr.State(0.0)
+    num_words_slider = gr.Slider(label="Number of words", value=NUM_WORDS_DEFAULT, minimum=10, maximum=100, step=10)
 
-    submit.click(desc2sheet, inputs=[request, openai_api_key_state, temperature_state], outputs=[table])
-    request.submit(desc2sheet, inputs=[request, openai_api_key_state, temperature_state], outputs=[table])
+    submit.click(desc2sheet,
+                 inputs=[
+                     request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state],
+                 outputs=[table])
+
+    request.submit(desc2sheet,
+                   inputs=[
+                     request, openai_api_key_state, temperature_state, llm_chain_state, num_words_state],
+                   outputs=[table])
 
     openai_api_key_textbox.change(set_openai_api_key,
-                                  inputs=[openai_api_key_textbox, openai_api_key_state],
-                                  outputs=[openai_api_key_state])
+                                  inputs=[openai_api_key_textbox, openai_api_key_state, temperature_state,
+                                          llm_chain_state],
+                                  outputs=[openai_api_key_state, llm_chain_state])
 
     temperature_slider.change(update_temperature,
                               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()

requirements.txt

@@ -1,2 +1,3 @@
 openai
 gradio
+langchain