Update app.py

app.py

@@ -7,7 +7,7 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 css="""
 .user_chat{
-    background: #367bb9; #user_c
+    background: $user_c;
     border-radius: 10px 10px 0px 10px;
     padding: 10px;
     max-width: 100%;
@@ -17,7 +17,7 @@ css="""
 
 }
 .bot_chat{
-    background: #1c3c9b; #bot_c
+    background: #1c3c9b;
     border-radius: 10px 10px 10px 0px;
     padding: 10px;
     max-width: 100%;
@@ -28,7 +28,7 @@ css="""
 }
 .chat_card{
     background: #253251; #background_c
-    color:#FFFFFF; #font_c
+    color:#FFFFFF; #font_c;
     border-radius: 15px;
     border-width: thin;
     padding: 15px;
@@ -72,7 +72,7 @@ def get_screenshot(chat,css_in,w,h,user_c,bot_c,background_c,font_c):
     print(user_c)
     if user_c:
         user_c=user_c.strip('""').strip("''")
-        css=css.replace("background: #367bb9; #user_c",f"background: {user_c};")
+        css_in=css_in.replace("$user_c",user_c)
     html_body=""
     for user,bot in chat:
         html_body += html_user.replace("$chat",user)
@@ -82,7 +82,7 @@ def get_screenshot(chat,css_in,w,h,user_c,bot_c,background_c,font_c):
     filename=f'{make_filename(chat[0][0])}.png'
     print(filename)
     hti.screenshot(html_str=html, css_str=css_in, save_as=filename)
-    return filename,html
+    return filename,html,css
 
 
 def format_prompt(message, history):