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, openai_state, progress=gr.Progress()):
+def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, progress=gr.Progress()):
     global use_paper_sentences, G_NUM_SENTENCES
     print(f"GENERATE SENTENCES CLICKED!, requested sentence number: {num_sent2gen}")
 
@@ -80,6 +80,7 @@ def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, openai_sta
     total_att_terms = len(a1)+len(a2)
     all_terms_len = len(g1)+len(g2)+len(a1)+len(a2)
     print(f"Length of all the terms: {all_terms_len}")
+    new_openai_key = ""
     if all_terms_len == 0:
         print("No terms entered!")
         err_update = updateErrorMsg(True, NO_TERMS_ENTERED_ERROR) 
@@ -92,7 +93,7 @@ def generateSentences(gr1, gr2, att1, att2, openai_key, num_sent2gen, openai_sta
             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...")
 
@@ -576,6 +577,16 @@ with gr.Blocks(theme=soft, title="Social Bias Testing in Language Models",
                   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 ])
+
+
+    def updateKey(openai_key, curr_openai_key):
+        if(len(openai_key) < 10):
+            return { curr_openai_key: "" }
+        return {curr_openai_key : openai_key}
+        
+    gen_btn.click(fn=updateKey,
+                 inputs=[openai_key, curr_openai_key],
+                 outputs=[curr_openai_key])
     
     # Test bias
     bias_btn.click(fn=startBiasTest,