modified

app.py

@@ -19,7 +19,6 @@ logger.addHandler(console_handler)
 logger.addHandler(file_handler)
 
 # Initialize Groq Client
-
 client = Groq(api_key=os.getenv("GROQ_API_KEY_2"))
 
 # client = Groq(
@@ -39,10 +38,13 @@ def encode_image(uploaded_image):
         raise
 
 # Function to handle text and image inputs
-def customLLMBot(user_input, uploaded_image):
+def customLLMBot(user_input, uploaded_image, chat_history):
     try:
         logger.info("Processing input...")
 
+        # Append user input to the chat history
+        chat_history.append(("User", user_input))
+
         if uploaded_image is not None:
             # Encode the image to base64
             base64_image = encode_image(uploaded_image)
@@ -84,13 +86,17 @@ def customLLMBot(user_input, uploaded_image):
         LLM_reply = response.choices[0].message.content
         logger.debug(f"LLM reply: {LLM_reply}")
 
+        # Append the bot's response to the chat history
+        chat_history.append(("Bot", LLM_reply))
+
         # Generate audio for response
         audio_file = f"response_{uuid.uuid4().hex}.mp3"
         tts = gTTS(LLM_reply, lang='en')
         tts.save(audio_file)
         logger.info(f"Audio response saved as {audio_file}")
 
-        return [(user_input or "Image uploaded", LLM_reply)], audio_file
+        # Return the chat history and audio file
+        return [(entry[0], entry[1]) for entry in chat_history], audio_file
 
     except Exception as e:
         # Handle errors gracefully
@@ -100,6 +106,8 @@ def customLLMBot(user_input, uploaded_image):
 
 # Gradio Interface
 def chatbot_ui():
+    chat_history = []  # Initialize empty chat history
+
     with gr.Blocks() as demo:
         gr.Markdown("# Healthcare Chatbot Doctor")
 
@@ -122,7 +130,7 @@ def chatbot_ui():
         # Define actions
         def handle_submit(user_query, image):
             logger.info("User submitted a query.")
-            response, audio = customLLMBot(user_query, image)
+            response, audio = customLLMBot(user_query, image, chat_history)
             return response, audio, ""
 
         # Submit on pressing Enter key
@@ -152,4 +160,4 @@ def chatbot_ui():
 # Launch the interface
 chatbot_ui().launch(server_name="0.0.0.0", server_port=7860)
 
-#chatbot_ui().launch(server_name="localhost", server_port=7860)
+#chatbot_ui().launch(server_name="localhost", server_port=7860)