Update app.py

app.py

@@ -1,4 +1,3 @@
-from html import escape
 import re
 import streamlit as st
 import pandas as pd
@@ -6,14 +5,6 @@ import numpy as np
 from transformers import CLIPProcessor, CLIPModel
 from st_clickable_images import clickable_images
 
-@st.cache(
-    show_spinner=False,
-    hash_funcs={
-        CLIPModel: lambda _: None,
-        CLIPProcessor: lambda _: None,
-        dict: lambda _: None,
-    },
-)
 def load():
     model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14")
     processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
@@ -87,17 +78,6 @@ def image_search(query, corpus, n_results=24):
         for i in results
     ]
 
-description = """
-# Semantic image search
-**Enter your query and hit enter**
-"""
-
-howto = """
-- Click image to find similar images
-- Use "**;**" to combine multiple queries)
-- Use "**EXCLUDING**", to exclude a query
-"""
-
 def main():
     st.markdown(
         """
@@ -132,36 +112,42 @@ def main():
               </style>""",
         unsafe_allow_html=True,
     )
-    st.markdown(description)
-    st.markdown(howto)
+    
+    st.sidebar.markdown("# Semantic Image Search")
+    st.sidebar.markdown("**Enter your query and hit enter**")
+    st.sidebar.markdown("- Click image to find similar images")
+    st.sidebar.markdown("- Use '**;**' to combine multiple queries")
+    st.sidebar.markdown("- Use '**EXCLUDING**' to exclude a query")
 
     if "query" in st.session_state:
-        query = st.text_input("", value=st.session_state["query"])
+        query = st.sidebar.text_input("Query", value=st.session_state["query"])
     else:
-        query = st.text_input("", value="lighthouse")
-    corpus = st.radio("", ["Unsplash"])
-    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"]:
+        query = st.sidebar.text_input("Query", value="lighthouse")
+    corpus = st.sidebar.radio("Corpus", ["Unsplash"])
+    
+    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
-            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()
 
 if __name__ == "__main__":
     main()