Update app.py

app.py

@@ -106,7 +106,7 @@ def compress_history(formatted_prompt):
     return output
 
 
-def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.3,):
+def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.3,):
 #def question_generate(prompt, history):
     uid=uuid.uuid4()
     print(post_check)
@@ -116,7 +116,7 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
     seed = random.randint(1,1111111111111111)
     #agent=prompts.COMMENTER.format(focus=main_point[0])
     #bp,_,_=full_conv[0]
-    agent=prompts.COMMENTER.format(focus=full_conv[0][0])
+    agent=prompts.COMMENTER.format(focus=full_conv[0][0],persona=persona2)
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -185,7 +185,7 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
 
 
 
-def reply_generate(prompt, history,post_check,full_conv, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
+def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
 #def question_generate(prompt, history):
     uid=uuid.uuid4()
     print(post_check)
@@ -193,7 +193,7 @@ def reply_generate(prompt, history,post_check,full_conv, agent_name=agents[0], s
     print(f'full_conv::\n{full_conv}')
     print("###############\nRUNNING REPLY GENERATOR\n###############\n")
     seed = random.randint(1,1111111111111111)
-    agent=prompts.REPLY_TO_COMMENTER.format(focus=full_conv[0][0],comment=full_conv[1][1])
+    agent=prompts.REPLY_TO_COMMENTER.format(focus=full_conv[0][0],comment=full_conv[1][1],persona=persona1)
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -363,7 +363,7 @@ def load_html_OG(inp,title):
 
 
 
-def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
+def generate(prompt, history, post_check,full_conv,persona1, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
     html_out=""
     #main_point[0]=prompt
     #print(datetime.datetime.now())
@@ -653,8 +653,8 @@ with gr.Blocks() as app:
     with gr.Row():
         #persona1=gr.Textbox(label="Bot 1 Persona",info="Your personallity can be be described as...")
         #persona2=gr.Textbox(label="Bot 2 Persona",info="Your personallity can be be described as...")
-        persona1=gr.Dropdown(label="Bot 1 Persona",choices=[p for p in persona_names])
-        persona2=gr.Dropdown(label="Bot 2 Persona",choices=[p for p in persona_names])
+        persona1=gr.Dropdown(label="Bot 1 Persona",value=persona_names[0],type='index',choices=[p for p in persona_names])
+        persona2=gr.Dropdown(label="Bot 2 Persona",value=persona_names[3],type='index',choices=[p for p in persona_names])
         
     sumbox=gr.Textbox("Summary", max_lines=100)
     with gr.Column():
@@ -666,9 +666,9 @@ with gr.Blocks() as app:
     #app.load(load_models,m_choice,[chatbot]).then(load_html,None,html)
     app.load(load_models,m_choice,[chatbot])
     
-    sub_b = submit_b.click(generate, [msg,chatbot,post_handler,chat_handler,chat_handler,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
-    sub_c = submit_c.click(comment_generate, [msg,chatbot,post_handler,chat_handler],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
-    sub_r = submit_r.click(reply_generate, [msg,chatbot,post_handler,chat_handler],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
+    sub_b = submit_b.click(generate, [msg,chatbot,post_handler,chat_handler,chat_handler,persona1,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
+    sub_c = submit_c.click(comment_generate, [msg,chatbot,post_handler,chat_handler,persona2],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
+    sub_r = submit_r.click(reply_generate, [msg,chatbot,post_handler,chat_handler,persona1],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_e = msg.submit(generate, [msg,chatbot,post_handler,chat_handler,chat_handler,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e,sub_c,sub_r])