Update app.py

app.py

@@ -356,7 +356,6 @@ with gr.Blocks(css=custom_css) as app:
         
         submit_chatbot = gr.Button(value="Submit Your Message", variant="primary", elem_id="submit-chatbot", icon="fa-paper-plane")
         
-        # Adding the loading prop for the chatbot button
         chatbot = gr.Chatbot(label="Chat History", elem_id="chat-history", show_label=True)
         sentiment = gr.Textbox(label="Detected Sentiment")
         emotion = gr.Textbox(label="Detected Emotion")
@@ -365,13 +364,13 @@ with gr.Blocks(css=custom_css) as app:
         professionals = gr.DataFrame(label="Nearby Health Professionals", headers=["Name", "Address"])
         map_html = gr.HTML(label="Interactive Map")
 
+        # Here we add the loading state on the button itself
         submit_chatbot.click(
             app_function_chatbot,
             inputs=[user_input, location, query, chatbot],
-            outputs=[chatbot, sentiment, emotion, suggestions_markdown, professionals, map_html],
-            loading=True  # Enable loading spinner on button click
-        )
-        
+            outputs=[chatbot, sentiment, emotion, suggestions_markdown, professionals, map_html]
+        ).then(submit_chatbot.update(loading=True))
+
     with gr.Tab("Disease Prediction"):
         symptom1 = gr.Dropdown(X_train.columns.tolist(), label="Select Symptom 1")
         symptom2 = gr.Dropdown(X_train.columns.tolist(), label="Select Symptom 2")
@@ -381,16 +380,14 @@ with gr.Blocks(css=custom_css) as app:
         
         submit_disease = gr.Button(value="Predict Disease", variant="primary", icon="fa-stethoscope")
         
-        # Adding the loading prop for the disease prediction button
         disease_prediction_result = gr.Markdown(label="Predicted Diseases")  # Use Markdown for predictions
 
         submit_disease.click(
             lambda symptom1, symptom2, symptom3, symptom4, symptom5: predict_disease(
                 [symptom1, symptom2, symptom3, symptom4, symptom5]),
             inputs=[symptom1, symptom2, symptom3, symptom4, symptom5],
-            outputs=disease_prediction_result,
-            loading=True  # Enable loading spinner on button click
-        )
+            outputs=disease_prediction_result
+        ).then(submit_disease.update(loading=True))
 
 # Launch the Gradio application
 app.launch()