Update app.py

app.py

@@ -39,6 +39,14 @@ async def read_root(response: Response):
             box-sizing: border-box;
             color: #e2e8f0; /* Light text color for general body text */
         }
+        code {
+                margin: 6px 0px;
+                display: block;
+                border-radius: 12px;
+                overflow-x: scroll;
+                background-color: #1e1e1e;
+                padding: 12px;
+        }
         .chat-container {
             display: flex;
             flex-direction: column;
@@ -52,7 +60,7 @@ async def read_root(response: Response):
             border: 1px solid #4a5568; /* Subtle border for dark mode */
         }
         .chat-header {
-            background: linear-gradient(to right, #2c5282, #4c78a8); /* Darker blue gradient header */
+            background: linear-gradient(to right, #6a0076, #12327a); /* Darker blue gradient header */
             color: #ffffff;
             padding: 1.5rem;
             text-align: center;
@@ -79,7 +87,7 @@ async def read_root(response: Response):
             animation: fadeIn 0.3s ease-out;
         }
         .user-message {
-            background-color: #4299e1; /* Blue for user messages in dark mode */
+            background-color: #12327a; /* Blue for user messages in dark mode */
             align-self: flex-end;
             color: #ffffff; /* White text for user messages */
             border-bottom-right-radius: 0.5rem;
@@ -333,8 +341,8 @@ async def read_root(response: Response):
                     throw new Error(`HTTP error! status: ${response.status}`);
                 }
 
-                const data = await response.text();
-                addMessage(data || "Sorry, I couldn't get a response.", 'assistant');
+                const data = await response.json();
+                addMessage(data.text || "Sorry, I couldn't get a response.", 'assistant');
             } catch (error) {
                 console.error('Error sending message:', error);
                 addMessage('Error: Could not connect to the assistant. Please try again.', 'assistant');
@@ -394,7 +402,7 @@ async def handle_chat(chat_request: ChatRequest, token: str = Cookie(None)):
                 messages=History[i],
             )
         History[i].append({"role": "assistant", "content": chat_request.prompt})
-        return stream.choices[0].message.content
+        return {"text": stream.choices[0].message.content}
     else: return 'Please stop. Just refresh the page.'
 
 @app.post("/history")