fix: clear image cache

- specify only for URL images

app.py

@@ -34,24 +34,24 @@ content_image, style_image = st.columns(2)
 with content_image:
     ContentColumnTitle = "## 🖼️ Content Image 🖼️"
     ContentImageSelectionPrompt = "Pick a Content image"
-    content_image_file = upload_image(
+    content_image_file, content_image_upload_method = upload_image(
         ContentColumnTitle, ContentImageSelectionPrompt, "Content", "content"
     )
 
 with style_image:
     StyleColumnTitle = "## 🎨 Style Image 🎨"
     StyleImageSelectionPrompt = "Pick a Style image"
-    style_image_file = upload_image(
+    style_image_file, style_image_upload_method = upload_image(
         StyleColumnTitle, StyleImageSelectionPrompt, "Style", "style"
     )
 
 if None not in (content_image_file, style_image_file):
-
-    # CLEAR IMAGES
-    clear_images = st.button(
-        label="🔄❌ Clear Image Cache ❌🔄",
-        on_click=remove_source_images(),
-    )
+    if "🔗 URL" in (content_image_upload_method, style_image_upload_method):
+        # CLEAR IMAGES
+        clear_images = st.button(
+            label="🔄❌ Clear URL Image Cache ❌🔄",
+            on_click=remove_source_images(),
+        )
 
     # STYLIZE CONTENT IMAGE
     stylize_image = st.button("🖼️🖌️🎨 Start Neural Style Transfer 🖼️🖌️🎨")

src/data/data.py

@@ -48,6 +48,6 @@ def upload_image(ColumnTitle, ImageSelectionPrompt, ImageType, KeyString):
     try:
         st.write(f"{ImageType} Image")
         st.image(imshow(image_file))
-        return image_file
+        return image_file, image_upload_method
     except:
         pass