new

app.py

@@ -8,7 +8,7 @@ import numpy as np
 import torch
 
 # ✅ Initialize Pinecone
-pc = Pinecone(api_key="pcsk_6r4DPn_4P9LckhZak3PhebvSebnEBKQZuzYFeJL2X93LtLxZVBxyJ93inBAktefa8usvJC")  # Replace with your API key
+pc = Pinecone(api_key="your-pinecone-api-key")  # Replace with your Pinecone API key
 index_name = "unsplash-index"
 unsplash_index = pc.Index(index_name)
 
@@ -26,16 +26,14 @@ def get_text_embedding(text):
     inputs = processor(text=[text], return_tensors="pt", padding=True, truncation=True)
     with torch.no_grad():
         text_features = model.get_text_features(**inputs)
-    embedding = text_features.detach().cpu().numpy().flatten().tolist()
-    return embedding
+    return text_features.detach().cpu().numpy().flatten().tolist()
 
 # ✅ Function to Generate Embedding from Image
 def get_image_embedding(image):
     inputs = processor(images=image, return_tensors="pt")
     with torch.no_grad():
         image_features = model.get_image_features(**inputs)
-    embedding = image_features.detach().cpu().numpy().flatten().tolist()
-    return embedding
+    return image_features.detach().cpu().numpy().flatten().tolist()
 
 # ✅ Function to Query Pinecone and Fetch Similar Images
 def search_similar_images(embedding, top_k=10):
@@ -47,6 +45,14 @@ def search_similar_images(embedding, top_k=10):
     )
     return results.get("matches", [])
 
+# ✅ Sidebar for Customization
+st.sidebar.title("⚙️ Settings")
+top_k = st.sidebar.slider("🔢 Number of Similar Images", 1, 20, 10)
+theme = st.sidebar.radio("🎨 Theme", ["Light Mode", "Dark Mode"], index=0)
+
+# Apply Theme
+st.set_page_config(page_title="Image Search App", layout="wide", initial_sidebar_state="expanded")
+
 # ✅ Streamlit UI
 st.title("🔍 Image & Text Search with CLIP & Pinecone")
 st.write("Search for images using text or upload an image to find similar ones!")
@@ -60,21 +66,25 @@ if st.button("🔍 Search by Text"):
         with st.spinner("Generating embedding..."):
             embedding = get_text_embedding(search_query)
         with st.spinner("Searching for similar images..."):
-            matches = search_similar_images(embedding, top_k=10)
+            matches = search_similar_images(embedding, top_k=top_k)
 
         st.subheader("🔎 Top Similar Images")
-        for match in matches:
-            score = match.get("score", 0)
-            photo_id = match.get("id", "Unknown ID")
-            url = match.get("metadata", {}).get("url", None)
-
-            st.write(f"**Photo ID**: {photo_id} | **Similarity Score**: {score:.4f}")
-            if url:
-                st.image(url, caption=f"Photo ID: {photo_id}", use_container_width=True)
-            else:
-                st.warning(f"Image URL not found for Photo ID: {photo_id}")
-    else:
-        st.warning("⚠️ Please enter a search query!")
+        
+        if matches:
+            cols = st.columns(3)  # Arrange images in 3 columns
+            for i, match in enumerate(matches):
+                score = match.get("score", 0)
+                photo_id = match.get("id", "Unknown ID")
+                url = match.get("metadata", {}).get("url", None)
+
+                with cols[i % 3]:  # Alternate images in columns
+                    st.write(f"📷 **Photo ID**: {photo_id} | 🔢 **Score**: {score:.4f}")
+                    if url:
+                        st.image(url, caption=f"Photo ID: {photo_id}", use_container_width=True)
+                    else:
+                        st.warning(f"⚠️ Image URL not found for Photo ID: {photo_id}")
+        else:
+            st.warning("⚠️ No similar images found!")
 
 # 📌 **Option 2: Image-to-Image Search**
 st.subheader("🖼️ Search by Image")
@@ -89,20 +99,51 @@ if uploaded_file:
             embedding = get_image_embedding(image)
 
         with st.spinner("Searching for similar images..."):
-            matches = search_similar_images(embedding, top_k=10)
+            matches = search_similar_images(embedding, top_k=top_k)
 
         st.subheader("🔎 Top Similar Images")
-        for match in matches:
-            score = match.get("score", 0)
-            photo_id = match.get("id", "Unknown ID")
-            url = match.get("metadata", {}).get("url", None)
-
-            st.write(f"**Photo ID**: {photo_id} | **Similarity Score**: {score:.4f}")
-            if url:
-                st.image(url, caption=f"Photo ID: {photo_id}", use_container_width=True)
-            else:
-                st.warning(f"Image URL not found for Photo ID: {photo_id}")
-
-# Instructions
-st.write("---")
-st.write("Note: This app searches an Unsplash dataset indexed in Pinecone using CLIP embeddings for both text and images.")
+
+        if matches:
+            cols = st.columns(3)  # Arrange in 3 columns
+            for i, match in enumerate(matches):
+                score = match.get("score", 0)
+                photo_id = match.get("id", "Unknown ID")
+                url = match.get("metadata", {}).get("url", None)
+
+                with cols[i % 3]:  # Alternate images in columns
+                    st.write(f"��� **Photo ID**: {photo_id} | 🔢 **Score**: {score:.4f}")
+                    if url:
+                        st.image(url, caption=f"Photo ID: {photo_id}", use_container_width=True)
+                    else:
+                        st.warning(f"⚠️ Image URL not found for Photo ID: {photo_id}")
+
+            # 📥 **Download Option**
+            with st.expander("⬇️ Download Images"):
+                for match in matches:
+                    url = match.get("metadata", {}).get("url", None)
+                    if url:
+                        st.markdown(f"[Download Image]({url})", unsafe_allow_html=True)
+        else:
+            st.warning("⚠️ No similar images found!")
+
+# ✅ Add Footer
+st.markdown(
+    """
+    ---
+    💡 **About:** This app uses **CLIP & Pinecone** to perform **Text-to-Image & Image-to-Image Search**.  
+    Built with **Streamlit, Pinecone, and Hugging Face Transformers**.  
+    📩 Contact: [Your Email or GitHub](#)  
+    """
+)
+
+# ✅ Apply Dark Mode if selected
+if theme == "Dark Mode":
+    st.markdown(
+        """
+        <style>
+        body { background-color: #121212; color: white; }
+        .stTextInput, .stFileUploader, .stButton { background-color: #333; color: white; }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )