Update app.py

app.py

@@ -146,6 +146,20 @@ def create_valid_filename(invalid_filename: str) -> str:
                       '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '_', '-')
     return ''.join(char for char in valid_chars if char in allowed_chars)
 
+
+
+
+
+    
+def load_html(inp):
+    with open('index.html','r') as h:
+        html=h.read()
+        html = html.replace("$body",f"{inp}")
+    h.close()
+    return html
+
+
+
 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,):
     #main_point[0]=prompt
     #print(datetime.datetime.now())
@@ -184,7 +198,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
             formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
             post_cnt+=1
         else:
-            system_prompt=prompts.REPLY_TO_COMMENTER.format(focus=main_points[0])
+            system_prompt=prompts.REPLY_TO_COMMENTER.format(focus=main_point[0])
     
             generate_kwargs = dict(
                 temperature=temperature,
@@ -209,7 +223,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
         
             for response in stream:
                 output += response.token.text
-                yield '', [(prompt,output)],summary[0],json_obj, json_hist
+                yield '', [(prompt,output)],summary[0],json_obj, json_hist,html_out
             out_json = {"prompt":prompt,"output":output}
         
             prompt = question_generate(output, history)
@@ -258,15 +272,13 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
            
         prompt = question_generate(output, history)
         main_point[0]=prompt
-            
-    return prompt, history, summary[0],json_obj,json_hist 
 
-def load_html():
-    with open('index.html','r') as h:
-        html=h.read()
-        html = html.replace("$name","Test")
-    h.close()
-    return html
+
+        html_out=load_html(history)
+        yield prompt, history, summary[0],json_obj,json_hist,html_out
+    return prompt, history, summary[0],json_obj,json_hist,html_out
+
+
 
 
 with gr.Blocks() as app: