Update screenshot.py

screenshot.py

@@ -30,7 +30,7 @@ def create_ss(graph,width=4000,height=8000):
             print(html)            
             driver.implicitly_wait(int(wait_time*30))
             #driver.set_window_size(int(width), int(height))
-            obj = driver.find_element(By.ID, "chart_box")            
+            obj = driver.find_element(By.ID, "chart")            
             screenshot = obj.screenshot(f'{uid}-tmp.png')
             return(f'{uid}-tmp.png')  
         except Exception as e: