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"
@@ -97,13 +96,13 @@ def transcribe_audio(audio_path):
         print(f"Exception occurred: {e}")
         return str(e)
 
-# Function to fill in the DataFrame with answers
-def fill_dataframe(context):
-    data = []
+# Function to fill in the answers for the text boxes
+def fill_textboxes(context):
+    answers = []
     for question in questions:
         answer = generate_answer(question, context)
-        data.append({"Question": question, "Answer": answer})
-    return pd.DataFrame(data)
+        answers.append(answer)
+    return answers
 
 # Main Gradio app function
 def main(audio):
@@ -112,36 +111,16 @@ def main(audio):
     if "Error" in context:
         return context
     
-    df = fill_dataframe(context)
+    answers = fill_textboxes(context)
+    answers.insert(0, "")  # for "Child's Name"
+    answers.insert(0, "Dr. Charles Xavier")  # for "Doctor’s Name"
     
-    # Add doctor's and patient's name to the beginning of the DataFrame
-    df = pd.concat([pd.DataFrame({"Question": ["Doctor’s Name", "Child’s Name"], "Answer": ["Dr. Charles Xavier", ""]}), df])
-    
-    # Add a title to the DataFrame
-    df['Question'] = oral_health_assessment_form
-
-    # Convert DataFrame to HTML table with editable text boxes
-    table_html = df.to_html(index=False, escape=False, formatters={"Answer": lambda x: f'<input type="text" value="{x}" />'})
-    
-    return table_html
+    return answers
 
-# 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
-    
-    print("Submitted Table Data: ", tabledata)
-    return f"Data submitted successfully! {tabledata}"
+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 "Answers saved successfully."
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
@@ -150,12 +129,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             audio_input = gr.Audio(type="filepath", label="Record your audio")
             output_html = gr.HTML(label="Assessment Form")
-        with gr.Column():
             transcribe_button = gr.Button("Transcribe and Generate Form")
+            
+        with gr.Column():
+            textboxes = [gr.Textbox(label=question, value="", interactive=True) for question in oral_health_assessment_form]
             submit_button = gr.Button("Submit")
-
-    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"))
+    
+    transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes)
+    submit_button.click(fn=save_answers, inputs=textboxes, outputs=output_html)
 
 # Launch the app
 demo.launch()