fix: medical history

app.py

@@ -5,6 +5,7 @@ import os
 from supabase import create_client, Client
 from datetime import datetime
 import csv
+from typing import Optional
 
 # Add your AssemblyAI API key as Environment Variable
 aai.settings.api_key = os.environ['Assembly']
@@ -18,9 +19,10 @@ question_answerer = pipeline("question-answering", model='distilbert-base-cased-
 questions = [
     "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",
+    "What is the chief complaint regarding the patient's oral health?",
+    "Can you provide any relevant Medication history for the patient? If there is none, just say the word 'none', else elaborate",
     "Can you provide any relevant Dental history for the patient? If there is none, just say the word 'none', else elaborate",
-    "Give me about the clinical findings listed",
+    "Please give all the clinical findings which were listed",
     "What treatment plan do you recommend?"
 ]
 
@@ -52,12 +54,12 @@ oral_health_assessment_form = [
 ]
 
 # Function to generate answers for the questions
-def generate_answer(question, context):
+def generate_answer(question: str, context: str) -> str:
     result = question_answerer(question=question, context=context)
     return result['answer']
 
 # Function to handle audio recording and transcription
-def transcribe_audio(audio_path):
+def transcribe_audio(audio_path: str) -> str:
     print(f"Received audio file at: {audio_path}")
     
     # Check if the file exists and is not empty
@@ -89,7 +91,7 @@ def transcribe_audio(audio_path):
         return str(e)
 
 # Function to fill in the answers for the text boxes
-def fill_textboxes(context):
+def fill_textboxes(context: str) -> dict:
     answers = []
     for question in questions:
         answer = generate_answer(question, context)
@@ -100,10 +102,10 @@ def fill_textboxes(context):
         "Age": answers[0],
         "Gender": answers[1],
         "Chief complaint": answers[2],
-        "Medical history": "none",  # Medical history is not part of the questions
-        "Dental history": answers[3],
-        "Clinical Findings": answers[4],
-        "Treatment plan": answers[5],
+        "Medical history": answers[3],
+        "Dental history": answers[4],
+        "Clinical Findings": answers[5],
+        "Treatment plan": answers[6],
         "Referred to": ""
     }
 
@@ -111,19 +113,19 @@ def fill_textboxes(context):
 supabase: Client = create_client(url, key)
 
 # Main Gradio app function
-def main(audio, doctor_name, location):
+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
     
     answers = fill_textboxes(context)
-    answers_list = [doctor_name, location] + [""]  # Initial patient name field empty
+    answers_list = [doctor_name, location, ""]  # Initial patient name field empty
     answers_list += [answers.get(field, "") for field in form_fields]
     
     return answers_list
 
-def save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to):
+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:
     current_datetime = datetime.now().isoformat()
     answers_dict = {
         "Doctor’s Name": doctor_name,
@@ -151,7 +153,7 @@ def save_answers(doctor_name, location, patient_name, age, gender, chief_complai
         return f"Error saving answers: {e}"
 
 # Function to download table as CSV
-def download_table_to_csv():
+def download_table_to_csv() -> Optional[str]:
     # Fetch data from Supabase table
     response = supabase.table("oral_health_assessments").select("*").execute()
     
@@ -182,7 +184,7 @@ def download_table_to_csv():
     print("Downloaded table oral_health_assessments")
     return csv_file
 
-def gradio_download():
+def gradio_download() -> Optional[str]:
     file_path = download_table_to_csv()
     if file_path:
         return file_path
@@ -220,19 +222,19 @@ with gr.Blocks() as demo:
                     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)
             
-            def enable_transcribe_button(audio_path):
+            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, doctor_name, location):
+            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
                 
                 answers = fill_textboxes(context)
-                answers_list = [doctor_name, location] + [""]  # Initial patient name field empty
+                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
                 
@@ -243,7 +245,7 @@ with gr.Blocks() as demo:
             save_button = gr.Button("Save Form")
             save_output = gr.HTML(label="Save Output")
             
-            def handle_submission(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):
                 return save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to)
             
             save_button.click(fn=handle_submission, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred], outputs=save_output)