Hotfix: avoid gr.JSON schema crash in Gradio

app.py

@@ -146,7 +146,7 @@ def load_selected_model(checkpoint_label):
     }
     status = f"Loaded `{experiment}` on `{device}` (`{cfg['model_type']}`)"
     suggested_out = os.path.join("analysis", "outputs_ui", experiment)
-    return bundle, status, model_info, cfg["inference"]["num_steps"], suggested_out
+    return bundle, status, json.dumps(model_info, ensure_ascii=False, indent=2), cfg["inference"]["num_steps"], suggested_out
 
 
 def apply_preset(preset_name):
@@ -246,7 +246,7 @@ def generate_from_ui(
     }
     log_path = save_generation(model_bundle["experiment"], record)
     status = f"Inference done. Saved: `{log_path}`"
-    return output_text, status, record
+    return output_text, status, json.dumps(record, ensure_ascii=False, indent=2)
 
 
 def _run_analysis_cmd(task, ckpt_path, output_dir, input_text="dharmo rakṣati rakṣitaḥ", phase="analyze"):
@@ -375,7 +375,7 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
             load_btn = gr.Button("Load Selected Model", variant="primary")
 
     load_status = gr.Markdown("Select a model and load.")
-    model_info = gr.JSON(label="Loaded Model Details")
+    model_info = gr.Textbox(label="Loaded Model Details (JSON)", lines=12, interactive=False)
 
     with gr.Tabs():
         with gr.Tab("1) Task Runner"):
@@ -435,7 +435,7 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
                         interactive=False,
                     )
                     run_status = gr.Markdown("")
-                    run_record = gr.JSON(label="Inference Metadata")
+                    run_record = gr.Textbox(label="Inference Metadata (JSON)", lines=12, interactive=False)
                 with gr.Column(scale=1, elem_classes=["panel"]):
                     preset = gr.Radio(["Manual", "Literal", "Balanced", "Creative"], value="Balanced", label="Preset")
                     temperature = gr.Slider(0.4, 1.2, value=0.70, step=0.05, label="Temperature")