Update app.py

app.py

@@ -25,20 +25,6 @@ questions = [
     "Please give all the clinical findings which were listed"
 ]
 
-# List of form fields in the correct order
-form_fields = [
-    "Age",
-    "Gender",
-    "Chief complaint",
-    "Medical history",
-    "Dental history",
-    "Clinical Findings",
-    "Referred to",
-    "Treatment plan",
-    "Calculus",
-    "Stains"
-]
-
 # Oral Health Assessment Form
 oral_health_assessment_form = [
     "Doctor’s Name",
@@ -65,7 +51,6 @@ def generate_answer(question: str, context: str) -> str:
 def transcribe_audio(audio_path: str) -> str:
     print(f"Received audio file at: {audio_path}")
     
-    # Check if the file exists and is not empty
     if not os.path.exists(audio_path):
         return "Error: Audio file does not exist."
     
@@ -73,14 +58,11 @@ def transcribe_audio(audio_path: str) -> str:
         return "Error: Audio file is empty."
     
     try:
-        # Transcribe the audio file using AssemblyAI
         transcriber = aai.Transcriber()
         print("Starting transcription...")
         transcript = transcriber.transcribe(audio_path)
-        
         print("Transcription process completed.")
         
-        # Handle the transcription result
         if transcript.status == aai.TranscriptStatus.error:
             print(f"Error during transcription: {transcript.error}")
             return transcript.error
@@ -94,57 +76,44 @@ def transcribe_audio(audio_path: str) -> str:
         return str(e)
 
 # Function to fill in the answers for the text boxes
-def fill_textboxes(context: str) -> dict:
+def fill_textboxes(context: str) -> list:
     answers = []
     for question in questions:
         answer = generate_answer(question, context)
         answers.append(answer)
     
-    # Map answers to form fields in the correct order
-    return {
-        "Age": answers[0] if len(answers) > 0 else "",
-        "Gender": answers[1] if len(answers) > 1 else "",
-        "Chief complaint": answers[2] if len(answers) > 2 else "",
-        "Medical history": answers[3] if len(answers) > 3 else "",
-        "Dental history": answers[4] if len(answers) > 4 else "",
-        "Clinical Findings": answers[5] if len(answers) > 5 else "",
-        "Referred to": "",  # Default value
-        "Treatment plan": "",  # Default value
-        "Calculus": "",  # Default value
-        "Stains": "",  # Default value
-        "Doctor’s Name": "",  # Default value
-        "Location": ""  # Default value
-    }
+    # Map answers to form fields in the correct order and return as a list
+    return [
+        answers[0] if len(answers) > 0 else "",
+        answers[1] if len(answers) > 1 else "",
+        answers[2] if len(answers) > 2 else "",
+        answers[3] if len(answers) > 3 else "",
+        answers[4] if len(answers) > 4 else "",
+        answers[5] if len(answers) > 5 else "",
+        "",  # Referred to
+        "",  # Treatment plan
+        "",  # Calculus
+        "",  # Stains
+        "",  # Doctor’s Name
+        "",  # Location
+    ]
 
 # Supabase configuration
 supabase: Client = create_client(url, key)
 
 # Main Gradio app function
-def handle_transcription(audio: str, doctor_name: str, location: str) -> dict:
+def handle_transcription(audio: str, doctor_name: str, location: str) -> list:
     context = transcribe_audio(audio)
     if "Error" in context:
-        return {field: context for field in form_fields}
+        return [context] * 13  # Return error for all fields
     
     answers = fill_textboxes(context)
-    answers.update({
-        "Doctor’s Name": doctor_name,
-        "Location": location
-    })
     
-    return {
-        "Age": answers.get("Age", ""),
-        "Gender": answers.get("Gender", ""),
-        "Chief complaint": answers.get("Chief complaint", ""),
-        "Medical history": answers.get("Medical history", ""),
-        "Dental history": answers.get("Dental history", ""),
-        "Clinical Findings": answers.get("Clinical Findings", ""),
-        "Referred to": answers.get("Referred to", ""),
-        "Treatment plan": answers.get("Treatment plan", ""),
-        "Calculus": answers.get("Calculus", ""),
-        "Stains": answers.get("Stains", ""),
-        "Doctor’s Name": doctor_name,
-        "Location": location
-    }
+    # Insert Doctor’s Name and Location in the appropriate fields
+    answers[-2] = doctor_name
+    answers[-1] = location
+    
+    return answers
 
 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:
     current_datetime = datetime.now().isoformat()
@@ -166,7 +135,6 @@ def save_answers(doctor_name: str, location: str, patient_name: str, age: str, g
     }
     print("Saved answers:", answers_dict)
     
-    # Insert data into Supabase
     try:
         response = supabase.table('oral_health_assessments').insert(answers_dict).execute()
         print("Data inserted into Supabase:", response.data)
@@ -177,28 +145,21 @@ def save_answers(doctor_name: str, location: str, patient_name: str, age: str, g
 
 # Function to download table as CSV
 def download_table_to_csv() -> Optional[str]:
-    # Fetch data from Supabase table
     response = supabase.table("oral_health_assessments").select("*").execute()
     
-    # Check if data is available
     if not response.data:
         print("No data found in the table.")
         return None
 
     data = response.data
-
-    # Prepare CSV data
     csv_data = []
 
-    # Add header row
     if len(data) > 0:
         csv_data.append(data[0].keys())
 
-    # Add data rows
     for row in data:
         csv_data.append(row.values())
 
-    # Save CSV data to file (replace 'your_table.csv' with desired filename)
     csv_file = "your_table.csv"
     with open(csv_file, "w", newline='') as f:
         writer = csv.writer(f)
@@ -218,7 +179,6 @@ with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
     
     with gr.Tabs() as tabs:
-        # Default tab for Doctor's Name and Location
         with gr.Tab("Doctor Info"):
             doctor_name_input = gr.Textbox(label="Doctor's Name", interactive=True)
             location_input = gr.Textbox(label="Location", interactive=True)
@@ -230,7 +190,6 @@ with gr.Blocks() as demo:
             
             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"):
             audio_input = gr.Audio(type="filepath", label="Record your audio", elem_id="audio_input")
             transcribe_button = gr.Button("Transcribe and Generate Form", elem_id="transcribe_button", interactive=False)
@@ -259,12 +218,25 @@ with gr.Blocks() as demo:
             oha_output = gr.Textbox(label="OHA Output", value="", interactive=False)
             save_button = gr.Button("Save to Supabase", elem_id="save_button", interactive=True)
             
-            transcribe_button.click(fn=handle_transcription, inputs=[audio_input, doctor_name_display, location_display], outputs=textboxes_left + textboxes_right + [oha_output])
-            save_button.click(fn=save_answers, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right, outputs=oha_output)
+            # Update the transcription and form fields when the transcribe button is clicked
+            transcribe_button.click(
+                fn=handle_transcription, 
+                inputs=[audio_input, doctor_name_input, location_input], 
+                outputs=textboxes_left + textboxes_right + [doctor_name_display, location_display]
+            )
+            
+            # Save the form data to Supabase when the save button is clicked
+            save_button.click(
+                fn=save_answers, 
+                inputs=[doctor_name_input, location_input] + textboxes_left + textboxes_right, 
+                outputs=[oha_output]
+            )
         
         with gr.Tab("Download Data"):
             download_button = gr.Button("Download CSV")
-            download_output = gr.File()
-            download_button.click(fn=gradio_download, outputs=download_output)
+            download_output = gr.File(label="Download the CSV File", interactive=False)
+            
+            download_button.click(fn=gradio_download, inputs=[], outputs=download_output)
 
-demo.launch()
+# Launch the Gradio app
+demo.launch(share=True)