Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,6 +7,20 @@ import numpy as np
 import chromadb
 import tempfile
 
+# Create directory if not exists
+os.makedirs("demo_images", exist_ok=True)
+
+# Download 50 high-res (1024x768) random images
+for i in tqdm(range(50), desc="Downloading images"):
+    url = f"https://picsum.photos/seed/{i}/1024/768"
+    response = requests.get(url)
+    if response.status_code == 200:
+        with open(f"demo_images/img_{i+1:02}.jpg", "wb") as f:
+            f.write(response.content)
+    else:
+        print(f"Failed to download image {i+1}")
+
+
 # ----- Setup -----
 CACHE_DIR = tempfile.gettempdir()
 CHROMA_PATH = os.path.join(CACHE_DIR, "chroma_db")