Replace base64 encoding with artifact links

Co-authored-by: functionstackx <47992694+functionstackx@users.noreply.github.com>

Author: Copilot

.github/workflows/collect-results.yml

@@ -42,13 +42,19 @@ jobs:
           pip install -q matplotlib
           python3 utils/plot_perf.py results/ ${{ inputs.exp-name || 'all' }}
 
-      - name: Display graphs in summary
-        run: python3 utils/display_graphs.py ${{ inputs.exp-name || 'all' }} >> $GITHUB_STEP_SUMMARY
-
       - name: Upload performance graphs
         uses: actions/upload-artifact@v5
         with:
           name: graphs_${{ inputs.exp-name || 'all' }}
           path: |
             tput_vs_intvty_*_${{ inputs.exp-name || 'all' }}.png
             tput_vs_e2el_*_${{ inputs.exp-name || 'all' }}.png
+
+      - name: Display graphs in summary
+        run: |
+          python3 utils/display_graphs.py \
+            ${{ inputs.exp-name || 'all' }} \
+            ${{ github.repository }} \
+            ${{ github.run_id }} \
+            graphs_${{ inputs.exp-name || 'all' }} \
+            >> $GITHUB_STEP_SUMMARY

utils/display_graphs.py

@@ -1,17 +1,9 @@
 import sys
-import base64
 from pathlib import Path
 
 
-def encode_image_to_base64(image_path):
-    """Encode an image file to base64 string."""
-    with open(image_path, 'rb') as image_file:
-        encoded = base64.b64encode(image_file.read()).decode('utf-8')
-    return encoded
-
-
-def display_graphs(exp_name):
-    """Display performance graphs in GitHub Actions summary."""
+def display_graphs(exp_name, repo, run_id, artifact_name):
+    """Display performance graphs in GitHub Actions summary with artifact links."""
     print("\n## Performance Graphs\n")
 
     # Find all generated graphs
@@ -21,34 +13,38 @@ def display_graphs(exp_name):
     intvty_graphs = sorted(current_dir.glob(f'tput_vs_intvty_*_{exp_name}.png'))
     e2el_graphs = sorted(current_dir.glob(f'tput_vs_e2el_*_{exp_name}.png'))
 
+    # Construct artifact download URL
+    artifact_url = f"https://github.com/{repo}/actions/runs/{run_id}/artifacts"
+    
     # Display interactivity graphs
     if intvty_graphs:
         print("### Throughput vs Interactivity\n")
         for graph in intvty_graphs:
             # Extract model name from filename
             model_name = graph.name.replace(f'tput_vs_intvty_', '').replace(f'_{exp_name}.png', '')
-            base64_image = encode_image_to_base64(graph)
             print(f"#### {model_name.upper()}\n")
-            print(f"![Throughput vs Interactivity - {model_name}](data:image/png;base64,{base64_image})\n")
+            print(f"📊 [{graph.name}]({artifact_url}) (download `{artifact_name}` artifact)\n")
 
     # Display end-to-end latency graphs
     if e2el_graphs:
         print("### Throughput vs End-to-End Latency\n")
         for graph in e2el_graphs:
             # Extract model name from filename
             model_name = graph.name.replace(f'tput_vs_e2el_', '').replace(f'_{exp_name}.png', '')
-            base64_image = encode_image_to_base64(graph)
             print(f"#### {model_name.upper()}\n")
-            print(f"![Throughput vs E2E Latency - {model_name}](data:image/png;base64,{base64_image})\n")
+            print(f"📊 [{graph.name}]({artifact_url}) (download `{artifact_name}` artifact)\n")
 
     if not intvty_graphs and not e2el_graphs:
         print("*No performance graphs were generated.*\n")
 
 
 if __name__ == '__main__':
-    if len(sys.argv) < 2:
-        print("Usage: python3 display_graphs.py <exp_name>")
+    if len(sys.argv) < 5:
+        print("Usage: python3 display_graphs.py <exp_name> <repo> <run_id> <artifact_name>")
         sys.exit(1)
 
     exp_name = sys.argv[1]
-    display_graphs(exp_name)
+    repo = sys.argv[2]
+    run_id = sys.argv[3]
+    artifact_name = sys.argv[4]
+    display_graphs(exp_name, repo, run_id, artifact_name)