Update app.py

app.py

@@ -144,25 +144,13 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
         do_sample=True,
         seed=seed,
     )
-    #history.append((prompt,""))
-    #formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
-    #[INST] {message} [/INST]
     formatted_prompt = f"[INST] {system_prompt}, {prompt} [/INST]"
     client=client_z[0]
-
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
-        
     output = ""
-
     for response in stream:
         output += response.token.text
     history.append((output,None))
-
-    #comment_cnt=post_check['comment']
-    #print(type(comment_cnt))
-    #post_check['comment']=comment_cnt+1
-
-
             
     reply_json= {'user':'','datetime':'','reply':''}
     
@@ -176,12 +164,6 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
                 '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':comment_out}    
     
-    
-    
-    
-    #out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
-    #full_conv[-1]+=(output,)
-    #full_conv.append((None,output,None))
     html_out=load_html(out_json)
 
     #out_json = {'user':list_of_users[0],'datetime':current_time,'file_name':filename,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output,'comment_list':[]}
@@ -261,16 +243,16 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
     #post_check['reply']=reply_cnt+1
 
     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]}
-    
+
+    #comment_out=post_check['comment_list']
+
+    post_check['comment_list'][0]['reply_list'].append(reply_json)
+    #print(comment_out)
+    #reply_out.append(reply_json)
+    #post_check['comment_list'][0]
     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)}  
-    #out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
-    #full_conv[-1]+=(output,)
-    #full_conv.append((None,output,None))
+    "prompt":post_check['prompt'],"output":post_check['output'],'comment_list':post_check['comment_list']}  
     html_out=load_html(out_json)
 
     #out_json = {'user':list_of_users[0],'datetime':current_time,'file_name':filename,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output,'comment_list':[]}
@@ -309,17 +291,15 @@ def create_valid_filename(invalid_filename: str) -> str:
 
 def load_html(conv):
     ht=""
-    comm=0
-    #for i,ea in enumerate(inp):
-        
-    #blog,comm,repl=ea
-    #print(f'outp:: {outp}')
-    #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>"""
+    ht+=f"""<pre class="bpost"><div class="bhead"><h2>{conv['title']}</h2><br><h5>{conv['user']}</h5><br>{conv['datetime']}</div>{conv['output']}</pre>"""
     if conv['comment_list']:
         for com in conv['comment_list']:
-            ht+=f"""<pre class="resp1"><div class="bhead"></div>{com}</pre>"""
+            ht+=f"""<pre class="resp1"><div class="bhead"><h5>{com['user']}</h5><br>{com['datetime']}</div>{com['comment']}</pre>"""
+            if com['reply_list']:
+                for repl in com['reply_list']:
+                    ht+=f"""<pre class="resp2"><div class="bhead"></div>{repl['reply']}</pre>"""
+        
         for i,ea in enumerate(conv['comment_list']):
             if conv['comment_list'][i]['reply_list']:
                 for repl in conv['comment_list'][i]['reply_list']: