Update app.py

app.py

@@ -571,12 +571,13 @@ image_urls = [
     "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/cmCZ5RTdSx3usMm7MwwWK.png",
 ]
 
-selected_image_url = random.choice(image_urls)
-selected_image_base64 = get_image_as_base64(selected_image_url)
-if selected_image_base64 is not None:
-    with st.sidebar:
-        st.markdown(f"![image](data:image/png;base64,{selected_image_base64})")
-else:
+try:
+    selected_image_url = random.choice(image_urls)
+    selected_image_base64 = get_image_as_base64(selected_image_url)
+    if selected_image_base64 is not None:
+        with st.sidebar:
+            st.markdown(f"![image](data:image/png;base64,{selected_image_base64})")
+except:
     st.sidebar.write("Failed to load the image.")
 
 # Ensure the directory for storing scores exists
@@ -1387,10 +1388,12 @@ if example_input:
     # Search AI
     query=example_input
     if query: 
-        result = search_arxiv(query)
-        #search_glossary(query)
-        search_glossary(result)
-    st.markdown(' ')
+        try:
+            result = search_arxiv(query)
+            #search_glossary(query)
+            search_glossary(result)
+        except:
+            st.markdown(' ')
                              
 st.write("Search history:")
 for example_input in session_state["search_queries"]: