Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import pipeline
 import pandas as pd
 import os
 
+# Replace with your AssemblyAI API key
 aai.settings.api_key = "62acec891bb04c339ec059b738bedac6"
 
 # Initialize question answering pipeline
@@ -117,13 +118,23 @@ def main(audio):
     
     # 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 df
+    return table_html
 
-# Function to handle form submission
-def submit_form(dataframe):
-    data_dict = dataframe.to_dict(orient='records')
-    print("Submitted Data:", data_dict)
+# Function to handle submit button and store table data in dictionary
+def submit_table_data(html_table):
+    # Extract input values from HTML table
+    tabledata = {}
+    for line in html_table.split('\n'):
+        if '<td>' in line:
+            question = line.split('<td>')[1].split('</td>')[0]
+            answer = line.split('value="')[1].split('"')[0]
+            tabledata[question] = answer
+    
+    print("Submitted Table Data: ", tabledata)
     return "Data submitted successfully!"
 
 # Create the Gradio interface
@@ -132,13 +143,13 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             audio_input = gr.Audio(type="filepath", label="Record your audio")
-            output_df = gr.Dataframe(headers=oral_health_assessment_form, label="Assessment Form", editable=True)
-            submit_button = gr.Button("Submit")
+            output_html = gr.HTML(label="Assessment Form")
         with gr.Column():
             transcribe_button = gr.Button("Transcribe and Generate Form")
+            submit_button = gr.Button("Submit")
 
-    transcribe_button.click(fn=main, inputs=audio_input, outputs=output_df)
-    submit_button.click(fn=submit_form, inputs=output_df, outputs="text")
+    transcribe_button.click(fn=main, inputs=audio_input, outputs=output_html)
+    submit_button.click(fn=submit_table_data, inputs=output_html, outputs="text")
 
 # Launch the app
 demo.launch()