Update app.py

app.py

@@ -97,7 +97,7 @@ def main(audio, doctor_name, location):
     answers.insert(0, location)  # Add location
     answers.insert(0, doctor_name)  # Add doctor name
     
-    return [doctor_name, location] + answers
+    return answers
 
 def save_answers(*args):
     current_datetime = datetime.now().isoformat()
@@ -164,7 +164,10 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit")
             info_output = gr.HTML(label="Submitted Info")
             
-            submit_button.click(fn=lambda name, loc: f"Doctor's Name: {name}<br>Location: {loc}", inputs=[doctor_name_input, location_input], outputs=info_output)
+            def submit_info(name, loc):
+                return f"Doctor's Name: {name}<br>Location: {loc}"
+            
+            submit_button.click(fn=submit_info, inputs=[doctor_name_input, location_input], outputs=info_output)
         
         # Second tab for OHA Form
         with gr.Tab("OHA Form"):
@@ -173,15 +176,29 @@ with gr.Blocks() as demo:
             
             with gr.Row(elem_id="textboxes_row"):
                 with gr.Column():
-                    textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=False) for i in range(2, len(oral_health_assessment_form)//2)]
+                    doctor_name_display = gr.Textbox(label="Doctor’s Name", value="", interactive=False)
+                    location_display = gr.Textbox(label="Location", value="", interactive=False)
+                    textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(2, len(oral_health_assessment_form)//2)]
                 with gr.Column():
-                    textboxes_right = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=False) for i in range(len(oral_health_assessment_form)//2, len(oral_health_assessment_form)-1)]
+                    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)-1)]
                     dropdown_referred = gr.Dropdown(choices=["NONE","ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS"], label="Referred to", interactive=True)
             
-            transcribe_button.click(fn=main, inputs=[audio_input, doctor_name_input, location_input], outputs=textboxes_left + textboxes_right + [doctor_name_input, location_input])
+            def update_textboxes(audio, doctor_name, location):
+                context = transcribe_audio(audio)
+                
+                if "Error" in context:
+                    return [context] * (len(oral_health_assessment_form) - 2)  # Adjust for the number of fields
+                
+                answers = fill_textboxes(context)
+                answers.insert(0, location)  # Add location
+                answers.insert(0, doctor_name)  # Add doctor name
+                
+                return [doctor_name, location] + answers
+
+            transcribe_button.click(fn=update_textboxes, inputs=[audio_input, doctor_name_input, location_input], outputs=[doctor_name_display, location_display] + textboxes_left + textboxes_right)
             submit_button = gr.Button("Submit", elem_id="submit_button")
             output_html = gr.HTML(label="Submitted Answers")
-            submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
+            submit_button.click(fn=save_answers, inputs=[doctor_name_display, location_display] + textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
         
         # Third tab for CSV download
         with gr.Tab("Download CSV"):