Update app.py

app.py

@@ -332,31 +332,31 @@ def create_download_link(filename, base64_str):
 
 # List of image URLs
 image_urls = [
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/KGvIFUpU1N-X2tX8hMRva.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/KvmVfcpAbowZIxcViGMmd.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/_ANl0q3ZGDa9CxQqpUmYP.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/SIGnnyVv7eLu8NnrtxdQP.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/5u06M9ue8FXK6Dsi2SF-I.png",
-    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/gTmhCUdpPaIJcmNXceDi4.png"
+    "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 URL from the list
 selected_image_url = random.choice(image_urls)
 
 # Get the base64 encoded string of the selected image
-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""")
-        # 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("Failed to load the image.")
+st.write(selected_image_url)
+try:
+    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""")
+            # 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("Failed to load the image.")
+except:
+    st.write('Sidebar Fail - Check your Images')
 # ---- Art Card Sidebar with random selection of image.