Update app.py

app.py

@@ -144,11 +144,8 @@ def create_ui():
         .minimal-button {min-width: 30px !important; height: 25px !important; line-height: 1 !important; font-size: 12px !important; padding: 2px 5px !important;}
         .space-row {margin-bottom: 5px !important;}
         #refresh-button {
-            min-width: 40px !important;
-            height: 40px !important;
-            border-radius: 20px !important;
-            padding: 0.5rem !important;
-            font-size: 1.2rem !important;
+            width: 100% !important;
+            margin-top: 5px !important;
         }
         """
 
@@ -170,16 +167,12 @@ def create_ui():
                     url_state = gr.State("")
                     last_url_state = gr.State("")
                     
-                    with gr.Row():
-                        screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=540)
-                        with gr.Column(scale=0.1):
-                            refresh_button = gr.Button("🔄", size="sm")
-    
+                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
+                    refresh_button = gr.Button("🔄 새로고침", elem_id="refresh-button")
+                    
                     app_py_content = gr.Code(language="python", label="메인 소스코드")
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
-            
-
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select(s),
@@ -190,15 +183,17 @@ def create_ui():
                         inputs=[url_state, last_url_state],
                         outputs=[screenshot_output, last_url_state]
                     )
+
+                def refresh_screenshot(url, last_url):
+                    print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
+                    return update_screenshot(url, last_url)
+
                 refresh_button.click(
-                    update_screenshot,
+                    refresh_screenshot,
                     inputs=[url_state, last_url_state],
                     outputs=[screenshot_output, last_url_state]
-                ).then(
-                    lambda: print("Refresh button clicked")
                 )
 
-
                 update_trigger.click(
                     update_screenshot,
                     inputs=[url_state, last_url_state],