Update app.py

app.py

@@ -9,7 +9,6 @@ import re
 from typing import List, Tuple
 import pandas as pd
 import io
-import base64
 import tempfile
 
 # Load environment variables
@@ -170,10 +169,10 @@ def groq_optimized_frequency_pattern(frequency: float, duration: float, material
         # Generate cymatic pattern
         cymatic_pattern = generate_cymatic_pattern(waveform)
 
-        # Generate plots and save them as temporary files
-        waveform_plot = plot_to_base64(create_waveform_plot(waveform, duration))
-        spectrum_plot = plot_to_base64(create_spectrum_plot(waveform, sample_rate))
-        cymatic_plot = plot_to_base64(create_cymatic_plot(cymatic_pattern))
+        # Generate plots
+        waveform_plot = create_waveform_plot(waveform, duration)
+        spectrum_plot = create_spectrum_plot(waveform, sample_rate)
+        cymatic_plot = create_cymatic_plot(cymatic_pattern)
 
         # Generate audio data
         audio_data = (waveform * 32767).astype(np.int16)
@@ -184,10 +183,10 @@ def groq_optimized_frequency_pattern(frequency: float, duration: float, material
         error_message = f"An error occurred: {str(e)}"
         return None, None, error_message, None, None, None
 
-def plot_to_base64(fig):
+def plot_to_file(fig):
     with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp:
         fig.savefig(tmp.name, format='png')
-        return tmp.name
+    return tmp.name
 
 def create_waveform_plot(waveform, duration):
     fig, ax = plt.subplots(figsize=(12, 6))
@@ -195,7 +194,6 @@ def create_waveform_plot(waveform, duration):
     ax.set_title('Waveform')
     ax.set_xlabel('Time [s]')
     ax.set_ylabel('Amplitude')
-    plt.close(fig)
     return fig
 
 def create_spectrum_plot(waveform, sample_rate):
@@ -205,7 +203,6 @@ def create_spectrum_plot(waveform, sample_rate):
     ax.set_ylabel('Frequency [Hz]')
     ax.set_xlabel('Time [sec]')
     ax.set_title('Frequency Spectrum')
-    plt.close(fig)
     return fig
 
 def create_cymatic_plot(cymatic_pattern):
@@ -213,16 +210,14 @@ def create_cymatic_plot(cymatic_pattern):
     ax.imshow(cymatic_pattern, cmap='viridis')
     ax.set_title('Simulated Cymatic Pattern')
     ax.axis('off')
-    plt.close(fig)
     return fig
 
 def export_data(waveform, sample_rate, duration):
     time = np.linspace(0, duration, len(waveform))
     df = pd.DataFrame({'Time': time, 'Amplitude': waveform})
-    csv_buffer = io.StringIO()
-    df.to_csv(csv_buffer, index=False)
-    csv_buffer.seek(0)
-    return csv_buffer.getvalue()
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.csv', delete=False, newline='') as tmp:
+        df.to_csv(tmp.name, index=False)
+    return tmp.name
 
 def cymatic_interface(frequency, duration, material_type, length, width, height):
     dimensions = (length, width, height)
@@ -233,9 +228,13 @@ def cymatic_interface(frequency, duration, material_type, length, width, height)
     
     waveform_plot, audio_data, description, spectrum_plot, waveform, cymatic_plot = results
     
-    csv_data = export_data(waveform, 44100, duration)
+    waveform_plot_file = plot_to_file(waveform_plot)
+    spectrum_plot_file = plot_to_file(spectrum_plot)
+    cymatic_plot_file = plot_to_file(cymatic_plot)
     
-    return waveform_plot, (44100, audio_data), description, spectrum_plot, waveform_plot, cymatic_plot, spectrum_plot, csv_data
+    csv_file = export_data(waveform, 44100, duration)
+    
+    return waveform_plot_file, (44100, audio_data), description, spectrum_plot_file, waveform_plot_file, cymatic_plot_file, spectrum_plot_file, csv_file
 
 # Gradio UI elements
 with gr.Blocks() as iface:
@@ -274,7 +273,7 @@ with gr.Blocks() as iface:
             description_output = gr.Textbox(label="Generated Description", lines=10)
         
         with gr.TabItem("Data Export"):
-            csv_output = gr.File(label="Download CSV", file_types=[".csv"])
+            csv_output = gr.File(label="Download CSV")
     
     error_output = gr.Textbox(label="Error Message", visible=False)