First Build

README.md

@@ -1,2 +1,5 @@
-# Sketch-Generator
+# Sketch Generator
  Convert images to sketch Using python and streamlit library
+
+# Suggession
+Try using high resolution images to get good results also the application does not perform well on images in dark theme.

app.py

@@ -0,0 +1,153 @@
+import streamlit as st
+import cv2
+import matplotlib.pyplot as pt
+import numpy as np
+from PIL import Image
+import io
+import requests
+import base64
+from tempfile import NamedTemporaryFile
+import logging
+
+logging.basicConfig(filename="logfile.log",
+                    format='%(asctime)s %(message)s',
+                    filemode='w')
+
+logger = logging.getLogger()
+logger.setLevel(logging.DEBUG)
+
+st.set_page_config(layout="wide")
+st.title("Sketch Generator")
+st.caption("Use high resolution image to get optimal results")
+      
+def verify_url(url):  
+    try:
+        encoded_content = url.split(',')[1]
+        image_data = base64.b64decode(encoded_content)
+        image = Image.open(io.BytesIO(image_data))
+        logger.info("Valid Base64 Url")
+        return True, image
+    except Exception as e:
+        logger.warning("Please Enter a valid Image url"+str(e))
+        return False, None
+    
+def http_url_image(nk04200111):
+    try:
+        # Send an HTTP request to the URL
+        response = requests.get(nk04200111)
+        response.raise_for_status()  # Raise an error for bad responses
+
+        # Open the image using PIL
+        image = Image.open(io.BytesIO(response.content))
+        logger.info("Valid http Url")
+        return True, image
+
+    except Exception as e:
+        logger.warning("Please Enter a valid Image url"+str(e))
+        return False, None
+
+def process_image(url_string,validation):
+    try:
+        if validation=="http":
+            result,upload = http_url_image(url_string)
+        elif validation=="data":
+            result,upload = verify_url(url_string)
+        if result:
+            image_np = np.array(upload)
+            firstimg=cv2.cvtColor(image_np,cv2.COLOR_BGR2GRAY)
+            blurs=cv2.bitwise_not(firstimg)
+            secondimg=cv2.GaussianBlur(blurs, (21, 21),sigmaX=30, sigmaY=30)
+            finalimg = cv2.divide(firstimg, 255 - secondimg, scale=256)
+            column1,column2,column3,column4=st.columns([1,4,4,1])
+            with column1:
+                pass
+            with column2:
+                st.image(upload, caption="Original Image", width=600, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
+            with column3:
+                if finalimg is not None:
+                    st.image(finalimg, caption="Final Image", width=600, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
+                            
+            with column4:
+                pass
+            logger.info("Image was processed successfully")
+        else:
+            st.write(f"The URL {url_string} does not contain a valid image.")
+            logger.warning(f"The URL {url_string} does not contain a valid image.")
+        return finalimg
+    except Exception as e:
+        logger.error("Exception occured while processing image: "+str(e))
+
+def get_base64_data(data, filename):
+    try:
+        base64_data = base64.b64encode(data).decode()
+        download_link = f'<a href="data:application/octet-stream;base64,{base64_data}" download="{filename}">Download {filename}</a>'
+        logger.info("Download Link created successfully")
+        return download_link
+    except Exception as e:
+        logger.error("Unable to create download link"+str(e))
+
+def save_image(finalimg):
+    try:
+        NKIMAGES=Image.fromarray(finalimg)
+        buffer = io.BytesIO()
+        NKIMAGES.save(buffer, format="PNG")
+        logger.info("Image was converted successfully")
+        # Provide a link for the user to download the processed image
+        st.markdown(get_base64_data(buffer.getvalue(), "Final Image.png"), unsafe_allow_html=True)
+    except Exception as e:
+        logger.error("Image conversion failed"+str(e))
+        
+
+def uploadfile():
+    try:
+        firstimg=None
+        secondimg=None
+        finalimg=None
+        val=1
+        selection=st.radio(" ", ["Enter Image Url:link:","Upload Image File:open_file_folder:"], index=None, disabled=False, horizontal=True, label_visibility="visible")
+        
+        if selection=="Enter Image Url:link:":
+            cols1,cols2=st.columns([5,1])
+            url_string=st.text_input(" ", type="default", placeholder="Enter the image Url", label_visibility="collapsed")
+            if st.button("Generate"):
+                if url_string !="" and url_string[:4]=="http":
+                    finalimg=process_image(url_string,"http")
+                    save_image(finalimg)
+                elif url_string !=""and url_string[:4]=="data":
+                    finalimg=process_image(url_string,"data")
+                    save_image(finalimg)
+                else:
+                    val=2
+            if val==2:
+                st.write("Enter a valid image url")
+        elif selection=="Upload Image File:open_file_folder:":
+            uploader=st.file_uploader("Upload the image here", type=None, accept_multiple_files=False, disabled=False, label_visibility="visible")
+            if uploader is None:
+                st.write("Please upload an image")
+            else:
+                upload=uploader.read()
+                if st.button("Generate"):
+                    image_np = np.array(Image.open(io.BytesIO(upload)))
+                    firstimg=cv2.cvtColor(image_np,cv2.COLOR_BGR2GRAY)
+                    blurs=cv2.bitwise_not(firstimg)
+                    secondimg=cv2.GaussianBlur(blurs, (21, 21),sigmaX=30, sigmaY=30)
+                    finalimg = cv2.divide(firstimg, 255 - secondimg, scale=256)
+                    
+                column1,column2,column3,column4=st.columns([1,4,4,1])
+                with column1:
+                    pass
+                with column2:
+                        st.image(upload, caption="Original Image", width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
+                with column3:
+                    if finalimg is not None:
+                        st.image(finalimg, caption="Final Image", width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
+                
+                with column4:
+                    pass
+                save_image(finalimg)
+                logger.info("Program ran successfully")
+    except Exception as e:
+        logger.error("Error occured in uploadfile"+str(e))
+        
+if __name__=="__main__":
+    uploadfile()

logfile.log

@@ -0,0 +1,6 @@
+2023-12-04 22:42:07,304 Starting new HTTPS connection (1): m.media-amazon.com:443
+2023-12-04 22:42:07,573 https://m.media-amazon.com:443 "GET /images/I/81eUYDkAdcL.jpg HTTP/1.1" 200 469714
+2023-12-04 22:42:07,624 Valid http Url
+2023-12-04 22:42:07,658 Image was processed successfully
+2023-12-04 22:42:07,716 Image was converted successfully
+2023-12-04 22:42:07,718 Download Link created successfully

requirements.txt

@@ -0,0 +1,53 @@
+altair==5.2.0
+attrs==23.1.0
+blinker==1.7.0
+cachetools==5.3.2
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+fonttools==4.46.0
+gitdb==4.0.11
+GitPython==3.1.40
+idna==3.6
+importlib-metadata==6.9.0
+Jinja2==3.1.2
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.2
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+mdurl==0.1.2
+numpy==1.26.2
+opencv-python==4.8.1.78
+packaging==23.2
+pandas==2.1.3
+Pillow==10.1.0
+protobuf==4.25.1
+pyarrow==14.0.1
+pydeck==0.8.1b0
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+pytz==2023.3.post1
+referencing==0.31.1
+requests==2.31.0
+rich==13.7.0
+rpds-py==0.13.2
+six==1.16.0
+smmap==5.0.1
+streamlit==1.29.0
+tenacity==8.2.3
+toml==0.10.2
+toolz==0.12.0
+tornado==6.4
+typing_extensions==4.8.0
+tzdata==2023.3
+tzlocal==5.2
+urllib3==2.1.0
+validators==0.22.0
+watchdog==3.0.0
+zipp==3.17.0