Update app.py

app.py

@@ -4,8 +4,7 @@ from transformers import pipeline
 import os
 from supabase import create_client, Client
 from datetime import datetime
-import pandas as pd
-import tempfile
+import csv
 
 # Add your AssemblyAI API key as Environment Variable
 aai.settings.api_key = os.environ['Assembly']
@@ -140,28 +139,43 @@ def save_answers(*args):
         print(f"Error inserting data into Supabase: {e}")
         return f"Error saving answers: {e}"
 
-def download_csv():
-    try:
-        response = supabase.table('oral_health_assessments').select("*").execute()
-        data = response.data
-        if data:
-            df = pd.DataFrame(data)
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as temp_file:
-                df.to_csv(temp_file.name, index=False)
-                temp_file_path = temp_file.name
-            return temp_file_path
-        else:
-            return "No data available"
-    except Exception as e:
-        print(f"Error fetching data from Supabase: {e}")
-        return f"Error: {e}"
+# Function to download table as CSV
+def download_table_to_csv():
+    # Fetch data from Supabase table
+    response = supabase.table("oral_health_assessments").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."
+    
+    data = response.data
+
+    # Prepare CSV data
+    csv_data = []
+
+    # Add header row
+    if len(data) > 0:
+        csv_data.append(data[0].keys())
+
+    # Add data rows
+    for row in data:
+        csv_data.append(row.values())
+
+    # Save CSV data to file
+    with open("oral_health_assessments.csv", "w", newline='') as f:
+        writer = csv.writer(f)
+        writer.writerows(csv_data)
+
+    print(f"Downloaded table 'oral_health_assessments' to 'oral_health_assessments.csv'")
+    return "Table downloaded successfully as 'oral_health_assessments.csv'"
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
-    download_button = gr.Button("Download CSV", elem_id="download_button")
     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():
@@ -174,7 +188,7 @@ 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_csv, inputs=None, outputs=gr.File(label="Download CSV"))
+    download_button.click(fn=download_table_to_csv, outputs=output_html)
 
-# Launches the app
+# Launch the app
 demo.launch(share=True)