Updated Streamlit App

src/streamlit_app.py

@@ -1694,9 +1694,24 @@ def render_v2_training_panel(image_size: int, key_prefix: str = "v2_top"):
 
     if latest_model.exists():
         st.success(f"Local v2 model exists: {latest_model}")
+        st.download_button(
+            "Download model_v2.pt",
+            data=latest_model.read_bytes(),
+            file_name="model_v2.pt",
+            mime="application/octet-stream",
+            key=f"{key_prefix}_download_model",
+        )
     if latest_metrics.exists():
         try:
-            st.json(json.loads(latest_metrics.read_text()))
+            metrics_text = latest_metrics.read_text()
+            st.json(json.loads(metrics_text))
+            st.download_button(
+                "Download metrics_v2.json",
+                data=metrics_text,
+                file_name="metrics_v2.json",
+                mime="application/json",
+                key=f"{key_prefix}_download_metrics",
+            )
         except Exception:
             st.caption(f"Metrics file exists: {latest_metrics}")
 
@@ -2671,9 +2686,24 @@ with st.expander("Train v2 model"):
 
     if latest_model.exists():
         st.success(f"Local v2 model exists: {latest_model}")
+        st.download_button(
+            "Download model_v2.pt",
+            data=latest_model.read_bytes(),
+            file_name="model_v2.pt",
+            mime="application/octet-stream",
+            key="v2_bottom_download_model",
+        )
     if latest_metrics.exists():
         try:
-            st.json(json.loads(latest_metrics.read_text()))
+            metrics_text = latest_metrics.read_text()
+            st.json(json.loads(metrics_text))
+            st.download_button(
+                "Download metrics_v2.json",
+                data=metrics_text,
+                file_name="metrics_v2.json",
+                mime="application/json",
+                key="v2_bottom_download_metrics",
+            )
         except Exception:
             st.caption(f"Metrics file exists: {latest_metrics}")