Update app.py

app.py

@@ -201,7 +201,7 @@ def comment_generate(prompt, history,post_check,full_conv,persona2, agent_name=a
 
 
 
-def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
+def reply_generate(prompt, history,post_check,full_conv,persona1,reply_to_comment, 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):
     current_time = str(datetime.datetime.now())
 
@@ -211,7 +211,7 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
     print(f'full_conv::\n{full_conv}')
     print("###############\nRUNNING REPLY GENERATOR\n###############\n")
     seed = random.randint(1,1111111111111111)
-    agent=prompts.REPLY_TO_COMMENTER.format(focus=full_conv[0][0],comment=full_conv[1][1],persona=persona[persona1]['personality'])
+    agent=prompts.REPLY_TO_COMMENTER.format(focus=post_check['output'],comment=post_check['comment_list'][reply_to_comment]['comment'],persona=persona[persona1]['personality'])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -246,7 +246,7 @@ def reply_generate(prompt, history,post_check,full_conv,persona1, agent_name=age
 
     #comment_out=post_check['comment_list']
 
-    post_check['comment_list'][0]['reply_list'].append(reply_json)
+    post_check['comment_list'][reply_to_comment]['reply_list'].append(reply_json)
     #print(comment_out)
     #reply_out.append(reply_json)
     #post_check['comment_list'][0]
@@ -305,58 +305,13 @@ def load_html(conv):
                 for repl in conv['comment_list'][i]['reply_list']:
                     ht+=f"""<pre class="resp2"><div class="bhead"></div>{repl}</pre>"""
     ht+=f"""</div>"""                
-    '''
-    if inp:
-        for i,ea in enumerate(inp):
-            
-            blog,comm,repl=ea
-            #print(f'outp:: {outp}')
-            #print(f'prom:: {prom}')
-            ht+=f"""<div class="div_box">"""
-            if blog:
-                #ht+=f"""<div class="bhead"><div><h1>$btitle</h1></div><div>$user_name</div></div>"""
-                ht+=f"""<pre class="bpost"><div class="bhead"><h2>{conv['title']}</h2><br><h5>{conv['user']}</h5></div>{blog}</pre>"""
-            if comm:
-                ht+=f"""<pre class="resp1"><div class="bhead"></div>{comm}</pre>"""
-            if repl:
-                ht+=f"""<pre class="resp2"><div class="bhead"></div>{repl}</pre>"""
-            ht+=f"""</div>"""                
-        
-    '''
-    with open('index.html','r') as h:
-        html=h.read()
-        html = html.replace("$body",f"{ht}")
-        #html = html.replace("$title",f"{title}")
-        #html = html.replace("$title",f"{title}")
-        #html = html.replace("$user_name",f"{user_name}")
-    h.close()
-    return html
-    
-def load_html_OG(inp,title):
-    ht=""
-    if inp:
-        for i,ea in enumerate(inp):
-            
-            outp,prom=ea
-            #print(f'outp:: {outp}')
-            #print(f'prom:: {prom}')
-            if i == 0:
-                ht+=f"""<div class="div_box">
-                <pre class="bpost">{outp}</pre>
-                <pre class="resp1">{prom}</pre>
-                </div>"""                
-            else:
-                ht+=f"""<div class="div_box">
-                <pre class="resp2">{outp}</pre>
-                <pre class="resp2">{prom}</pre>
-                </div>"""
+ 
     with open('index.html','r') as h:
         html=h.read()
         html = html.replace("$body",f"{ht}")
-        html = html.replace("$title",f"{title}")
     h.close()
     return html
-
+ 
 
 
 
@@ -506,12 +461,14 @@ with gr.Blocks() as app:
     chatbot=gr.Chatbot(visible=False)
     msg = gr.Textbox()
     with gr.Row():
-        submit_b = gr.Button("Blog Post")
-        submit_c = gr.Button("Comment")
-        submit_r = gr.Button("OP Reply")
-    with gr.Row():
-        stop_b = gr.Button("Stop")
-        clear = gr.ClearButton([msg, chatbot])
+        with gr.Column(scale=2):
+            submit_b = gr.Button("Blog Post")
+            submit_c = gr.Button("Comment")
+            reply_num= gr.Number(label="Replying to Comment:", value=0)
+            submit_r = gr.Button("OP Reply")
+        with gr.Column(scale=1):
+            stop_b = gr.Button("Stop")
+            clear = gr.ClearButton([msg, chatbot])
     
     with gr.Row():
         m_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],value=models[0],interactive=True)
@@ -534,7 +491,7 @@ with gr.Blocks() as app:
     
     sub_b = submit_b.click(generate, [msg,chatbot,post_handler,chat_handler,persona1,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,persona2],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
-    sub_r = submit_r.click(reply_generate, [msg,chatbot,post_handler,chat_handler,persona1],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
+    sub_r = submit_r.click(reply_generate, [msg,chatbot,post_handler,chat_handler,persona1,reply_num],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_e = msg.submit(generate, [msg,chatbot,post_handler,chat_handler,chat_handler,persona1,tokens],[msg,chatbot,post_handler,chat_handler,sumbox,sum_out_box,hist_out_box,html])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e,sub_c,sub_r])