Update app.py

app.py

@@ -62,25 +62,22 @@ from datetime import datetime
 import pathlib as Path
 from requests import get
 import urllib.request
+import gradio as gr
+from gradio import inputs, outputs
 
 
-# DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/visual_files/raw/main/level2.json"
-# DATA_FILENAME = "level2.json"
-# #DATA_FILE = os.path.join(DATASET_REPO_URL, DATA_FILENAME)
+DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/visual_files"
+DATA_FILENAME = "level2.json"
+DATA_FILE = os.path.join("level2", DATA_FILENAME)
 
-# # feedback_file = Path("https://huggingface.co/datasets/Seetha/visual_files/") / f"level2.json"
-# st.write(DATASET_REPO_URL)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+st.write("is none?", HF_TOKEN is None)
 
-# HF_TOKEN = os.environ.get("HF_TOKEN")
+st.write("hfh", huggingface_hub.__version__)
 
-# repo = Repository(
-#     local_dir="huggingface-hub", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-# )
-#dataset = load_dataset('Seetha/visual_files')
-
-# print("is none?", HF_TOKEN is None)
-
-# print("hfh", huggingface_hub.__version__)
+repo = Repository(
+    local_dir="level2", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
 
 
 def main():
@@ -501,6 +498,20 @@ def main():
     # f.seek(0)
     st.write(str(json_data))
     json.dump(json_data, f)
+
+  def store_message():
+    with open(DATA_FILE, "w") as f:
+        json.dump(json_data, f)
+        commit_url = repo.push_to_hub()
+        st.write(commit_url)
+
+    #return generate_html()
+  iface = gr.Interface(
+    store_message()  
+  )
+
+  iface.launch()
+
     #f.truncate()
 
   df_final1.to_csv('predictions.csv')