Update app.py

app.py

@@ -161,6 +161,8 @@ with col:
     
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
+    fname = uploaded_file.strip(".fits")
+    
     with fits.open(uploaded_file) as hdul:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
@@ -227,12 +229,11 @@ if uploaded_file is not None:
             except: image_decomposed = np.zeros((128,128))
             plot_decomposed(image_decomposed)
         
-        with col6:
-            shutil.make_archive("predictions", 'zip', "predictions")
-            with open('predictions.zip', 'rb') as f:
-                res = f.read()
-            st.markdown("")
-            st.markdown("")
-            st.markdown("")
-            # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
-            download = st.download_button(label="Download", data=res, file_name='prediction.zip', mime="application/octet-stream")
+            with col6:
+                shutil.make_archive("predictions", 'zip', "predictions")
+                with open('predictions.zip', 'rb') as f:
+                    res = f.read()
+                st.markdown("")
+                st.markdown("")
+                # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
+                download = st.download_button(label="Download", data=res, file_name=f'{fname}_{int(scale*128)}.zip', mime="application/octet-stream")