Update app.py

app.py

@@ -212,7 +212,7 @@ def load_html(inp,title):
 
 
 
-def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
+def generate(prompt, history, post_check, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
     html_out=""
     #main_point[0]=prompt
     #print(datetime.datetime.now())
@@ -237,7 +237,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
     full_conv=[]
     post_cnt=1
     seed = random.randint(1,1111111111111111)
-    if not history:
+    if not post_check:
         generate_kwargs = dict(
             temperature=temperature,
             max_new_tokens=max_new_tokens2,
@@ -317,9 +317,9 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
     
     
             html_out=load_html(full_conv,title)
-            yield prompt, history, summary[0],json_obj,json_hist,html_out
-    else:
-        pass            
+            post_check.append('blog':1)
+            yield prompt, history,post_check, summary[0],json_obj,json_hist,html_out
+       
 
 def generate_OG(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
     html_out=""
@@ -485,7 +485,7 @@ with gr.Blocks() as app:
     m_choice.change(load_models,m_choice,[chatbot])
     app.load(load_models,m_choice,[chatbot]).then(load_html,None,html)
     
-    sub_b = submit_b.click(generate, [msg,chatbot,tokens],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
+    sub_b = submit_b.click(generate, [msg,chatbot,post_handler,tokens],[msg,chatbot,post_handler,sumbox,sum_out_box,hist_out_box,html])
     sub_c = submit_c.click(comment_generate, [msg,chatbot,tokens],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_r = submit_r.click(reply_generate, [msg,chatbot,tokens],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_e = msg.submit(generate, [msg, chatbot,tokens], [msg, chatbot,sumbox,sum_out_box,hist_out_box,html])