Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 import assemblyai as aai
 from transformers import pipeline
-import pandas as pd
 import os
+from supabase import create_client, Client
 
 # 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')
@@ -104,6 +107,9 @@ def fill_textboxes(context):
         answers.append(answer)
     return answers
 
+# Supabase configuration
+supabase: Client = create_client(url, key)
+
 # Main Gradio app function
 def main(audio):
     context = transcribe_audio(audio)
@@ -120,7 +126,15 @@ def main(audio):
 def save_answers(*args):
     answers_dict = {oral_health_assessment_form[i]: args[i] for i in range(len(oral_health_assessment_form))}
     print("Saved answers:", answers_dict)
-    return f"Saved answers: {answers_dict}"
+    
+    # Insert data into Supabase
+    try:
+        response = supabase.table('oral_health_assessments').insert(answers_dict).execute()
+        print("Data inserted into Supabase:", response.data)
+        return f"Saved answers: {answers_dict}"
+    except Exception as e:
+        print(f"Error inserting data into Supabase: {e}")
+        return f"Error saving answers: {e}"
 
 # Create the Gradio interface
 with gr.Blocks() as demo: