Update app.py

app.py

@@ -21,6 +21,7 @@ summary =[]
 main_point=[]
 summary.append("")
 main_point.append("")
+list_of_users=["user1","user2","user3"]
 
 models=[
     "mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -150,7 +151,7 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
     #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))
-    html_out=load_html(full_conv,None)
+    html_out=load_html(full_conv,post_check['title'])
 
     #out_json = {'user':list_of_users[0],'datetime':current_time,'file_name':filename,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output,'comment_list':[]}
     file_n = f'{post_check["filename"]}.json'
@@ -161,7 +162,8 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
         try:
             lod = json.loads(r.text)
             print(f'lod:: {lod}')
-            lod[0]['comment_list'].append({'user':list_of_users[1],'datetime':'','comment':output})
+            lod[0]['comment']=lod[0]['comment']+1
+            lod[0]['comment_list'].append({'user':list_of_users[1],'datetime':'','comment':output,'reply_list':[]})
             #hist_out.append(out_json)
             #try:
             #    for ea in 
@@ -179,7 +181,7 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
             )
         except Exception as e:
             print(e)
-    return "",history,post_check,post_check,post_check,html_out
+    return "",history,lod[0],lod[0],lod[0],html_out
 
 
 
@@ -223,11 +225,36 @@ def reply_generate(prompt, history,post_check,full_conv, agent_name=agents[0], s
     #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,None,output))
-    html_out=load_html(full_conv,None)
+    html_out=load_html(full_conv,post_check['title'])
 
-    #history.append((output,history))
-    #[textbox, chatbot, textbox, json, json, html]
-    return "",history,post_check,post_check,post_check,html_out
+    file_n = f'{post_check["filename"]}.json'
+    print(file_n)
+    r = requests.get(f'{save_data}book1/{file_n}') 
+    print(f'status code main:: {r.status_code}')
+    if r.status_code==200:
+        try:
+            lod = json.loads(r.text)
+            print(f'lod:: {lod}')
+            lod[0]['reply']=lod[0]['reply']+1
+            lod[0]['comment_list'][0]['reply_list'].append({'user':list_of_users[1],'datetime':'','reply':output})
+            #hist_out.append(out_json)
+            #try:
+            #    for ea in 
+            with open(f'{uid}.json', 'w') as f:
+                json_hist=json.dumps(lod, indent=4)
+                f.write(json_hist)
+            f.close()
+        
+            upload_file(
+                path_or_fileobj =f"{uid}.json", 
+                path_in_repo = f"book1/{file_n}", 
+                repo_id =f"{username}/{dataset_name}", 
+                repo_type = "dataset", 
+                token=token,
+            )
+        except Exception as e:
+            print(e)]
+    return "",history,lod[0],lod[0],lod[0],html_out
 
 
     
@@ -333,7 +360,7 @@ def load_html_OG(inp,title):
     return html
 
 
-list_of_users=["user1","user2","user3"]
+
 
 def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
     html_out=""