Update app.py

app.py

@@ -28,7 +28,7 @@ css="""
 }
 .chat_card{
     background: #253251;
-    color:white;
+    color:#ffffff;
     border-radius: 15px;
     border-width: thin;
     padding: 15px;
@@ -64,11 +64,13 @@ def make_filename(inp):
         first_name=inp[:30]
     first_name=first_name.replace(" ","_")
     print(first_name)
-    last_name=random.randint(1,1111111111)
+    last_name=random.randint(1,99999999999999)
     filename=first_name+str(last_name)
     return filename
-def get_screenshot(chat,css_in,w,h):
     
+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_body=""
     for user,bot in chat:
         html_body += html_user.replace("$chat",user)
@@ -141,6 +143,12 @@ with gr.Blocks(css=css) as app:
                 with gr.Row():
                     im_height=gr.Number(label="Height",value=1000)
                     im_width=gr.Number(label="Width",value=500)
+                with gr.Row():
+                    user_c=gr.Color_Picker(value="#367bb9")
+                    bot_c=gr.Color_Picker(value="#1c3c9b")
+                with gr.Row():
+                    background_c=gr.Color_Picker(value="#253251")
+                    font_c=gr.Color_Picker(value="#ffffff")                    
                 wait_time=gr.Number(label="Wait Time",value=3000)
                 theme=gr.Radio(label="Theme", choices=["light","dark"],value="light")
                 chatblock=gr.Dropdown(label="Chatblocks",choices=[c for c in range(1,40)],multiselect=True)
@@ -149,7 +157,7 @@ with gr.Blocks(css=css) as app:
                 img=gr.Image(type='filepath')
     css_box=gr.Textbox(lines=50,value=css)
     html_view=gr.HTML()
-    btn.click(chat_inf,[sys_inp,inp,chat_b],chat_b)
+    btn.click(chat_inf,[sys_inp,inp,chat_b,user_c,bot_c,background_c,font_c],chat_b)
     im_btn.click(get_screenshot,[chat_b,css_box,im_width,im_height],[img,html_view])
     #im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
     #app.load(get_screenshot,inp,img)