update(12)

src/app.py

@@ -193,9 +193,11 @@ if st.session_state.page == "Enrich Companies":
                             "approx_revenue": lead_revenue,
                             "business_type": lead_business_type,
                             "website_url": lead_web_url,
+                            "score": None
                         }
                         lead_data = {"companies": [lead_data]}
                         cleaned_data = clean_json.clean_json_f(lead_data)
+                        print(cleaned_data)
                         cleaned_data_obj = json.loads(cleaned_data)
                         cleaned_data_obj = add_leads_f(
                             main_lead_info_file, cleaned_data_obj

src/post_extraction_tools/clean_json.py

@@ -31,11 +31,13 @@ def clean_json_f(input_json: object) -> str:
                 "email": company.get("email", "").strip() if company.get("email") else None,
                 "approx_revenue": company.get("approx_revenue", "").strip() if company.get("approx_revenue") else None,
                 "business_type": company.get("business_type", "").strip(),
-                "website_url": company.get("website_url", "").strip()
+                "website_url": company.get("website_url", "").strip(),
+                "score": company.get("score", None)
             }
             cleaned_companies.append(cleaned_company)
         
         cleaned_data = {"companies": cleaned_companies}
+        print(cleaned_data)
         return json.dumps(cleaned_data, indent=2)
     
     except json.JSONDecodeError:

src/post_extraction_tools/lead_scoring.py

@@ -71,7 +71,7 @@ class LeadScoring:
     def score(self, leads: object, conditions: object):
         # scored_leads = []
         for lead in leads["companies"]:
-            if "score" not in lead:
+            if lead["score"] is None:
                 score = 0
                 if "industry_type" in conditions:
                     try: