improved csv handling

app.py

@@ -214,20 +214,24 @@ def getBeatsv2(audio:gr.Audio):
     fig = plotBeattimes(times_label_one, audiodata, sr, times_label_zero)
 
 
-    
+    featuredf = featuredf.drop(columns=["S1 to S2", "S2 to S1"])
 
 
     return fig, featuredf.to_markdown(), (sr, audiodata)
 
-def updateBeatsv2(beattimes_table:gr.Markdown, audio:gr.Audio, uploadeddf:gr.File=None)-> go.Figure:
-
-    df = mdpd.from_md(df)
+def updateBeatsv2(audio:gr.Audio, uploadeddf:gr.File=None)-> go.Figure:
 
     sr, audiodata = getaudiodata(audio)
 
 
     if uploadeddf != None:
-        beattimes_table = pd.read_csv(uploadeddf)
+        beattimes_table = pd.read_csv(
+            filepath_or_buffer=uploadeddf,
+            sep=";",
+            decimal=",",
+            encoding="utf-8-sig")
+    else:
+        raise FileNotFoundError("No file uploaded")
 
     s1_times = beattimes_table[beattimes_table["Label (S1=0/S2=1)"] == 0]["Beattimes"].to_numpy()
     s2_times = beattimes_table[beattimes_table["Label (S1=0/S2=1)"] == 1]["Beattimes"].to_numpy()
@@ -236,16 +240,24 @@ def updateBeatsv2(beattimes_table:gr.Markdown, audio:gr.Audio, uploadeddf:gr.Fil
 
     return fig, beattimes_table.to_markdown()
 
-def download_df (df: str):
+def download_df (beattimes_table: str):
 
-    df = mdpd.from_md(df)
+    df = mdpd.from_md(beattimes_table)
+    df['Beattimes'] = df['Beattimes'].astype(float)
+    df['Label (S1=0/S2=1)'] = df['Label (S1=0/S2=1)'].astype(int)
 
-    print(df)
 
     temp_dir = tempfile.gettempdir()
-    temp_path = os.path.join(temp_dir, "feature_data.csv")
+    temp_path = os.path.join(temp_dir, "beattimes.csv")
 
-    df.to_csv(temp_path, index=False)
+    
+    df.to_csv(
+        index=False,
+        columns=["Beattimes", "Label (S1=0/S2=1)"],
+        path_or_buf=temp_path, 
+        sep=";",
+        decimal=",",
+        encoding="utf-8-sig")
 
     return temp_path
 
@@ -291,7 +303,7 @@ with gr.Blocks() as app:
 
         csv_download.click(download_df, inputs=[beattimes_table], outputs=[csv_download])
         getBeatsbtn.click(getBeatsv2, inputs=audiofile, outputs=[beats_wave_plot, beattimes_table, cleanedaudio])
-        updateBeatsbtn.click(updateBeatsv2, inputs=[beattimes_table, audiofile, uploadDF], outputs=[beats_wave_plot, beattimes_table])
+        updateBeatsbtn.click(updateBeatsv2, inputs=[audiofile, uploadDF], outputs=[beats_wave_plot, beattimes_table])
 
         gr.Examples(
             examples=example_files,