Spaces:

daqc
/

hugging-research

Sleeping

daqc commited on Aug 22

Commit

e061bc4

verified ·

1 Parent(s): 1b3e476

Update scripts/report_generator.py

Files changed (1) hide show

scripts/report_generator.py CHANGED Viewed

@@ -25,6 +25,7 @@ class HFLinkReportTool(Tool):
             import json as _json
             doc_title = title or "Report"
             query = (query or "").strip()
             # Extract URLs
             urls = re.findall(r"https?://[^\s)\]]+", final_answer or "")
@@ -129,7 +130,7 @@ class HFLinkReportTool(Tool):
   <script src=\"https://cdn.jsdelivr.net/npm/dompurify@3.1.6/dist/purify.min.js\"></script>
 </head>
 <body>
-  <div class=\"container\">{('<div class=\\"header\\"><div><div class=\\"title\\">' + title + '</div></div></div>') if title else ''}
     <h2>You may be interested <span class=\"warn\" title=\"Links may be AI‑generated and might not resolve.\">⚠️</span></h2>
     <div class=\"stats\">{chips_section()}</div>
     {section('Models', cats['models'], 'models')}

             import json as _json
             doc_title = title or "Report"
             query = (query or "").strip()
+            header_html = f"<div class=\"header\"><div><div class=\"title\">{title}</div></div></div>" if title else ""
             # Extract URLs
             urls = re.findall(r"https?://[^\s)\]]+", final_answer or "")
   <script src=\"https://cdn.jsdelivr.net/npm/dompurify@3.1.6/dist/purify.min.js\"></script>
 </head>
 <body>
+  <div class=\"container\">{header_html}
     <h2>You may be interested <span class=\"warn\" title=\"Links may be AI‑generated and might not resolve.\">⚠️</span></h2>
     <div class=\"stats\">{chips_section()}</div>
     {section('Models', cats['models'], 'models')}