Update app_main.py

app_main.py

@@ -238,6 +238,9 @@ def extract_images_from_pdf(pdf_path, output_json_path):
     except Exception as e:
         raise RuntimeError(f"❌ Error in extract_images_from_pdf: {str(e)}")
 
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+os.environ["HF_HOME"] = "/app/cache"
+
 def similarity_matching(input_json_path: str) -> str:
     import uuid, shutil, tempfile
     from langchain_experimental.open_clip.open_clip import OpenCLIPEmbeddings
@@ -318,6 +321,8 @@ def similarity_matching(input_json_path: str) -> str:
     # ============================== #
     #     COMPUTE SIMILARITIES       #
     # ============================== #
+    if not sprite_features or not img_features:
+        raise ValueError("❌ Embedding generation failed: One of the image feature lists is empty.")
     similarity = np.matmul(np.array(sprite_features), np.array(img_features).T)
     most_similar_indices = np.argmax(similarity, axis=1)