Require user to enter OpenAI API key

app.py

@@ -1,3 +1,4 @@
+import os
 import openai
 import gradio as gr
 from langchain import OpenAI
@@ -7,6 +8,18 @@ import datetime
 gpt_only_prompt = "Calculate the following, giving only the final answer:\n"
 prompt = ""
 
+os.environ["OPENAI_API_KEY"] = ""
+
+
+def set_openai_api_key(api_key, pal_chain_sta):
+    if api_key != "":
+        os.environ["OPENAI_API_KEY"] = api_key
+
+        llm = OpenAI(model_name='code-davinci-002', temperature=0, max_tokens=512)
+        pal_chain = PALChain.from_math_prompt(llm, verbose=True)
+        return pal_chain
+
+
 def openai_create(prompt):
     print("prompt: " + prompt)
 
@@ -27,19 +40,19 @@ def openai_create(prompt):
 def calc_gpt_only(math_problem):
     answer = openai_create(gpt_only_prompt + math_problem + "\n")
 
-    print("math problem: " + math_problem)
+    print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
+    print("calc_gpt_only math problem: " + math_problem)
     print("calc_gpt_only answer: " + answer)
 
     html = "<pre>" + answer + "</pre>"
     return html
 
 
-def calc_gpt_pal(math_problem):
-    llm = OpenAI(model_name='code-davinci-002', temperature=0, max_tokens=512)
-    pal_chain = PALChain.from_math_prompt(llm, verbose=True)
+def calc_gpt_pal(math_problem, pal_chain):
     answer = pal_chain.run(math_problem)
 
-    print("math problem: " + math_problem)
+    print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
+    print("calc_gpt_pal math problem: " + math_problem)
     print("calc_gpt_pal answer: " + answer)
 
     html = "<pre>" + answer + "</pre>"
@@ -49,7 +62,12 @@ def calc_gpt_pal(math_problem):
 block = gr.Blocks(css=".gradio-container {background-color: lightgray}")
 
 with block:
-    title = gr.Markdown("""<h3><center>Comparing GPT math techniques</center></h3>""")
+    with gr.Row():
+        title = gr.Markdown("""<h3><center>Comparing GPT math techniques</center></h3>""")
+
+        openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key (sk-...)",
+                                            show_label=False, lines=1, type='password')
+
     answer_html = gr.Markdown()
 
     request = gr.Textbox(label="Math question:",
@@ -62,7 +80,11 @@ with block:
         examples=["42 times 81",
                   "Olivia has $23. She bought five bagels for $3 each. How much money does she have left?",
                   "What is the sum of the first 21 Fibonacci numbers?",
-                  "Jane quit her job on Mar 20, 2020. 176 days have passed since then. What is the date tomorrow in MM/DD/YYYY?"],
+                  "Jane quit her job on Mar 20, 2020. 176 days have passed since then. What is the date tomorrow in "
+                  "MM/DD/YYYY?",
+                  "If y = 8−5x+4x2, what is the value of y when x = −2?",
+                  "A line parallel to y = 4x + 6 passes through (5, 10). What is the y-coordinate of the point where "
+                  "this line crosses the y-axis?"],
         inputs=request
     )
 
@@ -76,7 +98,13 @@ with block:
 
     gr.HTML("<center>Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗</a></center>")
 
+    pal_chain_state = gr.State()
+
     gpt_only.click(calc_gpt_only, inputs=[request], outputs=[answer_html])
-    gpt_pal.click(calc_gpt_pal, inputs=[request], outputs=[answer_html])
+    gpt_pal.click(calc_gpt_pal, inputs=[request, pal_chain_state], outputs=[answer_html])
+
+    openai_api_key_textbox.change(set_openai_api_key,
+                                  inputs=[openai_api_key_textbox, pal_chain_state],
+                                  outputs=[pal_chain_state])
 
 block.launch()