Final working

app.py

@@ -39,15 +39,14 @@ def list_png_files(directory):
     png_files.sort(key=lambda x: os.path.getmtime(x), reverse=True)  # Newest first
     return png_files
 
-def update_images(image_component, directory="/home/user/app/"):
-    """Update the image component with the latest image from the directory."""
-    png_files = list_png_files(directory)
+def update_images(image_component):
+    """Update the image component with the latest image from the current directory."""
+    current_directory = os.getcwd()  # Get the current working directory
+    png_files = list_png_files(current_directory)
     if png_files:
-        # Load the most recent image file
-        return png_files[0]
+        return png_files[0]  # Load the most recent image file
     return "No images available"
 
-
 def create_chat_widget():
     with gr.Blocks() as chatblock:
         # Declare chatbot and txt here so they are accessible later
@@ -88,25 +87,31 @@ def create_chat_widget():
     
 def new_chat():
     return [], ""
-
-
 def update_bot(text, chatbot):
-    if text:  # Append user input to the chat with "User:" prefix
-        chatbot.append(("User", f"User: {text}"))
+    if text.strip():  # Ensure the input text is not empty or just whitespace
+        # Append user input to the chat with specific formatting
+        chatbot.append(("User: " + text.strip(), ""))
+    
+    # Simulate getting a response from the interpreter
     response_json = interpreter.chat(text, stream=True, display=False)
     formatted_response, images = json_to_markdown(response_json)
-    # Append text response with "Assistant:" prefix if it exists
+
+    # Append the assistant's response with the desired formatting
     if formatted_response.strip():
-        chatbot.append(("Assistant", f"Assistant: {formatted_response}"))
-    # Append images if any
+        chatbot.append(("Assistant: " + formatted_response.strip(), ""))
+
+    # Handle images here if necessary
     for img_path in images:
         if os.path.exists(img_path) and img_path.endswith('.png'):
-            chatbot.append(("Assistant", img_path))  # Assuming the path is accessible
+            # Optionally display image paths
+            chatbot.append(("Assistant (image):", img_path))
         else:
-            chatbot.append(("Assistant", "Image not found or path is invalid."))
+            chatbot.append(("Assistant:", "Image not found or path is invalid."))
+
     return chatbot, ""
 
 
+
 with gr.Blocks() as demo:
     with gr.Tab("HEXON Chatbot Assignment"):
         chat_interface = create_chat_widget()