Update app.py

app.py

@@ -4,13 +4,13 @@ from transformers import pipeline
 import os
 from supabase import create_client, Client
 from datetime import datetime
+import pandas as pd
 
 # Add your AssemblyAI API key as Environment Variable
 aai.settings.api_key = os.environ['Assembly']
 url: str = os.environ['DBUrl']
 key: str = os.environ['DBKey']
 
-
 # Initialize question answering pipeline
 question_answerer = pipeline("question-answering", model='distilbert-base-cased-distilled-squad')
 
@@ -62,6 +62,7 @@ oral_health_assessment_form = [
     "Any other finding",
     "Treatment plan",
 ]
+
 # Function to generate answers for the questions
 def generate_answer(question, context):
     result = question_answerer(question=question, context=context)
@@ -138,13 +139,24 @@ 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)
+            csv_data = df.to_csv(index=False)
+            return csv_data
+        else:
+            return "No data available"
+    except Exception as e:
+        print(f"Error fetching data from Supabase: {e}")
+        return f"Error: {e}"
+
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
-    
-    # Add HTML box to show the status
-    status_box = gr.HTML("<div id='status_box' style='width: 100px; height: 30px; background-color: red;'></div>")
-    
+    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")
     
@@ -157,16 +169,9 @@ with gr.Blocks() as demo:
     submit_button = gr.Button("Submit", elem_id="submit_button")
     output_html = gr.HTML(label="Submitted Answers")
 
-    # Update status box color based on audio input
-    def update_status(audio):
-        if audio:
-            status_box_value = "<div id='status_box' style='width: 100px; height: 30px; background-color: green;'></div>"
-        else:
-            status_box_value = "<div id='status_box' style='width: 100px; height: 30px; background-color: red;'></div>"
-        return status_box_value, main(audio)
-
-    transcribe_button.click(fn=update_status, inputs=audio_input, outputs=[status_box, textboxes_left + textboxes_right])
+    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"))
 
 # Launch the app
-demo.launch(share=True)
+demo.launch(share=True)