fix(run tests): hide button in dialog after clicking link

Author: aarthy-dk

testgen/ui/views/dialogs/run_tests_dialog.py

@@ -53,11 +53,13 @@ def run_tests_dialog(project_code: str, test_suite: TestSuiteMinimal | None = No
     button_container = st.empty()
     status_container = st.empty()
 
+    link_clicked = st.session_state.get(LINK_KEY)
     run_test_button = None
-    with button_container:
-        _, button_column = st.columns([.8, .2])
-        with button_column:
-            run_test_button = st.button("Run Tests", use_container_width=True, disabled=not test_suite_id)
+    if not link_clicked:
+        with button_container:
+            _, button_column = st.columns([.8, .2])
+            with button_column:
+                run_test_button = st.button("Run Tests", use_container_width=True, disabled=not test_suite_id)
 
     if run_test_button:
         button_container.empty()
@@ -69,7 +71,7 @@ def run_tests_dialog(project_code: str, test_suite: TestSuiteMinimal | None = No
             status_container.error(f"Test run encountered errors: {e!s}.")
 
     # The second condition is needed for the link to work
-    if run_test_button or st.session_state.get(LINK_KEY):
+    if run_test_button or link_clicked:
         with status_container.container():
             st.success(
                 f"Test run started for test suite **{test_suite_name}**."