modified

app.py

@@ -39,7 +39,7 @@ def customLLMBot(user_input, uploaded_image, chat_history):
         logger.info("Processing input...")
 
         # Append user input to the chat history
-        chat_history.append((user_input, None))
+        chat_history.append(("user", user_input))
 
         if uploaded_image is not None:
             # Encode the image to base64
@@ -84,7 +84,7 @@ def customLLMBot(user_input, uploaded_image, chat_history):
         logger.debug(f"LLM reply: {LLM_reply}")
 
         # Append the bot's response to the chat history
-        chat_history[-1] = (user_input, LLM_reply)
+        chat_history.append(("bot", LLM_reply))
 
         # Generate audio for response
         audio_file = f"response_{uuid.uuid4().hex}.mp3"
@@ -98,23 +98,22 @@ def customLLMBot(user_input, uploaded_image, chat_history):
     except Exception as e:
         # Handle errors gracefully
         logger.error(f"Error in customLLMBot function: {e}")
-        return [(user_input, f"An error occurred: {e}")], None
+        return [(("user", user_input or "Image uploaded"), ("bot", f"An error occurred: {e}"))], None
 
 
 # Gradio Interface
 def chatbot_ui():
-    chat_history = []  # Initialize empty chat history
-
     with gr.Blocks() as demo:
         gr.Markdown("# Healthcare Chatbot Doctor")
 
+        # State for user chat history
+        chat_history = gr.State([])
+
         # 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_right=True,
-                    align_self_left=False,
-                    match=False,
+                    align_self_left=True,  # Align bot's response to the left
                 )
                 user_input = gr.Textbox(
                     label="Ask a health-related question",
@@ -129,30 +128,30 @@ def chatbot_ui():
                 audio_output = gr.Audio(label="Audio Response")
 
         # Define actions
-        def handle_submit(user_query, image):
+        def handle_submit(user_query, image, history):
             logger.info("User submitted a query.")
-            response, audio = customLLMBot(user_query, image, chat_history)
-            return response, audio, ""
+            response, audio = customLLMBot(user_query, image, history)
+            return response, audio, "", history
 
         # Submit on pressing Enter key
         user_input.submit(
             handle_submit,
-            inputs=[user_input, uploaded_image],
-            outputs=[chatbot, audio_output, user_input],
+            inputs=[user_input, uploaded_image, chat_history],
+            outputs=[chatbot, audio_output, user_input, chat_history],
         )
 
         # Submit on button click
         submit_btn.click(
             handle_submit,
-            inputs=[user_input, uploaded_image],
-            outputs=[chatbot, audio_output, user_input],
+            inputs=[user_input, uploaded_image, chat_history],
+            outputs=[chatbot, audio_output, user_input, chat_history],
         )
 
         # Action for clearing all fields
         clear_btn.click(
-            lambda: ([], "", None, None),
+            lambda: ([], "", None, []),
             inputs=[],
-            outputs=[chatbot, user_input, uploaded_image, audio_output],
+            outputs=[chatbot, user_input, uploaded_image, chat_history],
         )
 
     return demo