Update app.py

app.py

@@ -9,6 +9,7 @@ import json
 import requests
 from dotenv import load_dotenv
 import os
+import time
 import gradio as gr
 import subprocess
 
@@ -50,7 +51,7 @@ def screenshotName(url):
         url = url[:-1]
     else:
         url = url
-    ssname = f"SS_{url.replace('/', '')}.png"
+    ssname = f"{url.replace('/', '')}.png"
     return ssname
 
 def getScreenshots(driver, url):
@@ -61,6 +62,7 @@ def getScreenshots(driver, url):
 
 def saveScreenshot(url):
     driver = driverSetup()
+    time.sleep(15)
     screenshot_file = getScreenshots(driver, url)
     driver.quit()
     return screenshot_file