feat: add image capture

- URL image cache using exception handling
- image input using camera

app.py

@@ -34,24 +34,26 @@ content_image, style_image = st.columns(2)
 with content_image:
     ContentColumnTitle = "## 🖼️ Content Image 🖼️"
     ContentImageSelectionPrompt = "Pick a Content image"
-    content_image_file, content_image_upload_method = upload_image(
+    content_image_file = upload_image(
         ContentColumnTitle, ContentImageSelectionPrompt, "Content", "content"
     )
 
 with style_image:
     StyleColumnTitle = "## 🎨 Style Image 🎨"
     StyleImageSelectionPrompt = "Pick a Style image"
-    style_image_file, style_image_upload_method = upload_image(
+    style_image_file = upload_image(
         StyleColumnTitle, StyleImageSelectionPrompt, "Style", "style"
     )
 
 if None not in (content_image_file, style_image_file):
-    if "🔗 URL" in (content_image_upload_method, style_image_upload_method):
+    try:
         # CLEAR IMAGES
         clear_images = st.button(
-            label="🔄❌ Clear URL Image Cache ❌🔄",
+            label="🔄❌ Clear Image Cache ❌🔄",
             on_click=remove_source_images(),
         )
+    except:
+        pass
 
     # STYLIZE CONTENT IMAGE
     stylize_image = st.button("🖼️🖌️🎨 Start Neural Style Transfer 🖼️🖌️🎨")
@@ -65,6 +67,5 @@ if None not in (content_image_file, style_image_file):
         except:
             pass
 
-
 else:
     st.write("Please upload content and style image to go to next step.")

src/data/data.py

@@ -5,6 +5,21 @@ import tensorflow as tf
 from ..image_utils import load_img, imshow, transform_img
 
 
+def upload_image_url(ImageSelectionPrompt, ImageType, image_upload_method):
+    st.write(f"{ImageSelectionPrompt}: {image_upload_method}")
+    url = st.text_input(f"{ImageType} Image URL")
+    try:
+        image_path = tf.keras.utils.get_file(
+            os.path.join(os.getcwd(), f"{ImageType.lower()}.jpg"), url
+        )
+    except:
+        pass
+    try:
+        return load_img(image_path)
+    except:
+        pass
+
+
 def upload_image_file(ImageSelectionPrompt, ImageType, image_upload_method):
     st.write(f"{ImageSelectionPrompt}: {image_upload_method}")
     image_file = st.file_uploader(
@@ -17,17 +32,12 @@ def upload_image_file(ImageSelectionPrompt, ImageType, image_upload_method):
         pass
 
 
-def upload_image_url(ImageSelectionPrompt, ImageType, image_upload_method):
+def upload_image_capture(ImageSelectionPrompt, ImageType, image_upload_method):
     st.write(f"{ImageSelectionPrompt}: {image_upload_method}")
-    url = st.text_input(f"{ImageType} Image URL")
+    image_file = st.camera_input(f"Capture {ImageType} Image")
     try:
-        image_path = tf.keras.utils.get_file(
-            os.path.join(os.getcwd(), f"{ImageType.lower()}.jpg"), url
-        )
-    except:
-        pass
-    try:
-        return load_img(image_path)
+        image_file = image_file.read()
+        return transform_img(image_file)
     except:
         pass
 
@@ -35,19 +45,25 @@ def upload_image_url(ImageSelectionPrompt, ImageType, image_upload_method):
 def upload_image(ColumnTitle, ImageSelectionPrompt, ImageType, KeyString):
     st.write(ColumnTitle)
     image_upload_method = st.radio(
-        label="", options=["📁 File Upload", "🔗 URL"], key=KeyString
+        label="", options=["🔗 URL", "📁 File Upload", "📸 Capture"], key=KeyString
     )
+    if image_upload_method == "🔗 URL":
+        image_file = upload_image_url(
+            ImageSelectionPrompt, ImageType, image_upload_method
+        )
+
     if image_upload_method == "📁 File Upload":
         image_file = upload_image_file(
             ImageSelectionPrompt, ImageType, image_upload_method
         )
-    if image_upload_method == "🔗 URL":
-        image_file = upload_image_url(
+
+    if image_upload_method == "📸 Capture":
+        image_file = upload_image_capture(
             ImageSelectionPrompt, ImageType, image_upload_method
         )
     try:
         st.write(f"{ImageType} Image")
         st.image(imshow(image_file))
-        return image_file, image_upload_method
+        return image_file
     except:
         pass