Update app.py

app.py

@@ -20,62 +20,52 @@ interpreter.auto_run = True
 interpreter.llm.model = "gpt-4-turbo"
 interpreter.custom_instructions = "First ask the user what they want to do. Based on the input, describe the next steps. If user agrees, proceed; if not, ask what they want next.If it is anything to display , always at the end open up the file."
 
-def update_bot(text, chatbot):
-    response_json = interpreter.chat(text, stream=True, display=False)
-    formatted_response, images = json_to_markdown(response_json)
-    message = " ".join(formatted_response)
-    if message:
-        chatbot.append(("Assistant", message))
-    for img_path in images:
-        chatbot.append(("Assistant", img_path))  # Append image paths directly
-    return chatbot, ""
-
-
-def new_chat():
-    interpreter.messages = []
-    return [], ""
+def list_png_files(directory):
+    """List .png files in a given directory, sorted by modification time."""
+    png_files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith('.png')]
+    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)
+    if png_files:
+        # Load the most recent image file
+        return png_files[0]
+    return "No images available"
 
 def create_chat_widget():
     with gr.Blocks() as chatblock:
-        chatbot = gr.Chatbot(
-            [],
-            elem_id="chatbot",
-            elem_classes="chatbot",
-            layout="llm",
-            bubble_full_width=False,
-            height=600,
-        )
-        txt = gr.Textbox(
-            placeholder="Enter text and press enter to chat with the bot.",
-            show_label=False,
-        )
-        send_button = gr.Button("Send")
         with gr.Row():
+            chatbot = gr.Chatbot(
+                [],
+                elem_id="gpt4",
+                elem_classes="gpt4",
+                layout="llm",
+                bubble_full_width=False,
+                height=600,
+            )
+        with gr.Row():
+            txt = gr.Textbox(
+                placeholder="Enter text and press enter to chat with the bot.",
+                show_label=False,
+            )
+            send_button = gr.Button("Send")
             txt.submit(update_bot, inputs=[txt, chatbot], outputs=[chatbot, txt])
             send_button.click(update_bot, inputs=[txt, chatbot], outputs=[chatbot, txt])
-        return chatblock
 
-def json_to_markdown(json_data):
-    full_message = []
-    images = []
-    # Regex to detect URLs; this is a simple version and might need to be adapted
-    url_pattern = r'(http[s]?://\S+|sandbox:/\S+)'
-
-    for item in json_data:
-        if item['role'] == 'assistant' and item['type'] == 'message':
-            content = item.get('content', " ")
-            # Find all URLs in the content
-            urls = re.findall(url_pattern, content)
-            # Append any detected URLs to the images list
-            images.extend(urls)
-            # Remove URLs from the content
-            content = re.sub(url_pattern, "", content).strip()
-            if content:
-                full_message.append(content)
-    return full_message, images
+        with gr.Row():
+            image_display = gr.Image()
+            update_image_button = gr.Button("Update Image")
+            update_image_button.click(update_images, inputs=[], outputs=image_display)
 
+        return chatblock
 
+def update_bot(text, chatbot):
+    response_json = interpreter.chat(text, stream=True, display=False)
+    formatted_response = json_to_markdown(response_json)
+    chatbot.append((text, formatted_response))
+    return chatbot, ""
 
 with gr.Blocks() as demo:
     with gr.Tab("HEXON Chatbot Assignment"):