Update app.py

app.py

@@ -109,7 +109,7 @@ def main(audio):
     context = transcribe_audio(audio)
     
     if "Error" in context:
-        return context
+        return [context] * len(oral_health_assessment_form)
     
     answers = fill_textboxes(context)
     answers.insert(0, "")  # for "Child's Name"
@@ -120,23 +120,25 @@ 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 "Answers saved successfully."
+    return f"Saved answers: {answers_dict}"
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Audio Transcription and Question Answering App")
-    with gr.Row():
+    audio_input = gr.Audio(type="filepath", label="Record your audio", elem_id="audio_input")
+    transcribe_button = gr.Button("Transcribe and Generate Form", elem_id="transcribe_button")
+    
+    with gr.Row(elem_id="textboxes_row"):
         with gr.Column():
-            audio_input = gr.Audio(type="filepath", label="Record your audio")
-            output_html = gr.HTML(label="Assessment Form")
-            transcribe_button = gr.Button("Transcribe and Generate Form")
-            
+            textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(0, len(oral_health_assessment_form)//2)]
         with gr.Column():
-            textboxes = [gr.Textbox(label=question, value="", interactive=True) for question in oral_health_assessment_form]
-            submit_button = gr.Button("Submit")
+            textboxes_right = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(len(oral_health_assessment_form)//2, len(oral_health_assessment_form))]
     
-    transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes)
-    submit_button.click(fn=save_answers, inputs=textboxes, outputs=output_html)
+    submit_button = gr.Button("Submit", elem_id="submit_button")
+    output_html = gr.HTML(label="Submitted Answers")
+
+    transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes_left + textboxes_right)
+    submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right, outputs=output_html)
 
 # Launch the app
 demo.launch()