Update app.py

app.py

@@ -185,12 +185,12 @@ def reply_generate(prompt, history,post_check,full_conv, agent_name=agents[0], s
         output += response.token.text
     history.append((output,None))
 
-    comment_cnt=post_check['comment']
+    reply_cnt=post_check['reply']
     print(type(comment_cnt))
-    post_check['comment']=comment_cnt+1
+    post_check['reply']=reply_cnt+1
     #out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
     #full_conv[-1]+=(output,)
-    full_conv.append((None,output,None))
+    full_conv.append((None,None,output))
     html_out=load_html(full_conv,None)
 
     #history.append((output,history))
@@ -344,6 +344,7 @@ def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_pr
             prompt=prompt.strip(' \"')
 
         formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
+
         
         if len(formatted_prompt) < (40000):
             print(len(formatted_prompt))
@@ -353,15 +354,20 @@ def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_pr
             output = ""
             #if history:
             #    yield history
-        
+            if not prompt:
+                prompt_out == None
+            else: 
+                prompt_out=prompt
             for response in stream:
                 output += response.token.text
-                yield '', [(prompt,output)],post_check,full_conv,summary[0],json_obj, json_hist,html_out
+                yield "", [(prompt_out,output)],post_check,full_conv,summary[0],json_obj, json_hist,html_out
             
             if not title:
                 for line in output.split("\n"):
                     if "title" in line.lower() and ":" in line.lower():
                         title = line.split(":")[1]
+                        if title.startswith(' \"'):
+                            title=title.strip(' \"')                        
                         print(f'title:: {title}')
                         filename=create_valid_filename(f'{current_time}---{title}')
                         
@@ -406,7 +412,7 @@ def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_pr
     
            
             #prompt = question_generate(output, history)
-            #main_point[0]=prompt
+        #main_point[0]=output
         full_conv.append((output,None,None))