Update app.py

app.py

@@ -1,11 +1,11 @@
 #from __future__ import annotations
 from selenium import webdriver
-from PIL import Image
 import gradio as gr
 import uuid
 import re
 from PIL import Image
 from io import BytesIO
+from pathlib import Path
 from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By
@@ -59,8 +59,8 @@ def run_script(url: str, height: int, width: int):
             print (len(messages))
             for i,mes in enumerate(messages):
                 print (mes.text)
-                mes_shot = obj.screenshot(f'{i}{uid}-tmp.png')
-                mes_box.append(mes_shot)
+                obj.screenshot(f'{i}-{uid}-tmp.png')
+                mes_box.append(f'{i}-{uid}-tmp.png')
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")            
             #obj = driver.find_element_by_id("in_html")