Update app.py

app.py

@@ -6,10 +6,11 @@ import prompts
 import json
 import uuid
 import os
-
 token=os.environ.get("HF_TOKEN")
 username="omnibus"
 dataset_name="tmp"
+save_data=f'https://huggingface.co/datasets/{username}/{dataset_name}/raw/main/'
+
 api=HfApi(token="")
 VERBOSE=False
 
@@ -149,8 +150,28 @@ def comment_generate(prompt, history,post_check,full_conv, agent_name=agents[0],
     full_conv.append((None,output,None))
     html_out=load_html(full_conv,None)
 
-    #history.append((output,history))
-    #[textbox, chatbot, textbox, json, json, html]
+    #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':[]}
+    r = requests.get(f'{save_data}{post_check['filename']}') 
+    print(f'status code main:: {r.status_code}')
+    if r.status_code==200:
+        
+        lod = json.loads(r.text)
+        lod = lod['comment_list'].append({'user':list_of_users[1],'datetime':'','comment':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/{post_check['filename']}", 
+            repo_id =f"{username}/{dataset_name}", 
+            repo_type = "dataset", 
+            token=token,
+        )
     return "",history,post_check,post_check,post_check,html_out
 
 
@@ -379,7 +400,7 @@ def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_pr
                         print(f'title:: {title}')
                         filename=create_valid_filename(f'{current_time}---{title}')
                         
-            out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
+            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':[]}
 
             hist_out.append(out_json)
             #try:
@@ -425,7 +446,7 @@ def generate(prompt, history, post_check,full_conv, agent_name=agents[0], sys_pr
     
     
         html_out=load_html(full_conv,title,list_of_users[0])
-        post_check={'user':list_of_users[0],'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
+        post_check={'filename':filename,'user':list_of_users[0],'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0}
         yield prompt, history,post_check,full_conv,summary[0],json_obj,json_hist,html_out
     else:
         print("passing blog")