Update app.py

app.py

@@ -33,7 +33,10 @@ form_fields = [
     "Medical history",
     "Dental history",
     "Clinical Findings",
-    "Referred to"
+    "Referred to",
+    "Treatment plan",
+    "Calculus",
+    "Stains"
 ]
 
 # Oral Health Assessment Form
@@ -126,9 +129,6 @@ def main(audio: str, doctor_name: str, location: str) -> list:
     answers = fill_textboxes(context)
     answers_list = [doctor_name, location, ""]  # Initial patient name field empty
     answers_list += [answers.get(field, "") for field in form_fields]
-    answers_list.append("")  # Default for "Calculus"
-    answers_list.append("")  # Default for "Stains"
-    answers_list.append("")  # Default for "Treatment plan"
     return answers_list
 
 def save_answers(doctor_name: str, location: str, patient_name: str, age: str, gender: str, chief_complaint: str, medical_history: str, dental_history: str, clinical_findings: str, treatment_plan: str, referred_to: str, Calculus: str, stains: str) -> str:
@@ -238,16 +238,15 @@ with gr.Blocks() as demo:
 
             audio_input.change(fn=enable_transcribe_button, inputs=audio_input, outputs=transcribe_button)
             
-           
             def update_textboxes(audio: str, doctor_name: str, location: str):
                 context = transcribe_audio(audio)
                 
                 if "Error" in context:
-                    return [context] * (len(oral_health_assessment_form) - 5)  # Adjust for the number of fields
+                    return [context] * len(oral_health_assessment_form)  # Adjust for the number of fields
                 
                 answers = fill_textboxes(context)
                 answers_list = [doctor_name, location, ""]  # Initial patient name field empty
-                answers_list += [answers.get(field, "") for field in form_fields[:-1]]  # Exclude "Referred to"
+                answers_list += [answers.get(field, "") for field in form_fields]
                 return answers_list
 
             transcribe_button.click(