Update app.py

app.py

@@ -9,12 +9,15 @@ import plotly.express as px
 from plotly.subplots import make_subplots
 import plotly.graph_objects as go
 from datetime import date, timedelta
+from huggingface_hub import HfApi # ADDED: For permanent saving to Hugging Face
 
 # =====================================================
 # CONFIG / SECRETS
 # =====================================================
 API_TOKEN = os.getenv("LEADFEEDER_API_TOKEN")
 APP_PASSWORD = os.getenv("APP_PASSWORD")
+HF_TOKEN = os.getenv("HF_TOKEN") # ADDED: Your Hugging Face write token
+SPACE_ID = os.getenv("SPACE_ID") # ADDED: Automatically provided by HF Spaces
 
 if not API_TOKEN or not APP_PASSWORD:
     print("⚠️ WARNING: Secrets missing. App will launch but API calls will fail.")
@@ -64,15 +67,6 @@ def load_campaign_rules():
             return DEFAULT_CAMPAIGNS
     return DEFAULT_CAMPAIGNS
 
-def save_campaign_rules(rules_text):
-    try:
-        rules = json.loads(rules_text)
-        with open(CAMPAIGN_CONFIG_FILE, "w") as f:
-            json.dump(rules, f, indent=4)
-        return "✅ Configuration Saved!", rules
-    except json.JSONDecodeError:
-        return "❌ Error: Invalid JSON.", None
-
 def categorize_quality(score):
     if pd.isna(score): return "Unknown"
     try:
@@ -101,30 +95,23 @@ def apply_business_logic(df):
     if "lead_quality_score" in df.columns:
         df["Quality_Group"] = df["lead_quality_score"].apply(categorize_quality)
         
-    # 2. Campaign Logic (Hybrid Strict Match)
+    # 2. Campaign Logic (Smart Fallback)
     rules = load_campaign_rules()
     
     def resolve_campaign(row):
-        # Clean path to be safe
-        path = str(row.get("landing_page_path", "")).strip().lower()
-        
-        # SCENARIO 1: We HAVE a specific landing page path (e.g., /blog/..., /services/...)
-        # We strictly enforce keyword matching here to avoid generic pages getting grouped.
-        if path and path not in ["", "/", "/home", "nan", "none"]:
-            match = get_campaign_match(path, rules)
-            return match if match else "Uncategorized"
+        # Priority 1: Landing Page
+        if row.get("landing_page_path"):
+            match = get_campaign_match(row["landing_page_path"], rules)
+            if match: return match
+        # Priority 2: Website URL
+        if row.get("website_url"):
+            match = get_campaign_match(row["website_url"], rules)
+            if match: return match
+        # Priority 3: Industry
+        if row.get("primary_industry"):
+            match = get_campaign_match(row["primary_industry"], rules)
+            if match: return match
             
-        # SCENARIO 2: The path is missing (unenriched preview) OR it's a generic Homepage
-        # Here we fallback to website and industry so your dashboard actually populates!
-        website = str(row.get("website_url", "")).strip().lower()
-        industry = str(row.get("primary_industry", "")).strip().lower()
-        
-        match = get_campaign_match(website, rules)
-        if match: return match
-        
-        match = get_campaign_match(industry, rules)
-        if match: return match
-        
         return "Uncategorized"
 
     df["Campaign"] = df.apply(resolve_campaign, axis=1)
@@ -491,7 +478,6 @@ with gr.Blocks(title="Leadfeeder Campaign Pro") as demo:
         # --- TAB 4: SETTINGS ---
         with gr.Tab("⚙️ Campaign Settings"):
             gr.Markdown("### Manage Campaign Groups")
-            gr.Markdown("Select a campaign below and paste your keywords separated by commas.")
             
             init_rules = load_campaign_rules()
             camp_choices = list(init_rules.keys()) + ["+ Create New Campaign"]
@@ -513,7 +499,7 @@ with gr.Blocks(title="Leadfeeder Campaign Pro") as demo:
                     
             camp_dropdown.change(update_ui_on_select, inputs=[camp_dropdown], outputs=[new_camp_name, kw_input])
             
-            save_config_btn = gr.Button("💾 Save Configuration", variant="primary")
+            save_config_btn = gr.Button("💾 Save Configuration to Hugging Face", variant="primary")
             config_status = gr.Markdown()
             
             def save_easy_config(selected_camp, new_name, kw_string):
@@ -527,12 +513,33 @@ with gr.Blocks(title="Leadfeeder Campaign Pro") as demo:
                 if not target_camp:
                     return "❌ Error: Campaign name cannot be empty.", gr.update()
                 
+                # 1. Update the dictionary
                 rules[target_camp] = clean_kws
+                
+                # 2. Save file locally first
                 with open(CAMPAIGN_CONFIG_FILE, "w") as f:
                     json.dump(rules, f, indent=4)
                     
+                status_message = f"✅ Saved locally! Updated keywords for '{target_camp}'."
+                
+                # 3. Push to Hugging Face automatically
+                if HF_TOKEN and SPACE_ID:
+                    try:
+                        api = HfApi(token=HF_TOKEN)
+                        api.upload_file(
+                            path_or_fileobj=CAMPAIGN_CONFIG_FILE,
+                            path_in_repo=CAMPAIGN_CONFIG_FILE,
+                            repo_id=SPACE_ID,
+                            repo_type="space"
+                        )
+                        status_message = f"✅ Saved securely to Hugging Face Cloud! Updated keywords for '{target_camp}'."
+                    except Exception as e:
+                        status_message = f"⚠️ Saved locally, but failed to push to Hugging Face (Check HF_TOKEN). Error: {e}"
+                elif not HF_TOKEN:
+                    status_message = f"⚠️ Saved locally. To make this permanent on Hugging Face, add an HF_TOKEN secret in your Space settings."
+                
                 updated_choices = list(rules.keys()) + ["+ Create New Campaign"]
-                return f"✅ Saved successfully! Updated keywords for '{target_camp}'.", gr.update(choices=updated_choices, value=target_camp)
+                return status_message, gr.update(choices=updated_choices, value=target_camp)
 
             save_config_btn.click(save_easy_config, inputs=[camp_dropdown, new_camp_name, kw_input], outputs=[config_status, camp_dropdown])