Update app.py

app.py

@@ -13,7 +13,7 @@ from selenium.webdriver.common.by import By
 
 driver_type = 'chromedriver'
 driver=False
-def run_script(html: str, style: str,font_col,back_col):
+def run_script(html: str, style: str,height,width):
     uid=uuid.uuid4()
     font_col=font_col.strip("#")
     back_col=back_col.strip("#")
@@ -34,7 +34,7 @@ def run_script(html: str, style: str,font_col,back_col):
             driver.get(url)
             #html=driver.page_source
             driver.implicitly_wait(30)
-            driver.set_window_size(500, 4096)
+            driver.set_window_size(int(width), int(height))
             obj = driver.find_element(By.CLASS_NAME, "main")            
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")            
@@ -80,12 +80,12 @@ sty="""height:500;width:1000;"""
 
 with gr.Blocks() as app:
     with gr.Row():
-        inp = gr.Textbox(label="HTML",lines=8,max_lines=50,value=html)
+        inp = gr.Textbox(label="URL",lines=1,value=html)
         with gr.Column():
             style = gr.Textbox(label="CSS",lines=8,max_lines=50,value=sty)
             with gr.Row():
-                font_col=gr.ColorPicker(label="Font Color",value="#000000")
-                back_col=gr.ColorPicker(label="Background Color",value="#ffffff")
+                height=gr.Number(label="Height", value=4096)
+                width=gr.Number(label="width",value=512)
     btn= gr.Button()
     with gr.Row():
         with gr.Column():
@@ -94,5 +94,5 @@ with gr.Blocks() as app:
             outp = gr.HTML()
             cook = gr.Textbox(label="URL")
             html=gr.Textbox(visible=False)
-    btn.click(run_script,[inp,style,font_col,back_col],[outim,outp,cook,html])
+    btn.click(run_script,[inp,style,height,width],[outim,outp,cook,html])
 app.launch()