dropdown for stains and calculus

app.py

@@ -22,8 +22,8 @@ questions = [
     "What is the chief complaint regarding the patient's oral health?",
     "List the Medical history mentioned",
     "Give the Dental history in detail",
-    "Please give all the clinical findings which were listed",
-    "What treatment plan is recommended?"
+    "Please give all the clinical findings which were listed"
+    # "What treatment plan is recommended?" -- Removed as we are using a dropdown for this
 ]
 
 # List of form fields in the correct order
@@ -34,7 +34,7 @@ form_fields = [
     "Medical history",
     "Dental history",
     "Clinical Findings",
-    "Treatment plan",
+    # "Treatment plan", -- Removed
     "Referred to"
 ]
 
@@ -50,7 +50,9 @@ oral_health_assessment_form = [
     "Dental history",
     "Clinical Findings",
     "Treatment plan",
-    "Referred to"
+    "Referred to",
+    "Options",
+    "Stains"
 ]
 
 # Function to generate answers for the questions
@@ -105,7 +107,6 @@ def fill_textboxes(context: str) -> dict:
         "Medical history": answers[3],
         "Dental history": answers[4],
         "Clinical Findings": answers[5],
-        "Treatment plan": answers[6],
         "Referred to": ""
     }
 
@@ -117,15 +118,17 @@ def main(audio: str, doctor_name: str, location: str) -> list:
     context = transcribe_audio(audio)
     
     if "Error" in context:
-        return [context] * (len(oral_health_assessment_form) - 2)  # Adjust for the number of fields
+        return [context] * (len(oral_health_assessment_form) - 5)  # 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]
+    answers_list.append("")  # Default for "Options"
+    answers_list.append("")  # Default for "Stains"
     
     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) -> str:
+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, options: str, stains: str) -> str:
     current_datetime = datetime.now().isoformat()
     answers_dict = {
         "Doctor’s Name": doctor_name,
@@ -139,6 +142,8 @@ def save_answers(doctor_name: str, location: str, patient_name: str, age: str, g
         "Clinical Findings": clinical_findings,
         "Treatment plan": treatment_plan,
         "Referred to": referred_to,
+        "Options": options,
+        "Stains": stains,
         "Submission Date and Time": current_datetime
     }
     print("Saved answers:", answers_dict)
@@ -219,36 +224,46 @@ with gr.Blocks() as demo:
                     patient_name_input = gr.Textbox(label="Patient’s Name", value="", interactive=True)
                     textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(3, len(oral_health_assessment_form)//2)]
                 with gr.Column():
-                    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)]
+                    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)-5)]
                     dropdown_referred = gr.Dropdown(choices=["NONE","ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS"], label="Referred to", interactive=True)
+                    dropdown_options = gr.Dropdown(choices=["+", "++", "+++"], label="Options", interactive=True)
+                    dropdown_stains = gr.Dropdown(choices=["+", "++", "+++"], label="Stains", interactive=True)
+                    dropdown_treatment_plan = gr.Dropdown(choices=["Scaling", "Filling", "Pulp therapy/RCT", "Extraction", "Medication", "Referral"], label="Treatment plan", interactive=True)
             
             def enable_transcribe_button(audio_path: str):
                 return gr.update(interactive=True)
 
             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) - 3)  # Adjust for the number of fields
+                    return [context] * (len(oral_health_assessment_form) - 5)  # 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.append(answers.get("Referred to", ""))  # Ensure "Referred to" is included
-                
                 return answers_list
 
-            transcribe_button.click(fn=update_textboxes, inputs=[audio_input, doctor_name_input, location_input], outputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred])
+            transcribe_button.click(
+                fn=update_textboxes, 
+                inputs=[audio_input, doctor_name_input, location_input], 
+                outputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred, dropdown_treatment_plan, dropdown_options, dropdown_stains]
+            )
             
             save_button = gr.Button("Save Form")
             save_output = gr.HTML(label="Save Output")
             
-            def handle_submission(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):
-                return save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to)
+            def handle_submission(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, options: str, stains: str):
+                return save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to, options, stains)
             
-            save_button.click(fn=handle_submission, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred], outputs=save_output)
+            save_button.click(
+                fn=handle_submission, 
+                inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred, dropdown_treatment_plan, dropdown_options, dropdown_stains], 
+                outputs=save_output
+            )
         
         # Third tab for Save and Download
         with gr.Tab("Download Data"):