update download functionality to work in HF Spaces

app.py

@@ -1,5 +1,6 @@
 import re
 import os
+from io import BytesIO
 from dotenv import load_dotenv
 
 import gradio as gr
@@ -23,17 +24,14 @@ logs_df = PandasDataFrame(columns=logs_columns)
 
 def download_logs():
     global logs_df
-    # Check for the current operating system's desktop path
-    if os.name == 'nt':  # For Windows
-        desktop = os.path.join(os.path.join(os.environ['USERPROFILE']), 'Desktop')
-    else:  # For macOS and Linux
-        desktop = os.path.join(os.path.join(os.path.expanduser('~')), 'Desktop')
 
-    # Define the path to save the CSV file on the desktop
-    file_path = os.path.join(desktop, 'classification_logs.csv')
+    # Convert the DataFrame into a CSV byte stream
+    output = BytesIO()
+    logs_df.to_csv(output, index=False)
+    output.seek(0)  # Rewind the buffer
 
-    # Save the DataFrame to the CSV file on the desktop
-    logs_df.to_csv(file_path)
+    # Return the CSV byte stream and filename for Gradio to generate a download link
+    return output, "classification_logs.csv"
 
 
 def build_context(row):