Update app.py

app.py

@@ -68,7 +68,7 @@ def image_search(query, corpus, max_results=3):
         dot_product2 = dot_product2 / np.max(dot_product2, axis=0, keepdims=True)
         dot_product -= np.max(np.maximum(dot_product2, 0), axis=1)
 
-    results = np.argsort(dot_product)[::-1]
+    results = np.argsort(dot_product)[-1 : -n_results - 1 : -1]
     return [
         (
             df[k].iloc[i]["path"],
@@ -122,29 +122,32 @@ def main():
         query = st.sidebar.text_input("Query", value="lighthouse")
     corpus = "Unsplash"
     
+    # Wrap the content inside st.spinner for the "Submit" button
     if st.sidebar.button("Submit"):
-        if len(query) > 0:
-            results = image_search(query, corpus)
-            clicked = clickable_images(
-                [result[0] for result in results],
-                titles=[result[1] for result in results],
-                div_style={
-                    "display": "flex",
-                    "justify-content": "center",
-                    "flex-wrap": "wrap",
-                },
-                img_style={"margin": "2px", "height": "200px"},
-            )
-            if clicked >= 0:
-                change_query = False
-                if "last_clicked" not in st.session_state:
-                    change_query = True
-                else:
-                    if clicked != st.session_state["last_clicked"]:
+        with st.spinner("Searching..."):
+            time.sleep(2)  # Simulate a loading delay (replace with actual image search function)
+            if len(query) > 0:
+                results = image_search(query, corpus)
+                clicked = clickable_images(
+                    [result[0] for result in results],
+                    titles=[result[1] for result in results],
+                    div_style={
+                        "display": "flex",
+                        "justify-content": "center",
+                        "flex-wrap": "wrap",
+                    },
+                    img_style={"margin": "2px", "height": "200px"},
+                )
+                if clicked >= 0:
+                    change_query = False
+                    if "last_clicked" not in st.session_state:
                         change_query = True
-                if change_query:
-                    st.session_state["query"] = f"[{corpus}:{results[clicked][2]}]"
-                    st.experimental_rerun()
+                    else:
+                        if clicked != st.session_state["last_clicked"]:
+                            change_query = True
+                    if change_query:
+                        st.session_state["query"] = f"[{corpus}:{results[clicked][2]}]"
+                        st.experimental_rerun()
 
     st.sidebar.info("""
     Enter your query and hit enter
@@ -155,4 +158,4 @@ def main():
                     """)
     
 if __name__ == "__main__":
-    main()
+    main()