add urls for prs to summary

app.py

@@ -21,6 +21,7 @@ from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 from tqdm.asyncio import tqdm as atqdm
 from tqdm.auto import tqdm
 import random
+from huggingface_hub import get_discussion_details
 
 cache.setup("mem://")
 
@@ -48,8 +49,8 @@ def get_models(user_or_org):
             iter(
                 list_models(
                     filter=model_filter,
-                    sort="downloads",
-                    direction=-1,
+                    # sort="downloads",
+                    # direction=-1,
                     cardData=True,
                     full=True,
                 )
@@ -185,6 +186,16 @@ For a hands-on example of how such metadata can play a pivotal role in mapping m
 This PR was requested via the [Librarian Bot](https://huggingface.co/librarian-bot)  [metadata request service](https://huggingface.co/spaces/librarian-bots/metadata_request_service) by request of [{opened_by}](https://huggingface.co/{opened_by})
 """
 
+
+PR_FROM_COMMIT_PATTERN = re.compile(r"pr%2F(\d{1,3})/README.md")
+
+
+def get_pr_url_from_commit_url(commit_url, repo_id):
+    re_match = re.search(PR_FROM_COMMIT_PATTERN, commit_url)
+    pr_number = int(re_match.groups()[0])
+    return get_discussion_details(repo_id=repo_id, discussion_num=pr_number).url
+
+
 def update_metadata(metadata_payload: Dict[str, str], user_making_request=None):
     metadata_payload["opened_pr"] = False
     regex_match = metadata_payload["match"]
@@ -214,7 +225,7 @@ def update_metadata(metadata_payload: Dict[str, str], user_making_request=None):
                             logger.info("previously opened PR to add base_model tag")
                             metadata_payload["opened_pr"] = True
                             return metadata_payload
-        model_card.push_to_hub(
+        commit_url = model_card.push_to_hub(
             repo_id,
             token=token,
             repo_type="model",
@@ -223,6 +234,9 @@ def update_metadata(metadata_payload: Dict[str, str], user_making_request=None):
             commit_description=template,
         )
         metadata_payload["opened_pr"] = True
+        metadata_payload["pr_url"] = get_pr_url_from_commit_url(
+            commit_url=commit_url, repo_id=repo_id
+        )
         return metadata_payload
     except HfHubHTTPError:
         return metadata_payload
@@ -246,7 +260,24 @@ def open_prs(profile: gr.OAuthProfile | None, user_or_org: str = None):
             )
         except Exception as e:
             logger.error(e)
-    return f"Opened {len([r for r in results if r['opened_pr']])} PRs"
+        if not results:
+            return "No PRs to open"
+        if not any(r["opened_pr"] for r in results):
+            return "No PRs to open"
+    message = "# ✨ Librarian Bot Metadata Request Summary ✨ \n\n"
+    message += (
+        f"Librarian bot has {len([r for r in results if r['opened_pr']])} PRs open"
+        " against your repos \n\n"
+    )
+    message += "# URLs for newly opened PRs\n"
+    for result in results:
+        if result["opened_pr"]:
+            print(result)
+            try:
+                message += f"- {result['pr_url']}\n"
+            except KeyError:
+                continue
+    return message
 
 
 # description_text = """
@@ -310,7 +341,8 @@ For a hands-on example of how such metadata can play a pivotal role in mapping m
 
 with gr.Blocks() as demo:
     gr.HTML(
-        "<h1 style='text-align:center;'><span>&#129302;</span> Librarian Bot Metadata Request Service <span>&#129302;</span></h1>"
+        "<h1 style='text-align:center;'><span>&#129302;</span> Librarian Bot Metadata"
+        " Request Service <span>&#129302;</span></h1>"
     )
     gr.Markdown(
         """<div style='text-align:center;'><img src='https://huggingface.co/spaces/davanstrien/librarian_bot_request_metadata/resolve/main/image.png' style='display:block;margin-left:auto;margin-right:auto;width:150px;'></div><p>"""