Allow to post report with detailed info

app.py

@@ -2,9 +2,15 @@ import gradio as gr
 import datasets
 import huggingface_hub
 import sys
+import os
 from pathlib import Path
 
 
+HF_REPO_ID = 'HF_REPO_ID'
+HF_SPACE_ID = 'SPACE_ID'
+HF_WRITE_TOKEN = 'HF_WRITE_TOKEN'
+
+
 theme = gr.themes.Soft(
     primary_hue="green",
 )
@@ -85,7 +91,7 @@ def try_submit(model_id, dataset_id, dataset_config, dataset_split, local):
         from giskard_cicd.loaders import HuggingFaceLoader
         from giskard_cicd.pipeline.runner import PipelineRunner
 
-        from cicd.automation import create_discussion
+        from automation import create_discussion_detailed
         supported_loaders = {
             "huggingface": HuggingFaceLoader(),
         }
@@ -103,8 +109,11 @@ def try_submit(model_id, dataset_id, dataset_config, dataset_split, local):
 
         report = runner.run(**runner_kwargs)
 
-        # TODO: Publish it
-        # rendered_report = report.to_markdown(template="github")
+        # TODO: Publish it with given repo id/model id
+        if os.environ.get(HF_REPO_ID) or os.environ.get(HF_SPACE_ID) and os.environ.get(HF_WRITE_TOKEN):
+            rendered_report = report.to_markdown(template="github")
+            repo = os.environ.get(HF_REPO_ID) or os.environ.get(HF_SPACE_ID)
+            create_discussion_detailed(repo, m_id, d_id, config, split, os.environ.get(HF_WRITE_TOKEN), rendered_report)
 
         # Cache locally
         rendered_report = report.to_html()

cicd/automation/__init__.py

@@ -1,3 +1,3 @@
-from .post_discussion import create_discussion
+from .post_discussion import create_discussion, create_discussion_detailed
 
-__all__ = ["create_discussion"]
+__all__ = ["create_discussion", "create_discussion_detailed"]

cicd/automation/post_discussion.py

@@ -1,5 +1,18 @@
 import huggingface_hub as hf_hub
+
 def create_discussion(repo_id, model_name, hf_token, report):
     # Create a discussion
     discussion = hf_hub.create_discussion(repo_id, title=f"Report for {model_name}", token=hf_token, description=report, repo_type="space")
     return discussion
+
+
+def create_discussion_detailed(repo_id, model_name, dataset_id, dataset_config, dataset_split, hf_token, report):
+    # Create a discussion
+    discussion = hf_hub.create_discussion(
+        repo_id,
+        title=f"Report for {model_name} on {dataset_id} ({dataset_config}, {dataset_split} set)",
+        token=hf_token,
+        description=report,
+        repo_type="space"
+    )
+    return discussion