Update app.py

app.py

@@ -75,7 +75,8 @@ def inference(image_path_or_url, learn):
     return tensor_to_pil(img_hr)
 
 # Streamlit application
-st.title("Image Inference with Fastai")
+st.title("Image to Sketch")
+st.write("Convert any image to its sketch version using a Pix2Pix GAN Model.")
 
 # Download the model file from the Hugging Face repository
 model_url = "https://huggingface.co/Hammad712/image2sketch/resolve/main/image2sketch.pkl"
@@ -93,14 +94,34 @@ with tempfile.TemporaryDirectory() as tmpdirname:
     shutil.move(model_file_path, os.path.join(tmpdirname, 'export.pkl'))
     learn = load_learner(tmpdirname)
 
+# Sidebar to display previous generated images
+st.sidebar.title("Previous Results")
+if 'generated_images' not in st.session_state:
+    st.session_state['generated_images'] = []
+
+for img in st.session_state['generated_images']:
+    st.sidebar.image(img, use_column_width=True)
+
 # Input for image URL or path
-image_path_or_url = st.text_input("Enter image path or URL", "")
+image_path_or_url = st.text_input("Enter image URL", "")
 
 # Run inference button
-if st.button("Run Inference"):
+if st.button("Convert"):
     if image_path_or_url:
         with st.spinner('Processing...'):
             high_res_image = inference(image_path_or_url, learn)
-            st.image(high_res_image, caption='High Resolution Image', use_column_width=True)
+            original_image = PIL.Image.open(BytesIO(requests.get(image_path_or_url).content)) if image_path_or_url.startswith('http') else PIL.Image.open(image_path_or_url)
+
+            # Display original and high-res images side by side
+            col1, col2 = st.columns(2)
+
+            with col1:
+                st.image(original_image, caption='Original Image', use_column_width=True)
+            with col2:
+                st.image(high_res_image, caption='Sketch Image', use_column_width=True)
+
+            # Save the generated image to session state for sidebar display
+            st.session_state['generated_images'].append(high_res_image)
+
     else:
         st.error("Please enter a valid image path or URL.")