Update app.py

app.py

@@ -16,6 +16,7 @@ question_answerer = pipeline("question-answering", model='distilbert-base-cased-
 
 # List of questions
 questions = [
+    "Which grade is the patient studying?",
     "How old is the patient?",
     "What is the gender?",
     "What is the chief complaint regarding the patient's oral health? If there is none, just say the word 'none' else elaborate",
@@ -29,13 +30,14 @@ questions = [
 oral_health_assessment_form = [
     "Doctor’s Name",
     "Child’s Name",
+    "Grade",
     "Age",
     "Gender",
     "Chief complaint",
     "Medical history",
     "Dental history",
     "Clinical Findings",
-    "miscellaneous findings",
+    "Any other finding",
     "Treatment plan",
     "Referred to"
 ]
@@ -93,13 +95,13 @@ def main(audio):
     context = transcribe_audio(audio)
     
     if "Error" in context:
-        return [context] * len(oral_health_assessment_form)
+        return [context] * (len(oral_health_assessment_form) - 1)
     
     answers = fill_textboxes(context)
     answers.insert(0, "")  # for "Child's Name"
     answers.insert(0, "Dr.Alvi")  # for "Doctor’s Name"
     
-    return answers + [""]  # Add empty string for "Referred to"
+    return answers
 
 def save_answers(*args):
     current_datetime = datetime.now().isoformat()
@@ -164,23 +166,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             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_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))]
-    
-    referred_to = gr.Dropdown(label="Referred to", choices=[
-        "ORAL MEDICINE & RADIOLOGY",
-        "PERIODONTICS",
-        "ORAL SURGERY",
-        "CONSERVATIVE AND ENDODONTICS",
-        "PROSTHODONTICS",
-        "PEDODONTICS",
-        "ORTHODONTICS"
-    ], value="", interactive=True)
+            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=["ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS", "none"], label="Referred to", interactive=True)
     
     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 + [referred_to])
-    submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [referred_to], outputs=output_html)
+    transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes_left + textboxes_right)
+    submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
 
     download_button = gr.Button("Download Table as CSV")
     download_csv_output = gr.File(label="Download CSV")
@@ -188,4 +181,4 @@ with gr.Blocks() as demo:
     download_button.click(fn=gradio_download, inputs=[], outputs=download_csv_output)
 
 # Launch the app
-demo.launch(share=True)
+demo.launch()