Update app.py

app.py

@@ -47,27 +47,42 @@ def fig_to_png_bytes(fig, dpi=600):
     buf.seek(0)
     return buf.getvalue()
 
-def render_plot_with_download(fig, *, title: str, filename: str, export_dpi: int = 600):
-    # 1) SAVE FIRST (before any clear)
+def render_plot_with_download(
+    fig,
+    *,
+    title: str,
+    filename: str,
+    export_dpi: int = 600,
+    key: str | None = None   # <-- ADD THIS
+):
+    """
+    Show compact plot + provide high-DPI PNG download.
+    Fixes:
+    • blank downloads
+    • figure clearing issues
+    • duplicate widget keys
+    """
+
+    # Save BEFORE any clearing
     png_bytes = fig_to_png_bytes(fig, dpi=export_dpi)
 
-    # 2) DISPLAY (do NOT clear)
-    st.pyplot(fig, clear_figure=False, use_container_width=False)
+    # Show
+    st.pyplot(fig, clear_figure=False, use_container_width=True)
 
-    # 3) DOWNLOAD
+    # Download
     st.download_button(
         label=f"Download {title} (PNG {export_dpi} dpi)",
         data=png_bytes,
         file_name=filename,
         mime="image/png",
-        key=f"dl_{filename}"
+        key=key or f"dl_{filename}"
     )
 
-    # 4) Prevent figure leakage / overlaps
     plt.close(fig)
 
 
 
+
 # ============================================================
 # Fixed schema definition (PLACEHOLDER FRAMEWORK)
 # ============================================================