Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,22 +5,29 @@ from PIL import Image
 import os
 import numpy as np
 import chromadb
+import requests
 import tempfile
+from tqdm import tqdm
 
-# Create directory if not exists
-os.makedirs("demo_images", exist_ok=True)
+# Get a temporary directory (automatically cleaned up after runtime ends)
+temp_dir = tempfile.gettempdir()
+demo_dir = os.path.join(temp_dir, "demo_images")
+os.makedirs(demo_dir, exist_ok=True)
 
-# Download 50 high-res (1024x768) random images
+print(f"Saving images to: {demo_dir}")
+
+# Download 50 high-resolution images (1024x768)
 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:
+        with open(os.path.join(demo_dir, f"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")