CSV fix

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
 import tempfile
 from data_generator import generate_test_data
+import pandas as pd
+import os
 
 def generate_data(prompt):
     try:
         df = generate_test_data(prompt)
-        
-        # Save DataFrame to a temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w", encoding="utf-8")
+        # Save to temporary CSV file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', newline='')
         df.to_csv(temp_file.name, index=False)
         temp_file.close()
-
         return df, temp_file.name
     except Exception as e:
         return f"Error: {str(e)}", None