Update app.py

app.py

@@ -100,13 +100,13 @@ def run_script1(url: str, height: int, width: int, check_b,check_h):
     else:
         return None, '<center>Please enter a valid URL of a website/host.',out_box,out
 
-def run_script(chat: list, height: int, width: int, check_b,check_h,wait_time):
+def run_script(chat: list, height: int, width: int, check_b,check_h,check_t,wait_time):
     mes_box=[]
     out_box=[]
     uid=uuid.uuid4()
     out=None
     is_url=True
-    url=f'https://omnibus-snap.hf.space/?chat={chat}&__theme=emerald'
+    url=f'https://omnibus-snap.hf.space/?chat={chat}&__theme={check_t}'
 
     if is_url:
         options = webdriver.ChromeOptions()
@@ -189,7 +189,9 @@ with gr.Blocks() as app:
                 stop_btn=gr.Button("Stop")
             
         with gr.Column():
-            check_h=gr.Checkbox(label="Show Header", value=True)
+            with gr.Row():
+                check_t=gr.Radio(label="Theme", choices=["light","dark"], value="dark")
+                check_h=gr.Checkbox(label="Show Header", value=True)
             check_b=gr.CheckboxGroup(label="Chatblocks", choices=[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20], value=[1])
             with gr.Row():
                 height=gr.Number(label="Height", value=4096)
@@ -202,6 +204,6 @@ with gr.Blocks() as app:
         outim = gr.Image()
     prev=gr.HTML()
     #load_btn.click(load_frame,inp,prev)
-    go = btn.click(run_script,[inp,height,width,check_b,check_h,wait_t],[outim,message,outgal,out])
+    go = btn.click(run_script,[inp,height,width,check_b,check_h,check_t,wait_t],[outim,message,outgal,out])
     stop_btn.click(None,None,None,cancels=go)
 app.queue(default_concurrency_limit=5).launch()