Update app.py

app.py

@@ -17,6 +17,7 @@ SPREADSHEET_ID = '1o0iKPxWYKYKEPjqB2YwrTgrLzvGyb9ULj9tnw_cfJb0'
 service = build('sheets', 'v4', credentials=creds)
 LEFT_MODEL = None
 RIGHT_MODEL = None
+PROMPT = None
 
 with open("article_list.pkl","rb") as articles:
     article_list = tuple(pickle.load(articles))
@@ -44,9 +45,10 @@ def get_matchup():
     return left, right
 
 def get_comp(prompt):
+    global PROMPT
     left, right = get_matchup()
-    left_output = inference(prompt,left)
-    right_output = inference(prompt,right)
+    left_output = inference(PROMPT,left)
+    right_output = inference(PROMPT,right)
     return left_output, right_output
 
 def get_article():
@@ -54,6 +56,7 @@ def get_article():
 
 
 def send_result(l_output, r_output, prompt, pick):
+    global PROMPT
     global LEFT_MODEL, RIGHT_MODEL
     # with open('results.csv','a') as res_file:
     #   writer = csv.writer(res_file)
@@ -62,12 +65,13 @@ def send_result(l_output, r_output, prompt, pick):
         pick = LEFT_MODEL
     else:
         pick = RIGHT_MODEL
-    row = [USER_ID,prompt,LEFT_MODEL,RIGHT_MODEL,pick]
+    row = [USER_ID,PROMPT,LEFT_MODEL,RIGHT_MODEL,pick]
     row = [str(x) for x in row]
     body = {'values': [row]}
     result = service.spreadsheets().values().append(spreadsheetId=SPREADSHEET_ID, range='A1:E1', valueInputOption='RAW', body=body).execute()
     print(f"Appended {result['updates']['updatedCells']} cells.")
     new_prompt = get_article()
+    PROMPT = new_prompt
     return new_prompt,gr.State.update(value=new_prompt)
 
 
@@ -126,6 +130,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     article = get_article()
     models = gr.State(value=get_matchup())
     prompt = gr.State(value=article)
+    PROMPT = article
     abst = gr.Textbox(value = article, label="Abstract", lines=10)
     action_btn = gr.Button(value="Get comparison")
     with gr.Group():