Update app.py

app.py

@@ -23,6 +23,22 @@ summary.append("")
 main_point.append("")
 list_of_users=["user1","user2","user3"]
 
+persona=[
+    {"name":"Mr. Nice Guy", "description":"Nice","personality":"friendly, caring, helpful and informative.  You always compliment people, and stick up for them, and you have no patience for bullies."},
+    {"name":"Mr. Mean Guy", "description":"Mean","personality":"a total asshole.  You think you are really smart, but really you are just ignorant and mean.  You don't have time for everybodies stupidity, and you let them know that in the comments."},
+    {"name":"Smarty Pants", "description":"Genius","personality":"intelligent, informative, know-it-all.  You are the smartest guy in the room and always one-up the blog poster to show how mart you are."},
+    {"name":"Try Hard", "description":"Not Genius","personality":"dimwitted, lacking understanding about any topic.  You always ask really irrelevant questions about the post."},
+    {"name":"Class Clown", "description":"Humerous","personality":"humerous, funny.  You turn everything into a joke.  Make a joke about the post."},
+]
+persona_names=[]
+for ea in persona:
+    persona_names.append(ea['name'])
+
+
+
+
+
+
 models=[
     "mistralai/Mixtral-8x7B-Instruct-v0.1",
     "mistralai/Mixtral-8x7B-Instruct-v0.2",
@@ -116,7 +132,7 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
     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],persona=persona2)
+    agent=prompts.COMMENTER.format(focus=full_conv[0][0],persona=persona2['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -163,7 +179,7 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
             lod = json.loads(r.text)
             print(f'lod:: {lod}')
             lod[0]['comment']=lod[0]['comment']+1
-            lod[0]['comment_list'].append({'user':list_of_users[1],'datetime':'','comment':output,'reply_list':[]})
+            lod[0]['comment_list'].append({'user':persona2['name'],'datetime':'','comment':output,'reply_list':[]})
             #hist_out.append(out_json)
             #try:
             #    for ea in 
@@ -193,7 +209,7 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
     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],persona=persona1)
+    agent=prompts.REPLY_TO_COMMENTER.format(focus=full_conv[0][0],comment=full_conv[1][1],persona=persona1['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -237,7 +253,7 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
             lod = json.loads(r.text)
             print(f'lod:: {lod}')
             lod[0]['reply']=lod[0]['reply']+1
-            lod[0]['comment_list'][0]['reply_list'].append({'user':list_of_users[1],'datetime':'','reply':output})
+            lod[0]['comment_list'][0]['reply_list'].append({'user':persona1['name'],'datetime':'','reply':output})
             #hist_out.append(out_json)
             #try:
             #    for ea in 
@@ -375,7 +391,7 @@ def generate(prompt, history, post_check,full_conv,persona1, agent_name=agents[0
     current_time=current_time.replace(":","-")
     current_time=current_time.replace(".","-")
     print (current_time)
-    agent=prompts.BLOG_POSTER
+    agent=prompts.BLOG_POSTER.format(persona=persona1['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -480,8 +496,8 @@ def generate(prompt, history, post_check,full_conv,persona1, agent_name=agents[0
         full_conv.append((output,None,None))
     
     
-        html_out=load_html(full_conv,title,list_of_users[0])
-        post_check={'filename':filename,'user':list_of_users[0],'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
+        html_out=load_html(full_conv,title,persona1['name'])
+        post_check={'filename':filename,'user':persona1['name'],'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
         yield prompt, history,post_check,full_conv,summary[0],json_obj,json_hist,html_out
     else:
         print("passing blog")
@@ -620,16 +636,7 @@ def generate_OG(prompt, history, agent_name=agents[0], sys_prompt="", temperatur
     return prompt, history, summary[0],json_obj,json_hist,html_out
 
 
-persona=[
-    {"name":"Mr. Nice Guy", "description":"Nice","personality":"friendly, caring, helpful and informative.  You always compliment people, and stick up for them, and you have no patience for bullies."},
-    {"name":"Mr. Mean Guy", "description":"Mean","personality":"a total asshole.  You think you are really smart, but really you are just ignorant and mean.  You don't have time for everybodies stupidity, and you let them know that in the comments."},
-    {"name":"Smarty Pants", "description":"Genius","personality":"intelligent, informative, know-it-all.  You are the smartest guy in the room and always one-up the blog poster to show how mart you are."},
-    {"name":"Try Hard", "description":"Not Genius","personality":"dimwitted, lacking understanding about any topic.  You always ask really irrelevant questions about the post."},
-    {"name":"Class Clown", "description":"Humerous","personality":"humerous, funny.  You turn everything into a joke.  Make a joke about the post."},
-]
-persona_names=[]
-for ea in persona:
-    persona_names.append(ea['name'])
+
 
 with gr.Blocks() as app: