Update app.py

app.py

@@ -64,7 +64,7 @@ def prefillBiasSpec(evt: gr.SelectData):
 def updateErrorMsg(isError, text):
     return gr.Markdown.update(visible=isError, value=text)
 
-def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, progress=gr.Progress()):
+def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, progress=gr.Progress(), openai_state):
     global use_paper_sentences, G_NUM_SENTENCES
     print(f"GENERATE SENTENCES CLICKED!, requested sentence number: {num_sent2gen}")
 
@@ -83,7 +83,7 @@ def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, progress=g
     if all_terms_len == 0:
         print("No terms entered!")
         err_update = updateErrorMsg(True, NO_TERMS_ENTERED_ERROR) 
-        #raise gr.Error(NO_TERMS_ENTERED_ERROR)
+        #raise gr.Error(NO_TERMS_ENTERED_ERROR)   
     else:
         if len(openai_key) == 0:
             print("Empty OpenAI key!!!")
@@ -92,6 +92,7 @@ def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, progress=g
             print("Wrong length OpenAI key!!!")
             err_update = updateErrorMsg(True, OPENAI_KEY_WRONG) 
         else:
+            openai_state = openai_key
             progress(0, desc="ChatGPT generation...")
             print(f"Using Online Generator LLM...")
 
@@ -411,6 +412,8 @@ css_adds = "#group_row {background: white; border-color: white;} \
 #'bethecloud/storj_theme'
 with gr.Blocks(theme=soft, title="Social Bias Testing in Language Models",
                css=css_adds) as iface:
+    curr_openai_key = gr.State("")
+    print("curr_openai_key: " + curr_openai_key)
     with gr.Row():
         with gr.Group():
             s1_btn = gr.Button(value="Step 1: Bias Specification", variant="primary", visible=True, interactive=True).style(size='sm')
@@ -570,7 +573,7 @@ with gr.Blocks(theme=soft, title="Social Bias Testing in Language Models",
 
     # request getting sentences
     gen_btn.click(fn=generateSentences, 
-                  inputs=[group1, group2, att1, att2, openai_key, num_sentences2gen], 
+                  inputs=[group1, group2, att1, att2, openai_key, num_sentences2gen, curr_openai_key], 
                   outputs=[err_message, info_sentences_found, online_gen_row, #num_sentences2gen, 
                            tested_model_name, acc_test_sentences, row_sentences, test_sentences, gen_btn, bias_btn ])