Update app.py

app.py

@@ -121,8 +121,6 @@ def take_screenshot(url):
         print(f"Screenshot error: {str(e)}")
         return Image.new('RGB', (1080, 720), color='lightgray')
             
-
-
 def create_ui():
     try:
         spaces_list = get_most_liked_spaces()
@@ -153,7 +151,11 @@ def create_ui():
                     info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
                     url_state = gr.State("")
                     last_url_state = gr.State("")
-                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
+                    
+                    with gr.Row():
+                        screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=540)
+                        refresh_button = gr.Button("🔄 새로고침", scale=0.1)
+                    
                     app_py_content = gr.Code(language="python", label="메인 소스코드")
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
@@ -168,6 +170,12 @@ def create_ui():
                         outputs=[screenshot_output, last_url_state]
                     )
 
+                refresh_button.click(
+                    update_screenshot,
+                    inputs=[url_state, last_url_state],
+                    outputs=[screenshot_output, last_url_state]
+                )
+
                 update_trigger.click(
                     update_screenshot,
                     inputs=[url_state, last_url_state],