update app

app.py

@@ -6,6 +6,9 @@ import torch
 import chess
 import chess.svg
 import chess.pgn
+from svglib.svglib import svg2rlg
+from reportlab.graphics import renderPM
+from PIL import Image
 
 
 vocab_size=33
@@ -33,10 +36,12 @@ def generate(prompt):
     pgn_str = StringIO(pgn)
     game = chess.pgn.read_game(pgn_str)
     img = chess.svg.board(game.board())
-    filename = f'moves-{pgn}.svg'
-    with open(filename, 'w') as f:
+    filename = f'./moves-{pgn}'
+    with open(filename + '.svg', 'w') as f:
         f.write(img)
-    plot = f'<img src="{filename}">'
+    drawing = svg2rlg(filename + '.svg')
+    renderPM.drawToFile(drawing, f"{filename}.png", fmt="PNG")
+    plot = Image.open(f'{filename}.png')
     return pgn, plot
 
 
@@ -51,7 +56,7 @@ with gr.Blocks() as demo:
     """)
     prompt = gr.Text(label="PGN")
     output = gr.Text(label="Next turn", interactive=False)
-    img = gr.Markdown()
+    img = gr.Image()
     submit = gr.Button("Submit")
     submit.click(generate, [prompt], [output, img])
     

requirements.txt

@@ -1,2 +1,4 @@
 torch
-chess
+chess
+svglib
+rlPyCairo