Update app.py

app.py

@@ -132,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['personality'])
+    agent=prompts.COMMENTER.format(focus=full_conv[0][0],persona=persona[persona2]['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -179,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':persona2['name'],'datetime':'','comment':output,'reply_list':[]})
+            lod[0]['comment_list'].append({'user':persona[persona2]['name'],'datetime':'','comment':output,'reply_list':[]})
             #hist_out.append(out_json)
             #try:
             #    for ea in 
@@ -209,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['personality'])
+    agent=prompts.REPLY_TO_COMMENTER.format(focus=full_conv[0][0],comment=full_conv[1][1],persona=persona[persona1]['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -253,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':persona1['name'],'datetime':'','reply':output})
+            lod[0]['comment_list'][0]['reply_list'].append({'user':persona[persona1]['name'],'datetime':'','reply':output})
             #hist_out.append(out_json)
             #try:
             #    for ea in 
@@ -391,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.format(persona=persona1['personality'])
+    agent=prompts.BLOG_POSTER.format(persona=persona[persona1]['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -496,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,persona1['name'])
-        post_check={'filename':filename,'user':persona1['name'],'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
+        html_out=load_html(full_conv,title,persona[persona1]['name'])
+        post_check={'filename':filename,'user':persona[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")