better python syntax hightlighting

app.py

@@ -27,7 +27,7 @@ class Config:
 
 Your task is to generate a helpful response to the user's query about using FiftyOne, and provide the appropriate function calls and class methods so that the user can understand how to use FiftyOne to achieve their goal.
 
-Responses should be short and concise, and include required syntax to achieve the FiftyOne user's goal"""
+Responses should be thorough, complete, helpful, and include required syntax to achieve the user's goal in FiftyOne."""
     
     def __repr__(self):
         """Secure representation that doesn't leak credentials."""
@@ -548,6 +548,78 @@ def create_demo():
     ::-webkit-scrollbar-thumb:hover {
         background: #e55d03 !important;
     }
+ 
+    /* Code block styling with Python syntax highlighting */
+    pre, code {
+        background-color: #f8f8f8 !important;
+        border: 1px solid #e1e1e8 !important;
+        border-radius: 4px !important;
+        font-family: "Consolas", "Monaco", "Courier New", monospace !important;
+        padding: 0.2em 0.4em !important;
+        font-size: 0.9em !important;
+    }
+    
+    pre {
+        padding: 0.8em 1em !important;
+        overflow-x: auto !important;
+        line-height: 1.45 !important;
+    }
+    
+    pre code {
+        background: none !important;
+        border: none !important;
+        padding: 0 !important;
+        font-size: inherit !important;
+    }
+    
+    /* Python syntax highlighting */
+    .language-python .keyword, 
+    .language-python .builtin {
+        color: #0086b3 !important; /* keywords and built-ins */
+    }
+    
+    .language-python .string {
+        color: #183691 !important; /* strings */
+    }
+    
+    .language-python .number {
+        color: #0086b3 !important; /* numbers */
+    }
+    
+    .language-python .comment {
+        color: #969896 !important; /* comments */
+    }
+    
+    .language-python .function, 
+    .language-python .class {
+        color: #795da3 !important; /* functions and classes */
+    }
+    
+    .language-python .operator {
+        color: #a71d5d !important; /* operators */
+    }
+    
+    /* Make code blocks in chatbot messages look nice */
+    .message-bubble-border pre {
+        background-color: #f8f8f8 !important;
+        border-radius: 8px !important;
+        border: 1px solid #e1e1e8 !important;
+        margin: 0.5em 0 !important;
+    }
+    
+    .message-bubble-border code {
+        font-family: "Consolas", "Monaco", "Courier New", monospace !important;
+    }
+    
+    /* For block code in user messages (dark background) */
+    .user .message-bubble-border pre {
+        background-color: rgba(255, 255, 255, 0.1) !important;
+        border: 1px solid rgba(255, 255, 255, 0.2) !important;
+    }
+    
+    .user .message-bubble-border code {
+        color: #f8f8f8 !important;
+    }
     """
     
     # Create theme with light mode base