Update app.py

app.py

@@ -103,10 +103,10 @@ def compress_history(formatted_prompt):
     return output
 
 
-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 comment_generate(prompt, history,post_check,full_conv, 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):
     print(post_check)
-    full_conv=history
+    #full_conv=history
     print(f'full_conv::\n{full_conv}')
     print("###############\nRUNNING QUESTION GENERATOR\n###############\n")
     seed = random.randint(1,1111111111111111)
@@ -206,11 +206,14 @@ def load_html(inp,title):
             blog,comm,repl=ea
             #print(f'outp:: {outp}')
             #print(f'prom:: {prom}')
-            ht+=f"""<div class="div_box">
-            <pre class="bpost">{blog}</pre>
-            <pre class="resp1">{comm}</pre>
-            <pre class="resp2">{repl}</pre>
-            </div>"""                
+            ht+=f"""<div class="div_box">"""
+            if blog:
+                ht+=f"""<pre class="bpost">{blog}</pre>"""
+            if comm:
+                ht+=f"""<pre class="resp1">{comm}</pre>"""
+            if repl:
+                ht+=f"""<pre class="resp2">{repl}</pre>"""
+            ht+=f"""</div>"""                
     with open('index.html','r') as h:
         html=h.read()
         html = html.replace("$body",f"{ht}")
@@ -352,7 +355,7 @@ def generate(prompt, history, post_check, agent_name=agents[0], sys_prompt="", t
     
         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
+        yield prompt, history,post_check,full_conv,summary[0],json_obj,json_hist,html_out
     else:
         print("passing blog")
 
@@ -493,6 +496,7 @@ def generate_OG(prompt, history, agent_name=agents[0], sys_prompt="", temperatur
 
 
 with gr.Blocks() as app:
+    chat_handler=gr.Stat()
     post_handler=gr.State()
     html = gr.HTML()
 
@@ -520,8 +524,8 @@ 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,tokens],[msg,chatbot,post_handler,sumbox,sum_out_box,hist_out_box,html])
-    sub_c = submit_c.click(comment_generate, [msg,chatbot,post_handler],[msg,chatbot,sumbox,sum_out_box,hist_out_box,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_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])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e,sub_c,sub_r])