Update app.py

app.py

@@ -67,10 +67,12 @@ if "file_processed" not in st.session_state:
     st.session_state.file_processed = False
 if "file_data" not in st.session_state:
     st.session_state.file_data = None
+if "last_uploaded_file" not in st.session_state:
+    st.session_state.last_uploaded_file = None
 
-# File processing function
+# File processing function with cache
 @st.cache_data
-def process_file(uploaded_file):
+def process_file(uploaded_file, _cache_key):
     if uploaded_file is None:
         return None
     
@@ -101,7 +103,7 @@ def process_file(uploaded_file):
         st.error(f"📄 Error processing file: {str(e)}")
         return None
 
-# Model loading function
+# Model loading function (cached, but only loaded once)
 @st.cache_resource
 def load_model(hf_token):
     if not TRANSFORMERS_AVAILABLE:
@@ -164,9 +166,18 @@ if "model" not in st.session_state:
 model = st.session_state.get("model")
 tokenizer = st.session_state.get("tokenizer")
 
+# Check for new file upload and clear cache
+if uploaded_file and uploaded_file != st.session_state.last_uploaded_file:
+    st.cache_data.clear()  # Clear all cached data
+    st.session_state.file_processed = False  # Reset processing state
+    st.session_state.file_data = None  # Clear previous file data
+    st.session_state.last_uploaded_file = uploaded_file  # Update last uploaded file
+
 # Process uploaded file once
 if uploaded_file and not st.session_state.file_processed:
-    file_data = process_file(uploaded_file)
+    # Use file name and size as a cache key to ensure uniqueness
+    cache_key = f"{uploaded_file.name}_{uploaded_file.size}"
+    file_data = process_file(uploaded_file, cache_key)
     if file_data:
         st.session_state.file_data = file_data
         st.session_state.file_processed = True