Update app.py

app.py

@@ -17,48 +17,48 @@ def submit(intelligibility, naturalness, expressiveness):
     debug_info.append(f"Current working directory: {os.getcwd()}")
     debug_info.append(f"Target directory: {csv_dir}")
     debug_info.append(f"Target file: {csv_file_path}")
-
+    
     try:
         # Create directory if it doesn't exist
         debug_info.append("Trying to create directory...")
         os.makedirs(csv_dir, exist_ok=True)
         debug_info.append(f"Directory created or already exists: {os.path.exists(csv_dir)}")
-
+        
         # List directory contents
         if os.path.exists(csv_dir):
             debug_info.append(f"Directory contents: {os.listdir(csv_dir)}")
-
+        
         # Check if file exists to add headers if new
         file_exists = os.path.isfile(csv_file_path)
         debug_info.append(f"File exists: {file_exists}")
-
+        
         # Try writing to the file
         debug_info.append("Attempting to write to file...")
         with open(csv_file_path, "a", newline="") as f:
             writer = csv.writer(f)
-
+            
             # Add headers if file is new
             if not file_exists:
                 writer.writerow(["Audio A", "Audio B", "Intelligibility", "Naturalness", "Expressiveness"])
-
+            
             writer.writerow([audio_a_path, audio_b_path, intelligibility, naturalness, expressiveness])
             f.flush()  # Force write to disk
             os.fsync(f.fileno())  # Ensure it's written to the physical disk
-
+        
         # Verify the file was written
         debug_info.append(f"File exists after write: {os.path.isfile(csv_file_path)}")
         if os.path.isfile(csv_file_path):
             debug_info.append(f"File size: {os.path.getsize(csv_file_path)} bytes")
-
+        
         # Read and display CSV content
         csv_content = ""
         if os.path.isfile(csv_file_path):
             with open(csv_file_path, "r") as f:
                 csv_content = f.read()
             debug_info.append(f"\nCSV Content:\n{csv_content}")
-
+        
         return "Thank you for your feedback! Data saved successfully.\n\nDebug info:\n" + "\n".join(debug_info)
-
+    
     except Exception as e:
         exc_type, exc_value, exc_traceback = sys.exc_info()
         tb_str = traceback.format_exception(exc_type, exc_value, exc_traceback)
@@ -67,23 +67,36 @@ def submit(intelligibility, naturalness, expressiveness):
         debug_info.append("".join(tb_str))
         return "Error occurred. Debug info:\n" + "\n".join(debug_info)
 
+# Add this function for downloading the CSV
+def get_csv_data():
+    if os.path.exists(csv_file_path):
+        # Return the path to the file for download
+        return csv_file_path
+    return None
+
 with gr.Blocks() as demo:
     gr.Markdown("# 🎧 MOS Evaluation")
     gr.Markdown("Compare Audio A and Audio B, then rate them below.")
-
+    
     with gr.Row():
         with gr.Column():
             gr.Audio(audio_a_path, label="Audio A", type="filepath")
         with gr.Column():
             gr.Audio(audio_b_path, label="Audio B", type="filepath")
-
+    
     intelligibility = gr.Slider(1, 5, value=3, step=1, label="Intelligibility")
     naturalness = gr.Slider(1, 5, value=3, step=1, label="Naturalness")
     expressiveness = gr.Slider(1, 5, value=3, step=1, label="Expressiveness")
-
+    
     submit_btn = gr.Button("Submit")
     output = gr.Textbox(label="Status", lines=15)  # Increased size for debug and CSV output
-
+    
     submit_btn.click(fn=submit, inputs=[intelligibility, naturalness, expressiveness], outputs=output)
+    
+    # Add download section - ADD THIS NEW SECTION
+    gr.Markdown("## Download Results")
+    download_btn = gr.Button("Download Results CSV")
+    csv_output = gr.File(label="CSV File")
+    download_btn.click(fn=get_csv_data, inputs=[], outputs=csv_output)
 
 demo.launch()