Update app.py

app.py

@@ -176,6 +176,8 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
     #print(datetime.datetime.now())
     uid=uuid.uuid4()
     current_time = str(datetime.datetime.now())
+    title=""
+    filename=create_valid_filename(f'{current_time}---{title}')
     
     current_time=current_time.replace(":","-")
     current_time=current_time.replace(".","-")
@@ -191,7 +193,6 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
     json_hist={}
     json_obj={}
     full_conv=[]
-    filename=create_valid_filename(f'{prompt}---{current_time}')
     post_cnt=1
     while True:
         seed = random.randint(1,1111111111111111)
@@ -208,6 +209,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
                 prompt=prompt.strip(' \"')
 
             formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
+            
             post_cnt+=1
         else:
             system_prompt=prompts.REPLY_TO_COMMENTER.format(focus=main_point[0])
@@ -236,6 +238,14 @@ 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,html_out
+            
+            if not title:
+                for line in output:
+                    if "title" in line.lower() and ":" in line.lower():
+                        title = line.split(":")[1]
+                        print(f'title:: {title}')
+                        filename=create_valid_filename(f'{current_time}---{title}')
+                        
             out_json = {"prompt":prompt,"output":output}
         
             prompt = question_generate(output, history)