Update app.py

app.py

@@ -23,61 +23,36 @@ from datetime import datetime
 # ---------------------------------------------
 # Dataset and Token links - change awacke1 to your own HF id, and add a HF_TOKEN copy to your repo for write permissions
 # This should allow you to save your results to your own Dataset hosted on HF. ---
-#DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/Carddata.csv"
-#DATASET_REPO_ID = "awacke1/Carddata.csv"
-#DATA_FILENAME = "Carddata.csv"
-#DATA_FILE = os.path.join("data", DATA_FILENAME)
-#HF_TOKEN = os.environ.get("HF_TOKEN")
-#SCRIPT = """
-
-#<script>
-#if (!window.hasBeenRun) {
-#    window.hasBeenRun = true;
-#    console.log("should only happen once");
-#    document.querySelector("button.submit").click();
-#}
-#</script>
-#"""
-
-#try:
-#    hf_hub_download(
-#        repo_id=DATASET_REPO_ID,
-#        filename=DATA_FILENAME,
-#        cache_dir=DATA_DIRNAME,
-#        force_filename=DATA_FILENAME
-#    )
-#except:
-#    print("file not found")
-#repo = Repository(
-#    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-#)
+DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/ASRLive.csv"
+DATASET_REPO_ID = "awacke1/ASRLive.csv"
+DATA_FILENAME = "ASRLive.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+SCRIPT = """
+
+try:
+    hf_hub_download(
+        repo_id=DATASET_REPO_ID,
+        filename=DATA_FILENAME,
+        cache_dir=DATA_DIRNAME,
+        force_filename=DATA_FILENAME
+    )
+except:
+    print("file not found")
+repo = Repository(
+    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
            
-#def store_message(name: str, message: str):
-#    if name and message:
-#        with open(DATA_FILE, "a") as csvfile:
-#            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
-#            writer.writerow(
-#                {"name": name.strip(), "message": message.strip(), "time": str(datetime.now())}
-#            )
-#        # uncomment line below to begin saving - 
-#        commit_url = repo.push_to_hub()
-#    return ""            
-
-#iface = gr.Interface(
-#    store_message,
-#    [
-#        inputs.Textbox(placeholder="Your name"),
-#        inputs.Textbox(placeholder="Your message", lines=2),
-#    ],
-#    "html",
-#    css="""
-#    .message {background-color:cornflowerblue;color:white; padding:4px;margin:4px;border-radius:4px; }
-#    """,
-#    title="Reading/writing to a HuggingFace dataset repo from Spaces",
-#    description=f"This is a demo of how to do simple *shared data persistence* in a Gradio Space, backed by a dataset repo.",
-#    article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL})",
-#)
-
+def store_message(name: str, message: str):
+    if name and message:
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
+            writer.writerow(
+                {"name": name.strip(), "message": message.strip(), "time": str(datetime.now())}
+            )
+        # uncomment line below to begin saving - 
+        commit_url = repo.push_to_hub()
+    return ""            
 
 # main -------------------------
 mname = "facebook/blenderbot-400M-distill"
@@ -114,7 +89,7 @@ def chat(message, history):
     history_useful = add_note_to_history(response, history_useful)
     list_history = history_useful[0].split('</s> <s>')
     history.append((list_history[-2], list_history[-1]))
-#    store_message(message, response) # Save to dataset - uncomment if you uncomment above to save inputs and outputs to your dataset
+    store_message(message, response) # Save to dataset - uncomment if you uncomment above to save inputs and outputs to your dataset
     return history, history
     
 
@@ -143,7 +118,7 @@ def transcribe(audio, state = ""):
         if type(transcriptions) == tuple and len(transcriptions) == 2:
             transcriptions = transcriptions[0]
         transcriptions = transcriptions[0]
-#    store_message(transcriptions, state) # Save to dataset - uncomment to store into a dataset - hint you will need your HF_TOKEN
+    store_message(transcriptions, state) # Save to dataset - uncomment to store into a dataset - hint you will need your HF_TOKEN
     state = state + transcriptions + " "
     return state, state
 
@@ -159,10 +134,10 @@ iface = gr.Interface(
     ],
     layout="horizontal",
     theme="huggingface",
-    title="🗣️LiveSpeechRecognition🧠Memory💾",
+    title="🗣️ASR-Live🧠Memory💾",
     description=f"Live Automatic Speech Recognition (ASR) with Memory💾 Dataset.",
     allow_flagging='never',
     live=True,    
-#    article=f"Result Output Saved to Memory💾 Dataset: [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
+    article=f"Result Output Saved to Memory💾 Dataset: [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
 )
 iface.launch()