Update screenshot.py

screenshot.py

@@ -8,7 +8,7 @@ import uuid
 driver_type = 'chromedriver'
 driver=False
 
-def create_ss(graph,width=4000,height=8000):
+def create_ss(graph,width=1000,height=8000):
     uid=uuid.uuid4()
     code_out=""
     for ea in graph.split("\n"):
@@ -29,7 +29,7 @@ def create_ss(graph,width=4000,height=8000):
             html=driver.page_source
             print(html)            
             driver.implicitly_wait(int(wait_time*30))
-            #driver.set_window_size(int(width), int(height))
+            driver.set_window_size(int(width), int(height))
             obj = driver.find_element(By.ID, "chart")            
             screenshot = obj.screenshot(f'{uid}-tmp.png')
             return(f'{uid}-tmp.png')