Update app.py

app.py

@@ -108,18 +108,22 @@ def on_select(space):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", ""
 
-def update_screenshot(url, last_url):
-    print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}")
-    if url and url != last_url:
+def update_screenshot(url, last_url, force_update=False):
+    print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}, Force update: {force_update}")
+    if url and (url != last_url or force_update):
         screenshot = take_screenshot(url)
         print("Screenshot updated")
         return screenshot, url
     print("No update needed")
     return gr.update(), last_url
-        
+
+def refresh_screenshot(url, last_url):
+    print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
+    return update_screenshot(url, last_url, force_update=True)
+    
 def take_screenshot(url):
     try:
-        print(f"Attempting to take screenshot of URL: {url}")
+        print(f"Taking screenshot of URL: {url}")
         client = Client("ginipick/selenium-screenshot-gradio")
         result = client.predict(url=url, api_name="/predict")
         print(f"Screenshot result: {result}")
@@ -127,10 +131,10 @@ def take_screenshot(url):
             return Image.open(result)
         else:
             print(f"Invalid result from API: {result}")
-            return Image.new('RGB', (540, 360), color='lightgray')
+            return Image.new('RGB', (600, 360), color='lightgray')
     except Exception as e:
         print(f"Screenshot error: {str(e)}")
-        return Image.new('RGB', (540, 360), color='lightgray')
+        return Image.new('RGB', (600, 360), color='lightgray')
             
 def create_ui():
     try:
@@ -193,6 +197,7 @@ def create_ui():
                     inputs=[url_state, last_url_state],
                     outputs=[screenshot_output, last_url_state]
                 )
+                
 
                 update_trigger.click(
                     update_screenshot,