Update app.py

app.py

@@ -197,9 +197,13 @@ def _filter_records(repo: str, pr: str, sha: str) -> List[dict]:
     records: List[dict] = []
     for file_path in file_paths:
         commit = _extract_commit_from_path(file_path)
-        if sha and not commit.lower().startswith(sha):
+        
+        # Fixed SHA matching: works with both short and full SHAs
+        # Check if stored commit starts with input SHA OR if input SHA starts with stored commit
+        if sha and not (commit.lower().startswith(sha) or sha.startswith(commit.lower())):
             print(f"DEBUG: Skipping {file_path} - commit {commit} doesn't match sha {sha}")
             continue
+            
         payload = _load_payload(file_path)
         if payload is None:
             print(f"DEBUG: Skipping {file_path} - failed to load payload")
@@ -415,7 +419,6 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
     str,  # metadata_info
     str,  # by_test_html
     str,  # by_model_html
-    str,  # markdown_summary
     str,  # pytest_commands
     str,  # raw_json
     str,  # status
@@ -427,13 +430,24 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
     
     print(f"DEBUG: Query called with repo='{repo}', pr='{pr}', sha='{sha}'")
 
+    # Validate SHA length if provided
+    if sha and len(sha) < 6:
+        return (
+            "**Error:** Commit SHA must be at least 6 characters.",
+            "",
+            "",
+            "",
+            json.dumps({"error": "Commit SHA must be at least 6 characters."}, indent=2),
+            "⚠️ Commit SHA must be at least 6 characters.",
+            ""
+        )
+
     if not pr:
         return (
             "**Error:** PR number is required.",
             "",
             "",
             "",
-            "",
             json.dumps({"error": "PR number is required."}, indent=2),
             "❌ Provide a PR number to search.",
             ""
@@ -448,7 +462,6 @@ def query(repo: str, pr: str, sha: str) -> Tuple[
             "",
             "",
             "",
-            "",
             json.dumps({"error": "No records found."}, indent=2),
             f"❌ No records found for PR {pr}.",
             ""
@@ -469,14 +482,12 @@ Files are organized as `pr-{{PR}}/sha-{{COMMIT}}/failure_summary.json`
     
     metadata = latest_record.get("metadata", {})
     
-    # Generate metadata info
+    # Generate simplified metadata info
+    commit_sha = latest_record.get("__commit", "N/A")
     metadata_lines = [
-        f"**Repository:** {metadata.get('repository', 'N/A')}",
-        f"**Branch:** {metadata.get('branch', 'N/A')}",
-        f"**PR:** #{metadata.get('pull_request_number', 'N/A')}",
-        f"**Commit:** `{metadata.get('commit_sha', 'N/A')[:12]}`",
-        f"**Workflow ID:** {metadata.get('workflow_id', 'N/A')}",
-        f"**Collected at:** {metadata.get('collected_at', 'N/A')}",
+        f"**Repository:** huggingface/transformers",
+        f"**PR:** #{pr}",
+        f"**Commit:** `{commit_sha}`",
         f"**Total failures:** {len(latest_record.get('failures', []))}",
     ]
     metadata_info = "\n\n".join(metadata_lines)
@@ -484,22 +495,19 @@ Files are organized as `pr-{{PR}}/sha-{{COMMIT}}/failure_summary.json`
     # Generate HTML tables
     by_test_html, by_model_html = _generate_html_tables(latest_record)
     
-    # Generate markdown summary
-    markdown_summary = _generate_markdown_summary(latest_record)
-    
     # Generate pytest commands
     pytest_commands = _generate_pytest_commands(latest_record)
     
     # Raw JSON
     raw_json = json.dumps(latest_record, indent=2)
     
-    status = f"✅ Showing latest result from {len(records)} record(s) for PR {pr}."
+    # Updated status message format
+    status = f"✅ Showing result from {len(records)} record (commit: {commit_sha}) PR {pr}."
     
     return (
         metadata_info,
         by_test_html,
         by_model_html,
-        markdown_summary,
         pytest_commands,
         raw_json,
         status,
@@ -556,8 +564,8 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
         with gr.Column(scale=1):
             sha_box = gr.Textbox(
                 label="Commit SHA", 
-                placeholder="50947fc",
-                info="Optional: commit SHA prefix"
+                placeholder="50947fc (min 6 chars)",
+                info="Optional: commit SHA (min 6 characters)"
             )
 
     with gr.Row():
@@ -582,18 +590,6 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
             
             by_model_html = gr.HTML(label="Model Failures")
         
-        with gr.Tab("📋 Copy for GitHub"):
-            gr.Markdown(
-                """
-                Copy the markdown below to paste directly into a GitHub comment.
-                """
-            )
-            markdown_output = gr.Textbox(
-                label="Markdown Summary",
-                lines=20,
-                max_lines=30,
-            )
-        
         with gr.Tab("🧪 Pytest Commands"):
             gr.Markdown(
                 """
@@ -642,7 +638,6 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
                 "",
                 "",
                 "",
-                "",
                 "💡 Enter a PR number above to get started",
                 ""
             )
@@ -655,7 +650,6 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
             metadata_box,
             by_test_html,
             by_model_html,
-            markdown_output,
             pytest_output,
             json_view,
             status_md,
@@ -677,7 +671,6 @@ with gr.Blocks(title="CircleCI Test Results Viewer") as demo:
             metadata_box,
             by_test_html,
             by_model_html,
-            markdown_output,
             pytest_output,
             json_view,
             status_md,