Update app.py

app.py

@@ -3,6 +3,7 @@ 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"
@@ -122,47 +123,23 @@ def main(audio):
     # 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}" />'})
     
-    # Add JavaScript to capture changes and submit
-    custom_js = """
-    <script>
-    function getTableData() {
-        var table = document.querySelector('table');
-        var data = [];
-        for (var i = 1, row; row = table.rows[i]; i++) {
-            var question = row.cells[0].innerText;
-            var answer = row.cells[1].querySelector('input').value;
-            data.push({question: question, answer: answer});
-        }
-        return data;
-    }
+    return table_html
 
-    function submitTableData() {
-        var data = getTableData();
-        fetch('/submit_table_data', {
-            method: 'POST',
-            headers: {
-                'Content-Type': 'application/json',
-            },
-            body: JSON.stringify(data),
-        })
-        .then(response => response.json())
-        .then(data => {
-            document.querySelector('#submit_status').innerText = 'Data submitted successfully!';
-        })
-        .catch((error) => {
-            console.error('Error:', error);
-        });
-    }
+# 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 = {}
 
-    document.querySelector('#submit_button').addEventListener('click', submitTableData);
-    </script>
-    """
+    # 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
     
-    return table_html + custom_js
-
-# Function to handle submit button and store table data in dictionary
-def submit_table_data(data):
-    tabledata = {item['question']: item['answer'] for item in data}
     print("Submitted Table Data: ", tabledata)
     return f"Data submitted successfully! {tabledata}"
 
@@ -175,11 +152,10 @@ with gr.Blocks() as demo:
             output_html = gr.HTML(label="Assessment Form")
         with gr.Column():
             transcribe_button = gr.Button("Transcribe and Generate Form")
-            submit_button = gr.Button("Submit", elem_id="submit_button")
-            submit_status = gr.Textbox(label="Submit Status", elem_id="submit_status")
+            submit_button = gr.Button("Submit")
 
     transcribe_button.click(fn=main, inputs=audio_input, outputs=output_html)
-    demo.add_event_listener('submit_table_data', fn=submit_table_data, inputs=None, outputs=submit_status)
+    submit_button.click(fn=submit_table_data, inputs=output_html, outputs=gr.Textbox(label="Submit Status"))
 
 # Launch the app
 demo.launch()