Update app.py

app.py

@@ -237,15 +237,15 @@ def run_main_app(semantic_concept, word, letter, font_name, num_steps, example=0
     combine_word(cfg.word, cfg.optimized_letter, cfg.font, cfg.experiment_dir)
 
     image = os.path.join(cfg.experiment_dir,f"{cfg.font}_{cfg.word}_{cfg.optimized_letter}.svg")
-    yield gr.update(value=[filename_init],visible=True),gr.update(visible=False),gr.update(value=[image],visible=True)
+    yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(value=image,visible=True)
 
-    # with open(filename_init, "r") as f0:
-    #     svg_content_init = f0.read()
+    with open(filename_init, "r") as f0:
+        svg_content_init = f0.read()
 
-    # with open(image, "r") as f1:
-    #     svg_content_final = f1.read()
+    with open(image, "r") as f1:
+        svg_content_final = f1.read()
         
-    # yield gr.update(value=f"<div>{svg_content_init}</div>",label="Initial Word",visible=True),gr.update(visible=False),gr.update(value=f"<div>{svg_content_final}</div>",label="Final Result",visible=True)
+    yield gr.update(value=f'<div style="width: 333px; height: 333px;">{svg_content_init}</div>',label="Initial Word",visible=True),gr.update(visible=False),gr.update(value=f"<div>{svg_content_final}</div>",label="Final Result",visible=True)
  
 
 with gr.Blocks() as demo:
@@ -297,9 +297,9 @@ with gr.Blocks() as demo:
             run = gr.Button('Generate')
 
         with gr.Column():
-            result0 = gr.Gallery(label="Initial Word", value=None)#gr.HTML("""<style>{width: 333px; height: 333px;}</style>""",label="Initial Word") #gr.Image(type="filepath", label="Initial Word",height=333) 
-            result1 = gr.Gallery(label="Optimization Process", value=None)#gr.Image(type="filepath", label="Optimization Process",height=100) 
-            result2 = gr.Gallery(label="Final Result", value=None)#gr.HTML(label="Final Result") #gr.Image(type="filepath", label="Final Result",visible=False,height=333) 
+            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) 
             
         
     with gr.Row():