Update app.py

app.py

@@ -3,7 +3,6 @@ import assemblyai as aai
 from transformers import pipeline
 import pandas as pd
 import os
-from bs4 import BeautifulSoup
 
 # Replace with your AssemblyAI API key
 aai.settings.api_key = "62acec891bb04c339ec059b738bedac6"
@@ -126,20 +125,8 @@ def main(audio):
     return table_html
 
 # Function to handle submit button and store table data in dictionary
-def submit_table_data(html_table):
-    # Parse the HTML table using BeautifulSoup
-    soup = BeautifulSoup(html_table, 'html.parser')
-    tabledata = {}
-
-    # Iterate through the rows of the table
-    rows = soup.find_all('tr')
-    for row in rows:
-        cells = row.find_all('td')
-        if len(cells) == 2:
-            question = cells[0].text.strip()
-            answer = cells[1].find('input')['value']
-            tabledata[question] = answer
-    
+def submit_table_data(answers):
+    tabledata = dict(zip(oral_health_assessment_form, answers))
     print("Submitted Table Data: ", tabledata)
     return f"Data submitted successfully! {tabledata}"
 
@@ -154,8 +141,10 @@ with gr.Blocks() as demo:
             transcribe_button = gr.Button("Transcribe and Generate Form")
             submit_button = gr.Button("Submit")
 
+    output_textboxes = [gr.Textbox(label=question) for question in oral_health_assessment_form]
+    
     transcribe_button.click(fn=main, inputs=audio_input, outputs=output_html)
-    submit_button.click(fn=submit_table_data, inputs=output_html, outputs=gr.Textbox(label="Submit Status"))
+    submit_button.click(fn=submit_table_data, inputs=output_textboxes, outputs=gr.Textbox(label="Submit Status"))
 
 # Launch the app
 demo.launch()