Update app.py

app.py

@@ -626,8 +626,14 @@ class XylariaChat:
 
                 if isinstance(image_bytes, bytes):
                     base64_image = base64.b64encode(image_bytes).decode("utf-8")
-                    image_html = f'<img src="data:image/png;base64,{base64_image}" alt="Generated Image" style="max-width: 100%; max-height: 400px;">'
-
+                    
+                    # Updated artifact section
+                    image_html = f"""
+<details>
+<summary>Generated Image</summary>
+<img src="data:image/png;base64,{base64_image}" alt="Generated Image" style="max-width: 100%; max-height: 400px;">
+</details>
+"""
                     # Append the /image command and generated image to chat history
                     chat_history.append([message, ""])
                     chat_history.append(["", image_html])
@@ -676,7 +682,25 @@ class XylariaChat:
                         chunk_content = chunk.choices[0].delta.content
                         full_response += chunk_content
                         
-                        updated_history[-1][1] = full_response
+                        # Updated artifact section for code output
+                        if "```" in full_response:
+                            parts = full_response.split("```")
+                            formatted_response = ""
+                            for i, part in enumerate(parts):
+                                if i % 2 == 0:
+                                    formatted_response += part
+                                else:
+                                    
+                                    formatted_response += f"""
+<details>
+<summary>Code Output</summary>
+```{part}```                        
+</details>
+"""
+                            updated_history[-1][1] = formatted_response
+                        else:
+                            updated_history[-1][1] = full_response
+                        
                         yield "", updated_history, None, None, None
             except Exception as e:
                 print(f"Streaming error: {e}")
@@ -843,6 +867,22 @@ class XylariaChat:
         #main-content {
             flex: 1;
             transition: flex 0.3s ease;
+        }
+                .output-markdown {
+            padding: 10px;
+            border: 1px solid #ccc;
+            border-radius: 5px;
+            margin-bottom: 10px;
+        }
+        .output-markdown summary {
+            font-weight: bold;
+            cursor: pointer;
+        }
+        .output-markdown code {
+            background-color: #f7f7f7;
+            padding: 2px 4px;
+            border-radius: 3px;
+            font-family: monospace;
         }
         """