Update app.py

app.py

@@ -152,7 +152,10 @@ def run_main_app(semantic_concept, word, letter, font_name, num_steps, example=0
     preprocess(cfg.font, cfg.word, cfg.optimized_letter, cfg.level_of_cc)
     filename_init = os.path.join("code/data/init/", f"{cfg.font}_{cfg.word}_scaled.svg").replace(" ", "_")
     if not example:
-        yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(visible=False)
+        with open(filename_init, "r") as f0:
+            svg_content_init = f0.read()
+        # yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(visible=False)
+        yield gr.update(value=f'<div>{svg_content_init}</div>',label="Initial Word",visible=True),gr.update(visible=False),gr.update(visible=False)
 
     sds_loss = SDSLoss(cfg, device, model)
 
@@ -207,7 +210,10 @@ def run_main_app(semantic_concept, word, letter, font_name, num_steps, example=0
         check_and_create_dir(filename)
         save_svg.save_svg(filename, w, h, shapes, shape_groups)
         if not example:
-            yield gr.update(visible=True),gr.update(value=filename, label=f'iters: {step} / {num_iter}', visible=True),gr.update(visible=False)
+            # yield gr.update(visible=True),gr.update(value=filename, label=f'iters: {step} / {num_iter}', visible=True),gr.update(visible=False)
+            with open(filename, "r") as f:
+                svg_content = f.read()
+            yield gr.update(visible=True), gr.update(value=f'<div>{svg_content}</div>',label=f'iters: {step} / {num_iter}',visible=True),gr.update(visible=False)
 
         x = img.unsqueeze(0).permute(0, 3, 1, 2)  # HWC -> NCHW
         x = x.repeat(cfg.batch_size, 1, 1, 1)
@@ -298,8 +304,8 @@ with gr.Blocks() as demo:
 
         with gr.Column():
             result0 = gr.HTML() #gr.Image(type="filepath", label="Initial Word",height=333) 
-            result1 = gr.Image(type="filepath", label="Optimization Process",height=100) 
-            result2 = gr.HTML(label="Final Result") #gr.Image(type="filepath", label="Final Result",visible=False,height=333) 
+            result1 = gr.HTML() #gr.Image(type="filepath", label="Optimization Process",height=100) 
+            result2 = gr.HTML() #gr.Image(type="filepath", label="Final Result",visible=False,height=333) 
             
         
     with gr.Row():