Upload RadGenome demo Space

demo_log.py

@@ -2039,8 +2039,8 @@ class IFrame:
     """
     iframe = """
     <iframe id="diagram-html"
-        onload='this.style.height=this.contentWindow.document.body.scrollHeight + 20px;'
-        src="{html_path}?t={time}"
+        onload='this.style.height=this.contentWindow.document.body.scrollHeight + 20 + "px";'
+        srcdoc="{html_content}"
     ></iframe>
     """
     template = """
@@ -2056,23 +2056,26 @@ class IFrame:
     </div>
     """
 
-    def __init__(self, html_path=""):
-        self.html_path = html_path
+    def __init__(self, html_content=""):
+        self.html_content = html_content
         self.icon = IFrame.icon
         self.iframe = ""
-        if self.html_path:
-            self.iframe = IFrame.iframe.format(html_path=self.html_path, time=time.time())
+        if self.html_content:
+            import html as _html
+            self.iframe = IFrame.iframe.format(html_content=_html.escape(html_content, quote=True))
             self.icon = ""
-    
+
     @property
     def style(self):
-        if self.html_path:
+        if self.html_content:
             return ""
         return "min-height: 400px;"
     
     def render(self):
         return self.template.format(style=self.style, label_icon=IFrame.icon, icon=self.icon, iframe=self.iframe)
 
+
+
 class TextBox:
     icon = """
     <svg class="html-icon" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 25.5 18.14" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round">
@@ -2582,8 +2585,8 @@ def build_app(args, device=None, proc=None, tok=None, vlm=None, cxas_model=None)
                                 img_wrapper_json.update(explain_json)
                                 json_output = img_wrapper_json
                                 
-                                html_path = render_diagram_html(json_output)
-                                iframe = IFrame(html_path=html_path)
+                                html_content = render_diagram_html(json_output)
+                                iframe = IFrame(html_content=html_content)
 
                                 logger.finalize_reasoning(reasoning_wrapper)
 

render.py

@@ -178,13 +178,6 @@ def render_diagram_html(data):
 	css = env.get_template("template.css")
 	js = env.get_template("script.js")
 
-	output_folder = Path("static").resolve()
-	output_folder.mkdir(parents=True, exist_ok=True)
-
-	html_path = str(output_folder / "diagram.html")
-	css_path = str(output_folder / "styles.css")
-	js_path = str(output_folder / "script.js")
-
 	# Setup
 	svg_html_ls = []
 	tbox_left_html_ls = []
@@ -268,35 +261,23 @@ def render_diagram_html(data):
 		connector_css_ls.append(templates.diagram_html.connector_css.render(**data_point).strip("\n"))
 		i += 1
 
-	# gradio_prefix = ""
-	# html = html.render(
-	# 	stylesheet = f"{gradio_prefix}{css_path.replace('static/', '')}",
-	# 	javascript = f"{gradio_prefix}{js_path.replace('static/', '')}",
-	gradio_prefix = "/gradio_api/file="
-	html = html.render(
-		stylesheet = f"{gradio_prefix}{css_path.replace(chr(92), '/')}",
-		javascript = f"{gradio_prefix}{js_path.replace(chr(92), '/')}",
-		image_path = to_b64(data["image_path"]),
-		svgs = "\n".join(svg_html_ls),
-		tboxs_left = "\n".join(tbox_left_html_ls),
-		tboxs_right = "\n".join(tbox_right_html_ls),
-	)
-	css = css.render(
+	css_content = css.render(
 		svgs = "\n".join(svg_css_ls),
 		tboxs = "\n".join(tbox_css_ls),
 		connectors = "\n".join(connector_css_ls),
 	)
+	js_content = js.render()
 
-	with open(html_path, "w") as f:
-		f.write(html)
-
-	with open(css_path, "w") as f:
-		f.write(css)
-
-	with open(js_path, "w") as f:
-		f.write(js.render())
+	html_content = html.render(
+		stylesheet_content = css_content,
+		javascript_content = js_content,
+		image_path = to_b64(data["image_path"]),
+		svgs = "\n".join(svg_html_ls),
+		tboxs_left = "\n".join(tbox_left_html_ls),
+		tboxs_right = "\n".join(tbox_right_html_ls),
+	)
 
-	return f"{gradio_prefix}{html_path}"
+	return html_content
 
 
 import lorem

templates/template.html

@@ -4,7 +4,7 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Medical Image Analysis</title>
-    <link rel="stylesheet" href="{{stylesheet}}">
+    <style>{{stylesheet_content}}</style>
 </head>
 <body>
 	<div class="container">
@@ -21,6 +21,6 @@
 {{tboxs_right}}
 		</div>
 	</div>
-	<script src="{{javascript}}" defer></script>
+	<script>{{javascript_content}}</script>
 </body>
 </html>