Update pages/Content2Lead.py

pages/Content2Lead.py

@@ -1,35 +1,59 @@
 import streamlit as st
 import requests
+from dotenv import load_dotenv
 import os
+load_dotenv()
+endpoint=os.getenv('blog_lead_endpoint')
 def main():
     st.title("Content2Lead Matcher")
     email_options = [
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]",
-    "[email protected]"    ,
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
         "[email protected]"
-]    # Creating a form
+    ]
+
+    account_stages = [
+        "Pre-Sales",
+        "Pre-Sales (Unresponsive, After Call)",
+        "Pre-Sales (Long-Term/ Cold)",
+        "Sales Opportunity",
+        "Closed Lost (Opportunity)",
+        "Current Client",
+        "Pre-Sales (Short-Term/ Hot)",
+        "Pre-Sales (Mid-Term/ Warm)",
+        "Project Cancelled"
+    ]
+
+    # Creating a form
     with st.form(key='blog2lead_form'):
-        blog_url = st.text_input("Enter the Content URL", "")
+        blog_urls = st.text_area("Enter the Content URLs (one per line)", "")
         email_address = st.selectbox("Select your email address", email_options)
+        selected_stages = st.multiselect("Select Account Stages (optional)", account_stages)
         submit_button = st.form_submit_button(label='Submit')
 
     if submit_button:
-        if blog_url and email_address:
+        urls = blog_urls.strip().split('\n')
+        if urls and email_address:
             # Define your data payload to send
             data_to_send = {
-                "url": blog_url,
-                "email_receiver": email_address
+                "urls": urls,
+                "email_receiver": email_address,
             }
 
-            endpoint = os.getenv("blog_lead_endpoint")
+            # Add the filter to the payload only if selected_stages is not empty
+            if selected_stages:
+                data_to_send["filter"] = {
+                    "Account Stage": {"$in": selected_stages}
+                }
+
             # Sending the POST request to FastAPI
             response = requests.post(endpoint, json=data_to_send)