Update app.py

app.py

@@ -22,8 +22,12 @@ interpreter.custom_instructions = "First ask the user what they want to do. Base
 
 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))
+    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, ""
 
 
@@ -59,11 +63,23 @@ def create_chat_widget():
 
 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', " ")
-            full_message.append(content)  # Append the content to the list
-    return "**Assistant:**\n" + " ".join(full_message)  # Join all parts into one string and format as a single message
+            # 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.Blocks() as demo: