add comet on all other operations

src/streamlit_app.py

@@ -4,6 +4,12 @@
 
 import os
 
+# ---- Disable Comet auto-patching (MUST be set BEFORE importing openai/comet_llm/comet_ml) ----
+# Disable all Comet auto-logging / monkey-patching
+os.environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
+# Optionally: only disable LLM auto-logging
+os.environ["COMET_DISABLE_AUTO_LOGGING_LLM"] = "1"
+
 # ====== Force all cache dirs to /tmp (writable in most environments) ======
 CACHE_BASE = "/tmp/cache"
 os.environ["HF_HOME"] = f"{CACHE_BASE}/hf_home"
@@ -16,25 +22,21 @@ os.environ["STREAMLIT_STATIC_DIR"] = f"{CACHE_BASE}/streamlit_static"
 
 # Create the directories before imports
 for path in os.environ.values():
-    if path.startswith(CACHE_BASE):
+    if isinstance(path, str) and path.startswith(CACHE_BASE):
         os.makedirs(path, exist_ok=True)
 
-# ====== Imports ======
+# ====== Now safe to import libraries ======
 import streamlit as st
 import torch
 from sentence_transformers import SentenceTransformer, util
 from transformers import CLIPProcessor, CLIPModel
 from datasets import load_dataset, get_dataset_split_names
 from PIL import Image
+
 import openai
+from openai import OpenAI   # OK to import after openai is present
 from opik import track, log_event
 
-import os
-os.environ["COMET_DISABLE_AUTO_LOGGING"] = "1"   # disable all auto-logging
-# or to only disable LLM patching:
-os.environ["COMET_DISABLE_AUTO_LOGGING_LLM"] = "1"
-
-from openai import OpenAI
 
 # ========== 🔑 API Key ==========
 openai.api_key = os.getenv("OPENAI_API_KEY")