Update app.py

app.py

@@ -1,6 +1,9 @@
-import gradio as gr
+import streamlit as st
 import numpy as np
 import similarSearch
+from PIL import Image
+
+df = pd.read_csv('./gallery.csv')
 
 # Function to find similar image IDs for a newly updated image
 def find_similar_images(new_image):
@@ -9,14 +12,24 @@ def find_similar_images(new_image):
     print(similar_ids)
     return similar_ids
 
-# Define Gradio interface
-iface = gr.Interface(
-    fn=find_similar_images, 
-    inputs=gr.inputs.Image(label="Upload Image"), 
-    outputs=gr.JSON(label="Similar Image IDs"),
-    title="Image Similarity Finder",
-    description="Upload an image to find similar image IDs in the database."
-)
+# Function to display images from their paths
+def display_images(image_ids):
+    for img_id in image_ids:
+        image_path = df.loc[df['seller_img_id'] == img_id, 'img_path'].values[0]
+        image = Image.open(image_path)
+        st.image(image, caption=img_id, use_column_width=True)
+
+def main():
+    st.title('Image Similarity Finder')
+    st.write('Upload an image to find similar image IDs in the database.')
+
+    uploaded_image_id = st.text_input('Enter Image ID', value='')
+
+    if st.button('Find Similar Images'):
+        similar_image_ids = find_similar_images(uploaded_image_id)
+        st.write('Similar Image IDs:', similar_image_ids)
+        st.write('Similar Images:')
+        display_images(similar_image_ids)
 
-# Launch the Gradio interface
-iface.launch()
+if __name__ == '__main__':
+    main()