Update app.py

app.py

@@ -7,7 +7,7 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 css="""
 .user_chat{
-    background: #367bb9;
+    background: #367bb9; #user_c
     border-radius: 10px 10px 0px 10px;
     padding: 10px;
     max-width: 100%;
@@ -17,7 +17,7 @@ css="""
 
 }
 .bot_chat{
-    background: #1c3c9b;
+    background: #1c3c9b; #bot_c
     border-radius: 10px 10px 10px 0px;
     padding: 10px;
     max-width: 100%;
@@ -27,8 +27,8 @@ css="""
     
 }
 .chat_card{
-    background: #253251;
-    color:#ffffff;
+    background: #253251; #background_c
+    color:#ffffff; #font_c
     border-radius: 15px;
     border-width: thin;
     padding: 15px;
@@ -70,7 +70,7 @@ def make_filename(inp):
     
 def get_screenshot(chat,css_in,w,h,user_c,bot_c,background_c,font_c):
     if user_c:
-        html_user=html_user.replace("background: #367bb9;",f"background: {user_c};")
+        html_user=css.replace("background: #367bb9; #user_c",f"background: {user_c};")
     html_body=""
     for user,bot in chat:
         html_body += html_user.replace("$chat",user)