Update app.py

app.py

@@ -316,48 +316,36 @@ FileSidebar()
 
 
 
-    
-# ---- Art Card Sidebar with Random Selection of image:
-def get_image_as_base64(url):
-    response = requests.get(url)
-    if response.status_code == 200:
-        # Convert the image to base64
-        return base64.b64encode(response.content).decode("utf-8")
-    else:
-        return None
+    import streamlit as st
+import random
+import os
 
-def create_download_link(filename, base64_str):
-    href = f'<a href="data:file/png;base64,{base64_str}" download="{filename}">Download Image</a>'
-    return href
+# Function to get a list of image paths from the /images directory
+def get_image_paths(directory="/images"):
+    image_paths = [os.path.join(directory, file) for file in os.listdir(directory) if file.endswith(('png', 'jpg', 'jpeg'))]
+    return image_paths
+
+# Function to select a random image path
+def select_random_image_path(image_paths):
+    return random.choice(image_paths)
 
-# List of image URLs
-image_urls = [
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/W1omJItftG3OkW9sj-Ckb.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/Djx-k4WOxzlXEQPzllP3r.png"
-]
+# Get list of image paths from the /images directory
+image_paths = get_image_paths()
 
-# Select a random URL from the list
-selected_image_url = random.choice(image_urls)
+# Select a random image path from the list
+selected_image_path = select_random_image_path(image_paths)
 
-# Get the base64 encoded string of the selected image
-st.write(selected_image_url)
+# Display the selected image in the sidebar
 try:
-    selected_image_base64 = get_image_as_base64(selected_image_url)
-    
-    if selected_image_base64 is not None:
+    if selected_image_path:
         with st.sidebar:
             st.markdown("""### Graphic Novel AI""")
-            # Display the image
-            st.markdown(f"![image](data:image/png;base64,{selected_image_base64})")
-            
-            # Create and display the download link
-            download_link = create_download_link("downloaded_image.png", selected_image_base64)
-            st.markdown(download_link, unsafe_allow_html=True)
+            st.image(selected_image_path, caption=selected_image_path)
     else:
-        st.sidebar.write("Failed to load the image.")
-except:
-    st.write('Sidebar Fail - Check your Images')
-# ---- Art Card Sidebar with random selection of image.
+        st.sidebar.write("No images found in the directory.")
+except Exception as e:
+    st.write(f'Sidebar Fail - Check your Images. Error: {e}')
+