Update app.py

app.py

@@ -151,7 +151,7 @@ def create_valid_filename(invalid_filename: str) -> str:
 
 
     
-def load_html(inp):
+def load_html(inp,title):
     ht=""
     if inp:
         for ea in inp:
@@ -159,12 +159,13 @@ def load_html(inp):
             print(f'outp:: {outp}')
             print(f'prom:: {prom}')
             ht+=f"""<div class="div_box">
-            <div class="resp">{outp}</div>
-            <div class="resp">{prom}</div>
+            <div class="resp1">{outp}</div>
+            <div class="resp2">{prom}</div>
             </div>"""
     with open('index.html','r') as h:
         html=h.read()
         html = html.replace("$body",f"{ht}")
+        html = html.replace("$body",f"{title}")
     h.close()
     return html
 
@@ -297,7 +298,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
         full_conv.append((output,prompt))
 
 
-        html_out=load_html(full_conv)
+        html_out=load_html(full_conv,title)
         yield prompt, history, summary[0],json_obj,json_hist,html_out
     return prompt, history, summary[0],json_obj,json_hist,html_out