Update app.py

app.py

@@ -62,7 +62,6 @@ oral_health_assessment_form = [
     "Any other finding",
     "Treatment plan",
 ]
-
 # Function to generate answers for the questions
 def generate_answer(question, context):
     result = question_answerer(question=question, context=context)
@@ -142,6 +141,10 @@ def save_answers(*args):
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
+    
+    # Add HTML box to show the status
+    status_box = gr.HTML("<div id='status_box' style='width: 100px; height: 30px; background-color: red;'></div>")
+    
     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")
     
@@ -154,8 +157,16 @@ with gr.Blocks() as demo:
     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)
+    # Update status box color based on audio input
+    def update_status(audio):
+        if audio:
+            status_box_value = "<div id='status_box' style='width: 100px; height: 30px; background-color: green;'></div>"
+        else:
+            status_box_value = "<div id='status_box' style='width: 100px; height: 30px; background-color: red;'></div>"
+        return status_box_value, main(audio)
+
+    transcribe_button.click(fn=update_status, inputs=audio_input, outputs=[status_box, textboxes_left + textboxes_right])
     submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right, outputs=output_html)
 
 # Launch the app
-demo.launch(share=True)
+demo.launch(share=True)