Update app.py

app.py

@@ -315,38 +315,47 @@ 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 random
-import os
-
-# 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)
+def create_download_link(filename, base64_str):
+    href = f'<a href="data:file/png;base64,{base64_str}" download="{filename}">Download Image</a>'
+    return href
 
-# Get list of image paths from the /images directory
-image_paths = get_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"
+]
 
-# Select a random image path from the list
-selected_image_path = select_random_image_path(image_paths)
+# Select a random URL from the list
+selected_image_url = random.choice(image_urls)
 
-# Display the selected image in the sidebar
+# Get the base64 encoded string of the selected image
+st.write(selected_image_url)
 try:
-    if selected_image_path:
+    selected_image_base64 = get_image_as_base64(selected_image_url)
+    
+    if selected_image_base64 is not None:
         with st.sidebar:
             st.markdown("""### Graphic Novel AI""")
-            st.image(selected_image_path, caption=selected_image_path)
+            # 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)
     else:
-        st.sidebar.write("No images found in the directory.")
-except Exception as e:
-    st.write(f'Sidebar Fail - Check your Images. Error: {e}')
-
-
-
+        st.sidebar.write("Failed to load the image.")
+except:
+    st.write('Sidebar Fail - Check your Images')
+# ---- Art Card Sidebar with random selection of image.