Add util to convert string to pgn format

src/util/pgn_util.py

@@ -1,13 +1,35 @@
+import io
 import logging
 import chess.pgn
 import tempfile
 import os
 import time
 import glob
+import gradio as gr
+
+
+def extract_game(png_input):
+    if isinstance(png_input, io.IOBase):
+        file = png_input
+    elif isinstance(png_input, str):
+        file = io.StringIO(png_input)
+    elif isinstance(png_input, gr.utils.NamedString):
+        file = png_input.name
+    else:
+        raise ValueError("Input must be a string or a file-like object.")
+
+    game = read_pgn(file)
+    return game
+
+
+def pgn_string_to_game(pgn_text: str) -> chess.pgn.Game:
+    pgn_io = io.StringIO(pgn_text)
+    game = chess.pgn.read_game(pgn_io)
+    return game
 
 
 # Load game
-def read_pgn(pgn_filepath):
+def read_pgn(pgn_filepath) -> str:
     with open(pgn_filepath) as f:
         game = chess.pgn.read_game(f)
     return game