deploy

README.md

@@ -0,0 +1,10 @@
+---
+title: Image Search Engine Streamlit
+emoji: 📈
+colorFrom: red
+colorTo: indigo
+sdk: streamlit
+sdk_version: 1.42.2
+app_file: app.py
+pinned: false
+---

app.py

@@ -0,0 +1,66 @@
+import streamlit as st
+import requests
+from PIL import Image
+
+# FastAPI Backend URL
+API_URL = "http://127.0.0.1:8000/search"
+
+st.title("🔍 Image Search Engine")
+
+# Choose Search Type (Text or Image)
+option = st.radio("Search by:", ("Text", "Image"))
+
+# 📝 Text-based Search
+if option == "Text":
+    query = st.text_input("Enter search query:")
+
+    if st.button("Search") and query:
+        try:
+            response = requests.get(f"{API_URL}/text", params={"query": query})
+            response.raise_for_status()
+            
+            results = response.json()
+
+            if "matches" in results and results["matches"]:
+                st.subheader("🔎 Similar Images")
+                for match in results["matches"]:
+                    image_url = match.get("url", "")
+                    if image_url:
+                        st.image(image_url, width=300)
+                    else:
+                        st.warning("⚠️ No image URL found.")
+            else:
+                st.warning("⚠️ No similar images found.")
+                
+        except requests.exceptions.RequestException as e:
+            st.error(f"❌ Error: {e}")
+
+# 📷 Image-based Search
+elif option == "Image":
+    uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
+
+    if uploaded_file:
+        image = Image.open(uploaded_file)
+        st.image(image, caption="📷 Uploaded Image", width=300)
+
+        if st.button("Search"):
+            try:
+                files = {"file": (uploaded_file.name, uploaded_file.getvalue(), uploaded_file.type)}
+                response = requests.post(f"{API_URL}/image", files=files)
+                response.raise_for_status()
+                
+                results = response.json()
+
+                if "matches" in results and results["matches"]:
+                    st.subheader("🔎 Similar Images")
+                    for match in results["matches"]:
+                        image_url = match.get("url", "")
+                        if image_url:
+                            st.image(image_url, width=300)
+                        else:
+                            st.warning("⚠️ No image URL found.")
+                else:
+                    st.warning("⚠️ No similar images found.")
+                    
+            except requests.exceptions.RequestException as e:
+                st.error(f"❌ Error: {e}")

requirements.txt

@@ -0,0 +1,3 @@
+streamlit>=1.20.0
+requests>=2.28.0
+pillow>=9.0.0

utils/api.py

@@ -0,0 +1,21 @@
+import requests
+from typing import Dict, Any
+
+API_URL = "http://127.0.0.1:8000/search"
+
+def search_by_text(query: str) -> Dict[str, Any]:
+    try:
+        response = requests.get(f"{API_URL}/text", params={"query": query})
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        return {"error": str(e)}
+
+def search_by_image(image_file) -> Dict[str, Any]:
+    try:
+        files = {"file": (image_file.name, image_file.getvalue(), image_file.type)}
+        response = requests.post(f"{API_URL}/image", files=files)
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        return {"error": str(e)}