image upload

app.py

@@ -3,7 +3,45 @@ This is the main file for the streamlit app.
 Author: Sungjin Kim
 Date: 2023-05-19
 """
+import os
+from typing import Union
 import streamlit as st
+import requests
+from PIL import Image
+
+def download_image(image_url:str, 
+                   fname:str=None, 
+                   default_fold:str="images/") -> Union[str, None]:
+    """
+    download image from url
+    """
+    try:
+        image = requests.get(image_url, timeout=5)
+        if fname:
+            name_ext = fname
+        else:
+            name_ext = image_url.split("/")[-1]
+        name_ext = default_fold + name_ext
+        with open(name_ext, "wb") as f:
+        # with open(name, "wb") as f:
+            f.write(image.content)
+        return name_ext
+    except Exception as e_name:
+        print("An error occured", e_name)
+        return None
+
+def save_rgba_image(uploaded_file_name:str, uploaded_file_buffer:object,
+                    Default_fold="images/"):
+    """
+    # Save the uploaded file to a specific location
+    """
+    file_path = os.path.join(Default_fold[:-1], uploaded_file_name)
+    with open(file_path, "wb") as file:
+        file.write(uploaded_file_buffer)
+    image_file = Image.open(file_path)
+    image_file_rgba = image_file.convert("RGBA")
+    image_file_rgba.save(file_path)
+    return file_path
 
 def main():
     """
@@ -11,6 +49,25 @@ def main():
     """
     st.set_page_config(page_title="Avatar Generation")
     st.header("Artistic Avatar Generation 💖")
+
+    mode = st.radio("Choose your option", ("Upload photo by file", "Upload photo by url"))
+    photo_filename = None
+
+    if mode == "Upload photo by file":
+        uploaded_file = st.file_uploader("Upload your photo", type=["png"])
+        if uploaded_file is not None:
+            st.image(uploaded_file, width=256, caption="Your photo")
+            # Save the uploaded file to a specific location
+            photo_filename = save_rgba_image(uploaded_file.name, uploaded_file.getbuffer())
+    else:
+        photo_url = st.text_input("Enter your photo url:",
+                              value="https://pngimg.com/uploads/monkey/monkey_PNG18720.png") 
+        if photo_url:
+            st.image(photo_url, width=256, caption="Your photo")
+            photo_filename = download_image(photo_url) # e.g., "images/monkey_PNG18720.png"
+
+    st.write("Your photo filename:", photo_filename)
+
     x = st.slider('Slect a value')
     st.write(x, 'squared is', x * x)
 

requirements.txt

@@ -1 +1,3 @@
-streamlit
+streamlit
+requests
+Pillow