Update app.py

app.py

@@ -103,8 +103,9 @@ def compress_history(formatted_prompt):
     return output
 
 
-def comment_generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
+def comment_generate(prompt, history,post_check, 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):
+    full_conv=history
     print("###############\nRUNNING QUESTION GENERATOR\n###############\n")
     seed = random.randint(1,1111111111111111)
     agent=prompts.COMMENTER.format(focus=main_point[0])
@@ -132,9 +133,16 @@ def comment_generate(prompt, history, agent_name=agents[0], sys_prompt="", tempe
 
     for response in stream:
         output += response.token.text
-    #history.append((output,history))
+    history.append((output,None))
 
-    return output
+    post_check['comment']=post_check['comment']+1
+    #out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
+    full_conv.append((history,))
+    html_out=load_html(full_conv,title)
+
+    #history.append((output,history))
+    #[textbox, chatbot, textbox, json, json, html]
+    return "",history,post_check,post_check,post_check,html_out
 
 def reply_generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
 #def question_generate(prompt, history):
@@ -314,12 +322,12 @@ def generate(prompt, history, post_check, agent_name=agents[0], sys_prompt="", t
            
             #prompt = question_generate(output, history)
             #main_point[0]=prompt
-            #full_conv.append((output,prompt))
+        full_conv.append((output,prompt))
     
     
-            html_out=load_html(full_conv,title)
-            post_check={'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
-            yield prompt, history,post_check, summary[0],json_obj,json_hist,html_out
+        html_out=load_html(full_conv,title)
+        post_check={'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
+        yield prompt, history,post_check, summary[0],json_obj,json_hist,html_out
     else:
         print("passing blog")