Update app.py

app.py

@@ -9,26 +9,22 @@ SAMPLE_FILE_URL = (
     "resolve/main/sample_keyword_data_cpc.csv"
 )
 
-# --- 1. Data Loading and Preprocessing ---
-class DataLoader:
-    def __init__(self, sample_file_url: str = SAMPLE_FILE_URL):
-        self.sample_file_url = sample_file_url
-
-    @st.cache_data
-    def load_csv(_self, uploaded_file_obj: st.runtime.uploaded_file_manager.UploadedFile | None) -> pd.DataFrame | None:
-        try:
-            if uploaded_file_obj:
-                df = pd.read_csv(uploaded_file_obj)
-            else:
-                df = pd.read_csv(_self.sample_file_url)
-        except Exception as e:
-            st.error(f"Error loading file: {e}")
-            return None
-        df.columns = [col.strip().lower() for col in df.columns]
-        if "cpc" not in df.columns:
-            st.warning("No `cpc` column found—simulating CPC values between 0.50–3.00 USD (for testing purposes only!)")
-            df["cpc"] = np.round(np.random.uniform(0.5, 3.0, size=len(df)), 2)
-        return df
+# --- ✅ CACHED CSV LOADER FUNCTION (moved out of class) ---
+@st.cache_data
+def load_csv(uploaded_file_obj: st.runtime.uploaded_file_manager.UploadedFile | None, sample_file_url: str) -> pd.DataFrame | None:
+    try:
+        if uploaded_file_obj:
+            df = pd.read_csv(uploaded_file_obj)
+        else:
+            df = pd.read_csv(sample_file_url)
+    except Exception as e:
+        st.error(f"Error loading file: {e}")
+        return None
+    df.columns = [col.strip().lower() for col in df.columns]
+    if "cpc" not in df.columns:
+        st.warning("No `cpc` column found—simulating CPC values between 0.50–3.00 USD (for testing purposes only!)")
+        df["cpc"] = np.round(np.random.uniform(0.5, 3.0, size=len(df)), 2)
+    return df
 
 # --- 2. SEO Forecasting Core ---
 class SeoCalculator:
@@ -109,8 +105,7 @@ class SeoCalculator:
 
 # --- 3. Streamlit App UI ---
 class SeoAppUI:
-    def __init__(self, data_loader: DataLoader, seo_calculator: SeoCalculator):
-        self.data_loader = data_loader
+    def __init__(self, seo_calculator: SeoCalculator):
         self.seo_calculator = seo_calculator
         self._set_page_config()
 
@@ -179,7 +174,7 @@ class SeoAppUI:
         st.download_button("📥 Download sample CSV", sample_bytes, file_name="sample_keyword_data_cpc.csv", mime="text/csv", key="download_sample_main")
         st.markdown("---")
         uploaded_file, target_position, conversion_rate, close_rate, mrr_per_customer, seo_cost, add_spend = self._get_sidebar_inputs()
-        df = self.data_loader.load_csv(uploaded_file)
+        df = load_csv(uploaded_file, SAMPLE_FILE_URL)
         if df is not None:
             metrics, df_results = self.seo_calculator.calculate_metrics(
                 df, target_position, conversion_rate, close_rate, mrr_per_customer, seo_cost, add_spend
@@ -191,7 +186,6 @@ class SeoAppUI:
 
 # --- App Runner ---
 if __name__ == "__main__":
-    data_loader = DataLoader()
     seo_calculator = SeoCalculator()
-    app_ui = SeoAppUI(data_loader, seo_calculator)
+    app_ui = SeoAppUI(seo_calculator)
     app_ui.run()