0.55 Spreadsheet setting vars

app.py

@@ -41,28 +41,11 @@ challenger_models = [{"id": "NousResearch/Meta-Llama-3.1-8B-Instruct",
 challenger_model = choice(challenger_models)
 model_info.append(challenger_model)
 shuffle(model_info)
+chatbot_a_name = model_info[0]['name']
+chatbot_b_name = model_info[1]['name']
 
 device = "cuda" 
 
-def get_google_credentials():
-    service_account_info = {
-        "type": "service_account",
-        "project_id": os.environ.get("GOOGLE_PROJECT_ID"),
-        "private_key_id": os.environ.get("GOOGLE_PRIVATE_KEY_ID"),
-        "private_key": os.environ.get("GOOGLE_PRIVATE_KEY").replace('\\n', '\n'),
-        "client_email": os.environ.get("GOOGLE_CLIENT_EMAIL"),
-        "client_id": os.environ.get("GOOGLE_CLIENT_ID"),
-        "auth_uri": os.environ.get("GOOGLE_AUTH_URI"),
-        "token_uri": os.environ.get("GOOGLE_TOKEN_URI"),
-        "auth_provider_x509_cert_url": os.environ.get("GOOGLE_AUTH_PROVIDER_CERT_URL"),
-        "client_x509_cert_url": os.environ.get("GOOGLE_CLIENT_CERT_URL")
-    }
-
-    credentials = Credentials.from_service_account_info(service_account_info)
-    return credentials
-
-logging.debug(get_google_credentials())
-
 try: 
     tokenizer_a = AutoTokenizer.from_pretrained(model_info[0]['id'])
     model_a = AutoModelForCausalLM.from_pretrained(
@@ -84,6 +67,31 @@ except Exception as e:
     logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
 
 
+def get_google_credentials():
+    service_account_info = {
+        "type": "service_account",
+        "project_id": os.environ.get("GOOGLE_PROJECT_ID"),
+        "private_key_id": os.environ.get("GOOGLE_PRIVATE_KEY_ID"),
+        "private_key": os.environ.get("GOOGLE_PRIVATE_KEY").replace('\\n', '\n'),
+        "client_email": os.environ.get("GOOGLE_CLIENT_EMAIL"),
+        "client_id": os.environ.get("GOOGLE_CLIENT_ID"),
+        "auth_uri": os.environ.get("GOOGLE_AUTH_URI"),
+        "token_uri": os.environ.get("GOOGLE_TOKEN_URI"),
+        "auth_provider_x509_cert_url": os.environ.get("GOOGLE_AUTH_PROVIDER_CERT_URL"),
+        "client_x509_cert_url": os.environ.get("GOOGLE_CLIENT_CERT_URL")
+    }
+
+    credentials = Credentials.from_service_account_info(service_account_info)
+    return credentials
+
+
+def get_google_sheet():
+    credentials = get_google_credentials()
+    client = gspread.authorize(credentials)
+    sheet = client.open("pharia_bot_battle_logs").sheet1  # Open your Google Sheet
+    return sheet
+
+
 def apply_pharia_template(messages, add_generation_prompt=False):
     """Chat template not defined in Pharia model configs. 
     Adds chat template for Pharia. Expects a list of messages. 
@@ -109,7 +117,10 @@ def apply_pharia_template(messages, add_generation_prompt=False):
 
 
 @spaces.GPU()
-def generate_both(system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens=2048, temperature=0.2, top_p=0.9, repetition_penalty=1.1):
+def generate_both(system_prompt, input_text, 
+                    chatbot_a, chatbot_b, 
+                    max_new_tokens=2048, temperature=0.2, 
+                    top_p=0.9, repetition_penalty=1.1):
     try: 
         text_streamer_a = TextIteratorStreamer(tokenizer_a, skip_prompt=True)
         text_streamer_b = TextIteratorStreamer(tokenizer_b, skip_prompt=True)
@@ -229,21 +240,29 @@ def generate_both(system_prompt, input_text, chatbot_a, chatbot_b, max_new_token
             except Exception as e:
                 logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
 
+    logging.debug(f'Output complete chatbot_a: {chatbot_a} chatbot_b: {chatbot_b}')
     return chatbot_a, chatbot_b
 
 def clear():
     return [], []
 
-def reveal_bot(selection, chatbot_a, chatbot_b):
+def handle_vote(selection, chatbot_a, chatbot_b):
     if selection == "Bot A kicks ass!":
-        chatbot_a.append(["🏆", f"Thanks, man. I am {model_info[0]['name']}"])
-        chatbot_b.append(["💩", f"Pffff … I am {model_info[1]['name']}"])
+        chatbot_a.append(["🏆", f"Thanks, man. I am {chatbot_a_name}"])
+        chatbot_b.append(["💩", f"Pffff … I am {chatbot_b_name}"])
+        chatbot_a_vote = "winner"
+        chatbot_b_vote = "looser"
     elif selection == "Bot B crushes it!":
-        chatbot_a.append(["🤡", f"Rigged … I am {model_info[0]['name']}"])
-        chatbot_b.append(["🥇", f"Well deserved! I am {model_info[1]['name']}"])  
+        chatbot_a.append(["🤡", f"Rigged … I am {chatbot_a_name}"])
+        chatbot_b.append(["🥇", f"Well deserved! I am {chatbot_b_name}"])
+        chatbot_a_vote = "looser"
+        chatbot_b_vote = "winner"
     else:
-        chatbot_a.append(["🤝", f"Lame … I am {model_info[0]['name']}"])
-        chatbot_b.append(["🤝", f"Dunno. I am {model_info[1]['name']}"])            
+        chatbot_a.append(["🤝", f"Lame … I am {chatbot_a_name}"])
+        chatbot_b.append(["🤝", f"Dunno. I am {chatbot_b_name}"])
+        chatbot_a_vote = "draw"
+        chatbot_b_vote = "draw"
+    logging.debug(f'Casting vote: {chatbot_a_name}: {chatbot_a_vote}, {chatbot_b_name}: {chatbot_b_vote}')
     return chatbot_a, chatbot_b
 
 with gr.Blocks() as demo:
@@ -285,7 +304,7 @@ with gr.Blocks() as demo:
             outputs=[system_prompt]
         )
 
-        better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) 
+        better_bot.select(handle_vote, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) 
         input_text.submit(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
         submit_btn.click(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
         clear_btn.click(clear, outputs=[chatbot_a, chatbot_b])