modified

app.py

@@ -112,9 +112,7 @@ def chatbot_ui():
         # Layout for chatbot and input box alignment
         with gr.Row():
             with gr.Column(scale=3):  # Main column for chatbot
-                chatbot = gr.Chatbot(label=None, elem_id="chatbot").style(
-                    align_self_left=True,  # Align bot's response to the left
-                )
+                chatbot = gr.Chatbot(label="Responses", elem_id="chatbot")
                 user_input = gr.Textbox(
                     label="Ask a health-related question",
                     placeholder="Describe your symptoms...",
@@ -131,20 +129,20 @@ def chatbot_ui():
         def handle_submit(user_query, image, history):
             logger.info("User submitted a query.")
             response, audio = customLLMBot(user_query, image, history)
-            return response, audio, "", history
+            return response, audio, None, history  # Clear the image after submission
 
         # Submit on pressing Enter key
         user_input.submit(
             handle_submit,
             inputs=[user_input, uploaded_image, chat_history],
-            outputs=[chatbot, audio_output, user_input, chat_history],
+            outputs=[chatbot, audio_output, uploaded_image, chat_history],
         )
 
         # Submit on button click
         submit_btn.click(
             handle_submit,
             inputs=[user_input, uploaded_image, chat_history],
-            outputs=[chatbot, audio_output, user_input, chat_history],
+            outputs=[chatbot, audio_output, uploaded_image, chat_history],
         )
 
         # Action for clearing all fields