Update app.py

app.py

@@ -396,6 +396,10 @@ def process_summary_with_stablemed(summary):
 # Main function to handle the Gradio interface logic
 
 def process_and_query(input_language=None, audio_input=None, image_input=None, text_input=None):
+    components['speech_to_text'].hide()
+    components['image_identification'].hide()
+    components['text_summarization'].hide()
+    components['results'].show()
     try:
         # Initialize the conditional variables
         combined_text = ""
@@ -448,23 +452,22 @@ def process_and_query(input_language=None, audio_input=None, image_input=None, t
 
 # Interface logic
 
-def clear(interface):
-    interface['language_selection'].reset()
-    interface['speech_to_text'].hide()
-    interface['image_identification'].hide()
-    interface['text_summarization'].hide()
-    interface['results'].hide()
+def clear(components):
+    components['language_selection'].reset()
+    components['speech_to_text'].hide()
+    components['image_identification'].hide()
+    components['text_summarization'].hide()
+    components['results'].hide()
 
-
-def on_language_change(language, interface):
+def on_language_change(language, components):
     if language:
-        interface['speech_to_text'].show()
-        interface['image_identification'].show()
-        interface['text_summarization'].show()
+        components['speech_to_text'].show()
+        components['image_identification'].show()
+        components['text_summarization'].show()
     else:
-        interface['speech_to_text'].hide()
-        interface['image_identification'].hide()
-        interface['text_summarization'].hide()
+        components['speech_to_text'].hide()
+        components['image_identification'].hide()
+        components['text_summarization'].hide()
 
 welcome_message = """
 # 👋🏻Welcome to ⚕🗣️😷MultiMed - Access Chat ⚕🗣️😷
@@ -587,39 +590,45 @@ languages = [
     "Standard Malay",
     "Zulu"
 ]
-
-
 with gr.Blocks(theme='ParityError/Anime') as iface:
+    components = {}
+
     with gr.Row() as language_selection:
         input_language = gr.Dropdown(languages, label="Select the language", value="English", interactive=True)
+        components['language_selection'] = language_selection
 
     with gr.Accordion("Speech to Text", open=False) as speech_to_text:
         audio_input = gr.Audio(label="Speak", type="filepath", sources="microphone")
         audio_output = gr.Markdown(label="Output text")
+        components['speech_to_text'] = speech_to_text
 
     with gr.Accordion("Image Identification", open=False) as image_identification:
         image_input = gr.Image(label="Upload image")
         image_output = gr.Markdown(label="Output text")
+        components['image_identification'] = image_identification
 
     with gr.Accordion("Text Summarization", open=False) as text_summarization:
         text_input = gr.Textbox(label="Input text", lines=5)
         text_output = gr.Markdown(label="Output text")
         text_button = gr.Button("Process text")
         hallucination_output = gr.Label(label="Hallucination Evaluation")
-        text_button.click(process_and_query, inputs=[input_language, audio_input, image_input, text_input], outputs=[text_output, hallucination_output])
+        components['text_summarization'] = text_summarization
 
     with gr.Row() as results:
         text_output = gr.Markdown()
         hallucination_output = gr.Label()
+        components['results'] = results
 
     clear_button = gr.Button("Clear")
-    clear_button.click(clear, inputs=[iface], outputs=[])
+    clear_button.click(clear, inputs=[components], outputs=[])
+
+    text_button.click(process_and_query, inputs=[input_language, audio_input, image_input, text_input, components], outputs=[text_output, hallucination_output])
 
 # Initially hide all blocks except language selection
-iface['speech_to_text'].hide()
-iface['image_identification'].hide()
-iface['text_summarization'].hide()
-iface['results'].hide()
+components['speech_to_text'].hide()
+components['image_identification'].hide()
+components['text_summarization'].hide()
+components['results'].hide()
 
 gr.Examples([
             ["What is the proper treatment for buccal herpes?"],