Update audio_save.py

audio_save.py

@@ -8,11 +8,15 @@ if not os.path.exists('./tarun_1234'):
 
 hf_token = os.environ['HF_TOKEN']
 
+api = HfApi(token = hf_token)
+
 @app.post("/save_audio")
-async def save_audio(audio: UploadFile = File(...), path: str = Form(...)):
+async def save_audio(audio: UploadFile = File(...), path: str = Form(...), script_name: str = Form(...)):
 
     # print('PATH ************************ >>>>>', path)
     folder_path = path.split('/')[1]
+    filename = path.split('/')[-1]
+    # script_name = script_name
 
     count = path.split('/')[-1].split('.')[0]
 
@@ -52,11 +56,11 @@ async def save_audio(audio: UploadFile = File(...), path: str = Form(...)):
     # df1 = pd.DataFrame(l,columns = ['script','response','transcription'])
         
 
-    # api.upload_file(
-    # path_or_fileobj=f'{folder_path}/transcription/{folder_path}.csv',
-    # path_in_repo=f"{folder_path}.csv",
-    # repo_id="hellojarvis/agent_test",
-    # repo_type="dataset",)
+    api.upload_file(
+    path_or_fileobj=path,
+    path_in_repo=f"{folder_path}/{script_name}/{filename}",
+    repo_id="hellojarvis/agent_data",
+    repo_type="dataset",)
 
     # api.upload_file(
     # path_or_fileobj=f'{folder_path}/transcription/state_wise_duplicates.csv',