merge code

src/streamlit_app.py

@@ -118,7 +118,10 @@ def get_chat_completion_openai(client, prompt: str):
     )
 
 @track
-def get_similar_prompt(query_embedding, text_embeddings):
+def get_similar_prompt(query):
+    text_embeddings = embed_dataset_texts(combined_texts)  # cached
+    query_embedding = embed_query_text(query)  # recalculated each time
+
     cos_scores = util.pytorch_cos_sim(query_embedding, text_embeddings)[0]
     top_result = torch.topk(cos_scores, k=1)
     idx = top_result.indices[0].item()
@@ -127,11 +130,10 @@ def get_similar_prompt(query_embedding, text_embeddings):
 
 if query:
     with st.spinner("Searching medical cases..."):
-        text_embeddings = embed_dataset_texts(combined_texts)  # cached
-        query_embedding = embed_query_text(query)  # recalculated each time
+
 
         # Compute similarity
-        selected = get_similar_prompt(query_embedding, text_embeddings)
+        selected = get_similar_prompt(combined_texts, query)
 
         # Show Image
         st.image(selected['image'], caption="Most relevant medical image", use_container_width=True)