Update dataframe.py

dataframe.py

@@ -1,21 +1,14 @@
 import pandas as pd
 from loading_file import df
-import streamlit as st
 
-# Precompute nearest applications
-@st.cache_data
-def precompute_nearest_applications(df):
-    """Precomputes the nearest applications for each record in the DataFrame."""
-    df["Application Number"] = df["Application Number"].astype(int)
-    df = df.sort_values(by="Application Number").reset_index(drop=True)
-
-    # Add columns for nearest before and after applications
-    df["Nearest Before"] = df["Application Number"].shift(1)
-    df["Nearest After"] = df["Application Number"].shift(-1)
-    df["Nearest Before Decision"] = df["Decision"].shift(1)
-    df["Nearest After Decision"] = df["Decision"].shift(-1)
-
-    return df
+# Precompute for faster access during searches
+@st.cache_data(ttl=3600)
+def precompute_dataframe():
+    # Sort by application number for better nearest neighbor calculation
+    if df is not None:
+        df["Application Number"] = df["Application Number"].astype(int)
+        return df.sort_values("Application Number").reset_index(drop=True)
+    return None
 
 # Precomputed DataFrame
-precomputed_df = precompute_nearest_applications(df)
+precomputed_df = precompute_dataframe()