Update app.py

app.py

@@ -248,7 +248,7 @@ def load_html_OG(inp,title):
 
 
 
-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):
+def generate(prompt, history, post_check,full_conv, 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())
@@ -272,6 +272,12 @@ def generate(prompt, history, post_check, agent_name=agents[0], sys_prompt="", t
     json_obj={}
     full_conv=[]
     post_cnt=1
+    if not post_check:
+        post_check={}
+    if not full_conv:
+        full_conv={}
+
+    
     seed = random.randint(1,1111111111111111)
     if not post_check:
         print("writing blog")
@@ -299,7 +305,7 @@ def generate(prompt, history, post_check, agent_name=agents[0], sys_prompt="", t
         
             for response in stream:
                 output += response.token.text
-                yield '', [(prompt,output)],{},summary[0],json_obj, json_hist,html_out
+                yield '', [(prompt,output)],post_check,full_conv,summary[0],json_obj, json_hist,html_out
             
             if not title:
                 for line in output.split("\n"):
@@ -524,7 +530,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,post_handler,chat_handler,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
+    sub_b = submit_b.click(generate, [msg,chatbot,post_handler,chat_handler,chat_handler,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
     sub_c = submit_c.click(comment_generate, [msg,chatbot,post_handler,chat_handler],[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])