Update app.py

app.py

@@ -162,19 +162,18 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
         output += response.token.text
     history.append((output,None))
 
-    comment_cnt=post_check['comment']
-    print(type(comment_cnt))
-    post_check['comment']=comment_cnt+1
+    #comment_cnt=post_check['comment']
+    #print(type(comment_cnt))
+    #post_check['comment']=comment_cnt+1
 
 
             
     reply_json= {'user':'','datetime':'','reply':''}
     
-    #comment_json= {'user':'','datetime':'','comment':'','reply_list':[reply_json]}
     
-    comment_json= {'user':persona[persona2]['name'],'datetime':current_time,'comment':output,'reply_list':[reply_json]}
+    comment_json= {'user':persona[persona2]['name'],'datetime':current_time,'comment':output,'reply_list':[]}
     
-    out_json = {'user':post_check['user'],'datetime':post_check['datetime'],'file_name':post_check['filename'],
+    out_json = {'user':post_check['user'],'datetime':post_check['datetime'],'file_name':post_check['file_name'],
                 'title':post_check['title'],'blog':1,'comment':post_check['comment']+1,'reply':post_check['reply'],
     "prompt":post_check['prompt'],"output":post_check['output'],'comment_list':post_check['comment_list'].append(comment_json)}    
     
@@ -259,14 +258,17 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
         output += response.token.text
     history.append((output,None))
 
-    reply_cnt=post_check['reply']
-    post_check['reply']=reply_cnt+1
+    #reply_cnt=post_check['reply']
+    #post_check['reply']=reply_cnt+1
 
-
-
-
-    out_json = {'user':persona[persona1]['name'],'datetime':post_check['datetime'],'file_name':post_check['filename'],
-                'title':post_check['title'],'blog':1,'comment':0,'reply':post_check['reply']+1,"prompt":post_check['prompt'],"output":post_check['output'],'comment_list':post_check['comment_list']}
+    reply_json= {'user':persona[persona1]['name'],'datetime':current_time,'reply':output}
+    
+    
+    #comment_json= {'user':persona[persona2]['name'],'datetime':current_time,'comment':output,'reply_list':[reply_json]}
+    
+    out_json = {'user':post_check['user'],'datetime':post_check['datetime'],'file_name':post_check['file_name'],
+                'title':post_check['title'],'blog':1,'comment':post_check['comment'],'reply':post_check['reply']+1,
+    "prompt":post_check['prompt'],"output":post_check['output'],'comment_list':post_check['comment_list']['reply_list'].append(reply_json)}  
 
     hist_out.append(out_json)
     #try:
@@ -349,10 +351,13 @@ def load_html(inp,conv):
     #print(f'prom:: {prom}')
     ht+=f"""<div class="div_box">"""
     ht+=f"""<pre class="bpost"><div class="bhead"><h2>{conv['title']}</h2><br><h5>{conv['user']}</h5></div>{conv['output']}</pre>"""
-    for com in conv['comment_list']:
-        ht+=f"""<pre class="resp1"><div class="bhead"></div>{com}</pre>"""
-    for repl in conv['comment_list'][comm]['reply_list']:
-        ht+=f"""<pre class="resp2"><div class="bhead"></div>{repl}</pre>"""
+    if conv['comment_list']:
+        for com in conv['comment_list']:
+            ht+=f"""<pre class="resp1"><div class="bhead"></div>{com}</pre>"""
+        for ea in conv['comment_list']:
+            if conv['comment_list'][ea]['reply_list']:
+                for repl in conv['comment_list'][ea]['reply_list']:
+                    ht+=f"""<pre class="resp2"><div class="bhead"></div>{repl}</pre>"""
     ht+=f"""</div>"""                
     '''
     if inp:
@@ -492,7 +497,7 @@ def generate(prompt, history, post_check,full_conv,persona1, agent_name=agents[0
             
             out_json = {'user':persona[persona1]['name'],'datetime':current_time,'file_name':filename,
                         'title':title,'blog':1,'comment':0,'reply':0,
-            "prompt":prompt,"output":output,'comment_list':[comment_json]}  
+            "prompt":prompt,"output":output,'comment_list':[]}  
 
             #hist_out.append(out_json)
             #try: