Update app.py

app.py

@@ -226,11 +226,26 @@ def create_ui():
                         """
                         <script type="module" src="https://gradio.s3-us-west-2.amazonaws.com/4.36.1/gradio.js"></script>
                         <gradio-app id="embedded-gradio-app" style="width:100%;height:600px;"></gradio-app>
+                        <script>
+                        function updateGradioApp(url) {
+                            const app = document.getElementById('embedded-gradio-app');
+                            if (app) {
+                                app.setAttribute('src', url);
+                            }
+                        }
+                        </script>
                         """
                     )
                     
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
+                def update_embedded_space(url):
+                    return f"""
+                    <script>
+                    updateGradioApp("{url}");
+                    </script>
+                    """
+
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select(s),
@@ -241,16 +256,13 @@ def create_ui():
                         inputs=[url_state, last_url_state],
                         outputs=[screenshot_output, last_url_state]
                     ).then(
-                        lambda url: f"""
-                        <script>
-                            document.getElementById('embedded-gradio-app').setAttribute('src', '{url}');
-                        </script>
-                        """,
+                        update_embedded_space,
                         inputs=[url_state],
                         outputs=[embedded_space]
                     )
 
 
+
                 def refresh_screenshot(url, last_url):
                     print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
                     return update_screenshot(url, last_url)