Save processed files and add download support

app.py

@@ -1,4 +1,5 @@
 import json
+import tempfile
 from pathlib import Path
 
 import gradio as gr
@@ -7,10 +8,12 @@ import spaces
 from document_processor import DocumentProcessor
 from normalizer import DocumentNormalizer
 
-
 processor = DocumentProcessor()
 normalizer = DocumentNormalizer()
 
+OUTPUT_DIR = Path(tempfile.gettempdir()) / "ai-tutor-output"
+OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+
 
 @spaces.GPU(duration=120)
 def process_document(file):
@@ -36,6 +39,27 @@ def process_document(file):
 
         output = normalized.model_dump()
 
+        base_name = path.stem
+        json_path = OUTPUT_DIR / f"{base_name}_normalized.json"
+        markdown_path = OUTPUT_DIR / f"{base_name}_markdown.md"
+        stats_path = OUTPUT_DIR / f"{base_name}_statistics.json"
+
+        json_path.write_text(
+            json.dumps(output, indent=2, ensure_ascii=False),
+            encoding="utf-8",
+        )
+        markdown_path.write_text(result["markdown"], encoding="utf-8")
+        stats_path.write_text(
+            json.dumps(normalized.statistics, indent=2, ensure_ascii=False),
+            encoding="utf-8",
+        )
+
+        download_files = [
+            (str(json_path), json_path.name),
+            (str(markdown_path), markdown_path.name),
+            (str(stats_path), stats_path.name),
+        ]
+
         return (
             json.dumps(
                 output,
@@ -48,6 +72,7 @@ def process_document(file):
                 indent=2,
                 ensure_ascii=False,
             ),
+            download_files,
         )
 
     except Exception as e:
@@ -99,6 +124,13 @@ with gr.Blocks(
         language="json",
     )
 
+    gr.Markdown("## Download")
+
+    download_output = gr.File(
+        label="Download Processed Files",
+        file_count="multiple",
+    )
+
     with gr.Tabs():
 
         with gr.Tab("AI Tutor JSON"):
@@ -119,6 +151,7 @@ with gr.Blocks(
             json_output,
             markdown_output,
             statistics_output,
+            download_output,
         ],
         api_name="process_document",
     )