Update document_processor_hf.py

document_processor_hf.py

@@ -5,9 +5,9 @@ import tempfile
 from typing import List, Dict, Any, Tuple, Optional
 import numpy as np
 from sentence_transformers import SentenceTransformer
-import streamlit as st
 from dataclasses import dataclass
 from collections import defaultdict
+import functools
 
 logger = logging.getLogger(__name__)
 
@@ -31,6 +31,20 @@ class Entity:
     start_pos: int
     end_pos: int
 
+# Simple caching decorator to replace Streamlit's cache
+def simple_cache(func):
+    """Simple caching decorator"""
+    cache = {}
+    
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        # Create a simple key from args (excluding self)
+        key = str(args[1:]) + str(sorted(kwargs.items()))
+        if key not in cache:
+            cache[key] = func(*args, **kwargs)
+        return cache[key]
+    return wrapper
+
 class DocumentProcessor:
     """
     Streamlined document processor for Hugging Face Spaces deployment.
@@ -78,9 +92,9 @@ class DocumentProcessor:
             'academic': ['abstract', 'introduction', 'literature review', 'methodology', 'results']
         }
     
-    @st.cache_resource
-    def _load_embedding_model(_self):
-        """Load embedding model with Streamlit caching"""
+    @simple_cache
+    def _load_embedding_model(self):
+        """Load embedding model with simple caching"""
         try:
             logger.info("🔄 Loading embedding model...")
             model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')