Update app.py

app.py

@@ -419,6 +419,7 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
     str,  # pytest_commands
     str,  # raw_json
     str,  # status
+    str,  # data_source_link
 ]:
     repo = repo.strip()
     pr = pr.strip()
@@ -434,7 +435,8 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
             "",
             "",
             json.dumps({"error": "PR number is required."}, indent=2),
-            "❌ Provide a PR number to search."
+            "❌ Provide a PR number to search.",
+            ""
         )
 
     records = _filter_records(repo, pr, sha)
@@ -448,11 +450,23 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
             "",
             "",
             json.dumps({"error": "No records found."}, indent=2),
-            f"❌ No records found for PR {pr}."
+            f"❌ No records found for PR {pr}.",
+            ""
         )
 
     # Use the latest record
     latest_record = records[0]
+    
+    # Generate data source link
+    source_path = latest_record.get("__source_path", "")
+    data_source_link = f"""
+---
+
+**Data source:** [{source_path}](https://huggingface.co/datasets/{DATASET_ID}/blob/main/{source_path})
+
+Files are organized as `pr-{{PR}}/sha-{{COMMIT}}/failure_summary.json`
+"""
+    
     metadata = latest_record.get("metadata", {})
     
     # Generate metadata info
@@ -488,7 +502,8 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
         markdown_summary,
         pytest_commands,
         raw_json,
-        status
+        status,
+        data_source_link
     )
 
 
@@ -598,6 +613,9 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
                 language="json",
                 lines=20,
             )
+    
+    # Dynamic data source link (replaces the hardcoded one)
+    data_source_md = gr.Markdown("")
 
     def get_url_params(request: gr.Request):
         """Get URL parameters from the request"""
@@ -625,7 +643,8 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
                 "",
                 "",
                 "",
-                "💡 Enter a PR number above to get started"
+                "💡 Enter a PR number above to get started",
+                ""
             )
     
     # Connect the search button
@@ -639,7 +658,8 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
             markdown_output,
             pytest_output,
             json_view,
-            status_md
+            status_md,
+            data_source_md
         ]
     )
     
@@ -660,20 +680,11 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
             markdown_output,
             pytest_output,
             json_view,
-            status_md
+            status_md,
+            data_source_md
         ]
     )
 
-    gr.Markdown(
-        """
-        ---
-        
-        **Data source:** [https://huggingface.co/datasets/transformers-community/circleci-test-results/blob/main/pr-42240/sha-50947fc72be7/failure_summary.json](https://huggingface.co/datasets/transformers-community/circleci-test-results/blob/main/pr-42240/sha-50947fc72be7/failure_summary.json)
-        
-        Files are organized as `pr-{PR}/sha-{COMMIT}/failure_summary.json`
-        """
-    )
-
 print("="*60)
 print("✓ GRADIO INTERFACE CREATED SUCCESSFULLY")
 print("="*60)