Update app.py

app.py

@@ -65,21 +65,28 @@ import urllib.request
 import gradio as gr
 from gradio import inputs, outputs
 
-
-DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/Visualization"
-DATA_FILENAME = "level2.json"
-DATA_FILE = os.path.join("level2", DATA_FILENAME)
-
-HF_TOKEN = os.environ.get("HF_TOKEN")
-st.write("is none?", HF_TOKEN)
-
-st.write("hfh", huggingface_hub.__version__)
-
-repo = Repository(
-    local_dir="level2", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+feedback_file = Path("Seetha/Visualization") / f"level2.json"
+feedback_folder = feedback_file.parent
+# DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/Visualization"
+# DATA_FILENAME = "level2.json"
+# DATA_FILE = os.path.join("level2", DATA_FILENAME)
+
+# HF_TOKEN = os.environ.get("HF_TOKEN")
+# st.write("is none?", HF_TOKEN)
+
+# st.write("hfh", huggingface_hub.__version__)
+
+# repo = Repository(
+#     local_dir="level2", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+# )
+scheduler = CommitScheduler(
+    repo_id="Visualization",
+    repo_type="dataset",
+    folder_path=feedback_folder,
+    path_in_repo="level2",
+    every=10,
 )
 
-
 def main():
 
   st.title("Text to Causal Knowledge Graph")
@@ -481,13 +488,22 @@ def main():
             'target': col,
             'value': int(df_tab.loc[row, col])
         })
-  
-
-  dataset = load_dataset("Seetha/Visualization")
-  st.write(dataset)
-  with open(DATA_FILE,'w') as f:
-      json.dump(json_data, f)
-      repo.push_to_hub()
+        
+  def save_feedback():
+    """
+    Append input/outputs and user feedback to a JSON Lines file using a thread lock to avoid concurrent writes from different users.
+    """
+    with scheduler.lock:
+        with feedback_file.open("w") as f:
+            json.dump(json_data, f)
+            #f.write(json.dumps({"input": input_text, "output_1": output_1, "output_2": output_2, "user_choice": user_choice}))
+           # f.write("\n")
+
+  # dataset = load_dataset("Seetha/Visualization")
+  # st.write(dataset)
+  # with open(DATA_FILE,'w') as f:
+     # json.dump(json_data, f)
+      #repo.push_to_hub()
   # Write JSON to file
   #with open(DATA_FILE, 'w') as f: #w+
   # with urllib.request.urlopen('https://huggingface.co/datasets/Seetha/visual_files/raw/main/level2.json') as response:
@@ -512,11 +528,12 @@ def main():
   #       st.write(commit_url)
 
   #   #return generate_html()
-  # iface = gr.Interface(
-  #   store_message()  
-  # )
+  #with gr.Blocks() as demo:
+  iface = gr.Interface(
+    save_feedback() 
+  )
 
-  # iface.launch()
+  iface.launch()
 
     #f.truncate()