diff --git a/html/style.css b/html/style.css
new file mode 100644
index 0000000..e2c99a3
--- /dev/null
+++ b/html/style.css
@@ -0,0 +1,10 @@
+.contain div div:nth-child(2n-1) div [data-testid="markdown"] p {
+ background: #bcffb0aa;
+ margin-left: auto;
+}
+[data-testid="markdown"] p {
+ width: fit-content;
+ padding: 1em;
+ background: #cececeaa;
+ border-radius: 16px;
+}
\ No newline at end of file
diff --git a/web_demo.py b/web_demo.py
index 39709a7..3ca6eb5 100644
--- a/web_demo.py
+++ b/web_demo.py
@@ -23,7 +23,7 @@ def predict(input, max_length, top_p, temperature, history=None):
yield [history] + updates
-with gr.Blocks() as demo:
+with gr.Blocks(css="html/style.css", analytics_enabled=False) as demo:
state = gr.State([])
text_boxes = []
for i in range(MAX_BOXES):