Show PGN text and provide option to download

app.py

@@ -37,9 +37,7 @@ def annotate_pgn(
         game, analysis_time=analysis_time_per_move, llm_character=style
     )
 
-    output_path = export_pgn(annotated_game)
-
-    return output_path
+    return str(annotated_game)
 
 
 def render_board(fen: Optional[str] = None, render_mode: str = "svg") -> str:
@@ -128,6 +126,11 @@ def play_chess(
 
 
 with gr.Blocks(title="ThinkSquare") as app:
+    def save_text_to_file(text):
+        with open("annotated_game.pgn", "w") as f:
+            f.write(text)
+        return "annotated_game.pgn"
+
     with gr.Tab("Play Chess"):
         gr.Markdown("### Play Chess with an engine")
         move_input = gr.Textbox(
@@ -178,7 +181,12 @@ with gr.Blocks(title="ThinkSquare") as app:
         )
         analyze_btn = gr.Button("Annotate PGN")
 
-        annotated_pgn_file = gr.File(label="Download Annotated PGN")
+        annotated_pgn_file = gr.Textbox(label="Annotated PGN")
+        download_button = gr.Button("Download Annotated PGN")
+        pgn_file_output = gr.File(label="Download your annotated PGN")
+        download_button.click(
+            fn=save_text_to_file, inputs=annotated_pgn_file, outputs=pgn_file_output
+        )
 
         analyze_btn.click(
             fn=annotate_pgn,
@@ -206,4 +214,6 @@ with gr.Blocks(title="ThinkSquare") as app:
             outputs=board_render_output,
         )
 
-app.launch(mcp_server=True)
+
+if __name__ == "__main__":
+    app.launch(mcp_server=True)