create app.py

app.py

@@ -0,0 +1,84 @@
+
+import streamlit as st
+import asyncio
+from PIL import Image
+from app.upscale import ImageUpscaler
+
+img_upscale = ImageUpscaler()
+
+async def async_upscaler(image_stream, factor):
+    return await img_upscale.upscaler(target_img=image_stream, factor=factor, byte_stream=True)
+
+def calculate_new_dimension(original_size, entered_size, dimension_type):
+    original_width, original_height = original_size
+    aspect_ratio = original_width / original_height
+
+    if dimension_type == 'width':
+        new_width = entered_size
+        new_height = round(new_width / aspect_ratio)
+        return new_width, new_height
+    else: 
+        new_height = entered_size
+        new_width = round(new_height * aspect_ratio)
+        return new_width, new_height
+
+def calculate_factor(original_size, target_size):
+    return target_size[0] / original_size[0], target_size[1] / original_size[1]
+
+def main():
+    if 'user_factor' not in st.session_state:
+        st.session_state.user_factor = None
+
+    password = st.text_input("Enter Password", type='password')
+    if password != '1bz@123':
+        st.stop()
+    
+    cols = st.columns([3,5,2])
+    logo_url = 'https://media.licdn.com/dms/image/D560BAQGQxI71sZ1vTQ/company-logo_200_200/0/1687829291546?e=1703721600&v=beta&t=5RnOK7-Ak8cj1EQose8HQHWUYXPXRpI-x2dnjGjUABg'
+    cols[0].image(logo_url, width=int(200 * 0.8))
+    cols[1].markdown("<h1 style='text-align: left; color: black;'>IMAGE UPSCALER</h1>", unsafe_allow_html=True)
+    cols[1].markdown("This is a PoC application for upscaling images.")  
+    cols[1].markdown("[onebyzero.ai](http://www.onebyzero.ai)", unsafe_allow_html=True)
+  
+    uploaded_image = st.file_uploader('Upload target image (Required)', type=['jpeg', 'jpg', 'png'])
+
+    if uploaded_image is not None:
+        image = Image.open(uploaded_image)
+        Image_stream = img_upscale.pil2byte(image)
+        original_size = image.size
+        st.write(f"Original Image Resolution: {original_size[0]} x {original_size[1]}")
+
+        col1, mid_col, col2 = st.columns([3, 1, 3])
+        with col1:
+            dimension_type = st.radio("Dimension Type:", ('width', 'height'))
+            entered_size = st.number_input("Enter Target Width or Height:", min_value=1, value=original_size[0])
+            if st.button("Calculate Target Resolution"):
+                if dimension_type == 'width':
+                    target_size = calculate_new_dimension(original_size, entered_size, 'width')
+                else:
+                    target_size = calculate_new_dimension(original_size, entered_size, 'height')
+                st.write(f"Target Resolution: {target_size[0]} x {target_size[1]}")
+                factor_x, factor_y = calculate_factor(original_size, target_size)
+                st.session_state.user_factor = round(min(factor_x, factor_y), 2)
+                st.write(f"Calculated Upscaling Factor: {st.session_state.user_factor}")
+
+        with mid_col:
+            st.markdown("<h2 style='text-align: center;'>OR</h2>", unsafe_allow_html=True)  
+
+        with col2:
+            factor_slider = st.slider('Choose Upscaling Factor:', min_value=1.0, max_value=10.0, value=1.0, step=0.5, format="")
+            st.write(f"Selected Upscaling Factor: {factor_slider}")
+
+        if st.button("Generate"):
+            factor = st.session_state.user_factor if st.session_state.user_factor is not None else factor_slider
+            with st.spinner('Generating images...'):    
+                result_image_stream = asyncio.run(async_upscaler(Image_stream, factor))
+            try:
+                st.image(result_image_stream.getvalue())
+            except Exception as e:
+                st.error(f"An error occurred during image upscaling: {e}")
+
+if __name__ == "__main__":
+    main()
+    st.markdown(""" --- """)
+    st.markdown("<h6 style='text-align: left; color: grey;'>Created by <a href='mailto:[email protected]'>Sandeep Patra</a></h6>", unsafe_allow_html=True)