Update src/streamlit_app.py

src/streamlit_app.py

@@ -28,10 +28,13 @@ from datasets import load_dataset, get_dataset_split_names
 from PIL import Image
 import openai
 import comet_llm
-from functools import wraps
-from opik import track as opik_track
+from opik import track
 
+os.environ["STREAMLIT_CONFIG_DIR"] = "/tmp/.streamlit"
+os.environ["STREAMLIT_CACHE_DIR"] = f"{CACHE_BASE}/streamlit_cache"
+os.environ["STREAMLIT_STATIC_DIR"] = f"{CACHE_BASE}/streamlit_static"
 
+os.makedirs("/tmp/.streamlit", exist_ok=True)
 
 
 # ========== 🔑 API Key ==========
@@ -111,13 +114,7 @@ st.title("🩺 Multimodal Medical Chatbot")
 query = st.text_input("Enter your medical question or symptom description:")
 uploaded_file = st.file_uploader("Upload an image to find similar medical cases:", type=["png", "jpg", "jpeg"])
 
-def safe_track(func):
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        return opik_track(func)(*args, **kwargs)
-    return wrapper
-
-@safe_track
+@track
 def get_chat_completion_openai(client, prompt: str):
     return client.chat.completions.create(
         model="gpt-4o",  # or "gpt-4" if you need the older GPT-4
@@ -126,7 +123,7 @@ def get_chat_completion_openai(client, prompt: str):
         max_tokens=150
     )
 
-@safe_track
+@track
 def get_similar_prompt(query):
     text_embeddings = embed_dataset_texts(combined_texts)  # cached
     query_embedding = embed_query_text(query)  # recalculated each time
@@ -177,6 +174,8 @@ if query:
             st.warning("OpenAI API key not found. Please set OPENAI_API_KEY as a secret environment variable.")
 
 if uploaded_file is not None:
+    print('uploading file')
+    print(uploaded_file)
     query_image = Image.open(uploaded_file).convert("RGB")
     st.image(query_image, caption="Your uploaded image", use_container_width=True)
 
@@ -199,5 +198,7 @@ if uploaded_file is not None:
         similar_img = data[int(idx)]['image']
         st.image(similar_img, caption=f"Similarity: {score:.3f}", use_container_width=True)
         st.markdown(f"**Case description:** {data[int(idx)]['complaints']}")
+else:
+    print("no image")
 
 st.caption("This chatbot is for educational purposes only and does not provide medical advice.")