Update app.py

app.py

@@ -142,13 +142,13 @@ def save_answers(*args):
 # Function to download table as CSV
 def download_table_to_csv():
     # Fetch data from Supabase table
-    response = supabase.table("oral_health_assessments").select("*").execute()
+    response = supabase.table(table_name).select("*").execute()
     
     # Check if data is available
     if not response.data:
         print("No data found in the table.")
-        return "No data found in the table."
-    
+        return None
+
     data = response.data
 
     # Prepare CSV data
@@ -162,21 +162,26 @@ def download_table_to_csv():
     for row in data:
         csv_data.append(row.values())
 
-    # Save CSV data to file
-    file_path = "oral_health_assessments.csv"
-    with open(file_path, "w", newline='') as f:
+    # Save CSV data to file (replace 'your_table.csv' with desired filename)
+    csv_file = "your_table.csv"
+    with open(csv_file, "w", newline='') as f:
         writer = csv.writer(f)
         writer.writerows(csv_data)
 
-    print(f"Downloaded table 'oral_health_assessments' to '{file_path}'")
-    return file_path
+    print(f"Downloaded table '{table_name}' to '{csv_file}'")
+    return csv_file
+
+def gradio_download():
+    file_path = download_table_to_csv()
+    if file_path:
+        return file_path
+    return None
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
     audio_input = gr.Audio(type="filepath", label="Record your audio", elem_id="audio_input")
     transcribe_button = gr.Button("Transcribe and Generate Form", elem_id="transcribe_button")
-    download_button = gr.Button("Download as CSV", elem_id="download_button")
     
     with gr.Row(elem_id="textboxes_row"):
         with gr.Column():
@@ -189,7 +194,11 @@ with gr.Blocks() as demo:
 
     transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes_left + textboxes_right)
     submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right, outputs=output_html)
-    download_button.click(fn=download_table_to_csv, outputs=gr.File())
+
+    download_button = gr.Button("Download Table as CSV")
+    download_csv_output = gr.File(label="Download CSV")
+
+    download_button.click(fn=gradio_download, inputs=[], outputs=download_csv_output)
 
 # Launch the app
 demo.launch(share=True)