added keep styling and image saving

app.py

@@ -24,7 +24,7 @@ st.sidebar.markdown('<p class="font">Afrodreams.AI</p>', unsafe_allow_html=True)
 with st.sidebar.expander("About the App"):
      st.write("""
         This app takes in your image and styles it with a unique african art.""")
-     style_weight = st.slider("Select Style Weight)
+     style_weight = st.slider("Select Style Weight", min_value=10, max_value=100, value=12)
 
 #Add file uploader to allow users to upload photos
 uploaded_file = st.file_uploader("", type=['jpg','png','jpeg'])
@@ -49,12 +49,14 @@ if uploaded_file is not None:
     params.backend = "mkl"
     params.image_size = 300
     params.content_image = uploaded_file
-    path = 'stylesv2'
-    styles = os.listdir(path)
-    params.style_weight = 12
     params.style_image = path + '/' + random.choice(styles)
-
+    keep_style = False
     if run==True:
+        # run image selection if keep style is false
+        if keep_style==False
+            path = 'stylesv2'
+            styles = os.listdir(path)
+            params.style_weight = style_weight
         st.session_state.submitted = True    
         my_bar.progress(neural_style.transfer(params))
 
@@ -64,9 +66,14 @@ if uploaded_file is not None:
                 result = Image.open('out.png')
                 st.image(result, width=500)
                 buf = BytesIO()
-                result.save(buf, format="JPEG")
+                result.save(buf, format="png")
+                if len(os.listdir('generated_sampples') <= 10:
+                    result.save(f"generated_sampples/{str(len(os.listdir('generated_sampples'))}.png")
                 byte_im = buf.getvalue()
                 run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
+        #keeping the current style by update the weight 
+        keep_style = st.sidebar.checkbox("Keep current style")
+