Update app.py

app.py

@@ -188,44 +188,7 @@ def refresh_dataset() -> str:
     return "Cleared cached manifest. Data will be reloaded on next search."
 
 
-with gr.Blocks(head="""
-<script>
-document.addEventListener('DOMContentLoaded', function() {
-    // Parse URL parameters
-    const params = new URLSearchParams(window.location.search);
-    const repo = params.get('repo') || '';
-    const pr = params.get('pr') || '';
-    const sha = params.get('sha') || '';
-    
-    if (repo || pr || sha) {
-        // Wait for Gradio to initialize
-        setTimeout(() => {
-            // Find and populate the input fields
-            const inputs = document.querySelectorAll('input[type="text"]');
-            if (inputs.length >= 3) {
-                if (repo) inputs[0].value = repo;
-                if (pr) inputs[1].value = pr;
-                if (sha) inputs[2].value = sha;
-                
-                // Trigger input events to update Gradio's state
-                inputs.forEach(input => {
-                    input.dispatchEvent(new Event('input', { bubbles: true }));
-                });
-                
-                // Auto-click search if PR is provided
-                if (pr) {
-                    setTimeout(() => {
-                        const buttons = document.querySelectorAll('button');
-                        const searchBtn = Array.from(buttons).find(btn => btn.textContent.includes('Search'));
-                        if (searchBtn) searchBtn.click();
-                    }, 500);
-                }
-            }
-        }, 1000);
-    }
-});
-</script>
-""") as demo:
+with gr.Blocks() as demo:
     gr.Markdown(
         """
         # CircleCI Test Collection Helper
@@ -263,6 +226,46 @@ document.addEventListener('DOMContentLoaded', function() {
 
     search_btn.click(query, inputs=[repo_box, pr_box, sha_box], outputs=[table, json_view, status])
     refresh_btn.click(refresh_dataset, outputs=status)
+    
+    # Add JavaScript to handle URL parameters
+    gr.HTML("""
+    <script>
+    (function() {
+        const params = new URLSearchParams(window.location.search);
+        const repo = params.get('repo') || '';
+        const pr = params.get('pr') || '';
+        const sha = params.get('sha') || '';
+        
+        if (repo || pr || sha) {
+            setTimeout(() => {
+                const inputs = document.querySelectorAll('input[type="text"]');
+                if (inputs.length >= 3) {
+                    if (repo) {
+                        inputs[0].value = repo;
+                        inputs[0].dispatchEvent(new Event('input', { bubbles: true }));
+                    }
+                    if (pr) {
+                        inputs[1].value = pr;
+                        inputs[1].dispatchEvent(new Event('input', { bubbles: true }));
+                    }
+                    if (sha) {
+                        inputs[2].value = sha;
+                        inputs[2].dispatchEvent(new Event('input', { bubbles: true }));
+                    }
+                    
+                    if (pr) {
+                        setTimeout(() => {
+                            const buttons = document.querySelectorAll('button');
+                            const searchBtn = Array.from(buttons).find(btn => btn.textContent.includes('Search'));
+                            if (searchBtn) searchBtn.click();
+                        }, 500);
+                    }
+                }
+            }, 1000);
+        }
+    })();
+    </script>
+    """)
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(ssr_mode=False)