Update app.py

app.py

@@ -1,17 +1,17 @@
 import os
 import sqlite3
+import smtplib
 import streamlit as st
 from typing import List, TypedDict
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+# LangChain & Graph Imports
 from langchain_openai import ChatOpenAI
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langgraph.graph import StateGraph, START, END
 from langgraph.checkpoint.sqlite import SqliteSaver
 
-# --- NEW TOOL IMPORTS ---
-import requests  # For Slack Webhooks
-from sendgrid import SendGridAPIClient
-from sendgrid.helpers.mail import Mail
-
 # --- 1. PAGE SETUP & CUSTOM CSS ---
 st.set_page_config(page_title="Strategic Intel-Agent", layout="wide", page_icon="🛡️")
 
@@ -33,40 +33,36 @@ with st.sidebar:
     with st.expander("🔑 API Credentials", expanded=True):
         openai_key = st.text_input("OpenAI API Key", value=os.getenv("OPENAI_API_KEY", ""), type="password")
         tavily_key = st.text_input("Tavily API Key", value=os.getenv("TAVILY_API_KEY", ""), type="password")
-        sg_key = st.text_input("SendGrid API Key (Optional)", type="password")
-        slack_webhook = st.text_input("Slack Webhook URL (Optional)", type="password")
+        
+    with st.expander("📧 Email Settings (SMTP)", expanded=False):
+        sender_email = st.text_input("Your Gmail Address")
+        app_password = st.text_input("App Password", type="password", help="Use a Google App Password, not your login password.")
 
     thread_id = st.text_input("Session ID", value="global_user_1")
     
     os.environ["OPENAI_API_KEY"] = openai_key
     os.environ["TAVILY_API_KEY"] = tavily_key
-    os.environ["SENDGRID_API_KEY"] = sg_key
     
     st.divider()
-    st.info("The agent will research, allow you to critique the data, and then distribute findings via the tools below.")
-
-# --- 3. UTILITY TOOLS ---
-def send_to_slack(text, webhook_url):
-    if not webhook_url: return False
-    payload = {"text": f"🚀 *New Market Intel:* \n {text[:500]}..."}
-    r = requests.post(webhook_url, json=payload)
-    return r.status_code == 200
-
-def send_to_email(content, recipient, api_key):
-    if not api_key or not recipient: return False
-    message = Mail(
-        from_email='reports@intel-agent.ai',
-        to_emails=recipient,
-        subject='Market Intelligence Report',
-        plain_text_content=content)
+    st.info("The agent will research, allow you to critique the data, and then distribute findings via email.")
+
+# --- 3. UTILITY TOOLS (NEW EMAIL TOOL) ---
+def send_email_smtp(content, recipient, sender, password):
+    if not all([content, recipient, sender, password]): return False
+    msg = MIMEMultipart()
+    msg['From'], msg['To'], msg['Subject'] = sender, recipient, "Market Intelligence Report"
+    msg.attach(MIMEText(content, 'plain'))
     try:
-        sg = SendGridAPIClient(api_key)
-        sg.send(message)
+        server = smtplib.SMTP('smtp.gmail.com', 587)
+        server.starttls()
+        server.login(sender, password)
+        server.sendmail(sender, recipient, msg.as_string())
+        server.quit()
         return True
     except Exception: return False
 
-# --- 4. AGENT LOGIC (Unchanged as requested) ---
-DB_PATH = "/tmp/agent_memory.db"
+# --- 4. AGENT LOGIC ---
+DB_PATH = "agent_memory.db"
 
 class AgentState(TypedDict):
     company_name: str
@@ -77,9 +73,13 @@ class AgentState(TypedDict):
 @st.cache_resource
 def initialize_agent():
     llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
-    search_tool = TavilySearchResults(max_results=3)
 
     def researcher(state: AgentState):
+        # Initializing tool inside node to prevent startup validation errors
+        if not os.environ.get("TAVILY_API_KEY"):
+            return {"research_data": [], "iteration": state.get("iteration", 0) + 1}
+        
+        search_tool = TavilySearchResults(max_results=3)
         iter_count = state.get("iteration", 0)
         query = f"Financial SWOT and strategy for {state['company_name']} 2025"
         if iter_count > 0:
@@ -112,13 +112,12 @@ config = {"configurable": {"thread_id": thread_id}}
 
 # --- 6. MAIN GUI ---
 st.title("🛡️ Strategic Intel-Agent")
-st.caption("v2.0: Agentic Intelligence with Slack & Email Integration")
+st.caption("v2.1: Agentic Intelligence with SMTP Email Integration")
 
-# Input Section
 with st.container():
     col_input, col_btn = st.columns([4, 1])
     with col_input:
-        company = st.text_input("Target Company Name", placeholder="e.g. NVIDIA, Tesla, Microsoft", label_visibility="collapsed")
+        company = st.text_input("Target Company Name", placeholder="e.g. NVIDIA, Tesla", label_visibility="collapsed")
     with col_btn:
         if st.button("🔍 Start Research"):
             st.session_state.report_output = None
@@ -131,22 +130,18 @@ current_state = agent_app.get_state(config)
 
 if current_state.values.get("research_data"):
     st.divider()
-    
-    # Header with Iteration Metric
     m1, m2 = st.columns([3, 1])
     with m1:
         st.subheader("📋 Intelligence Briefing")
     with m2:
         st.metric("Web Search Cycles", current_state.values.get("iteration", 0))
 
-    # Sources
     with st.expander("📂 View Raw Research Data", expanded=True):
         for item in current_state.values["research_data"]:
             st.markdown(f"**Source:** {item['url']}")
             st.caption(item['content'])
             st.divider()
 
-    # Action Buttons
     c1, c2 = st.columns(2)
     with c1:
         if st.button("✅ Approve & Generate SWOT", type="primary"):
@@ -161,7 +156,7 @@ if current_state.values.get("research_data"):
                 agent_app.invoke(None, config)
             st.rerun()
 
-# --- 7. REPORT & TOOL DISTRIBUTION CENTER ---
+# --- 7. REPORT & EMAIL CENTER ---
 if st.session_state.report_output:
     st.markdown("---")
     res_col, tool_col = st.columns([2, 1])
@@ -178,22 +173,11 @@ if st.session_state.report_output:
 
     with tool_col:
         st.markdown("### 🚀 Distribute Results")
-        
-        # Slack Tool
-        with st.container(border=True):
-            st.markdown("**Slack Channel**")
-            if st.button("📤 Post to Slack"):
-                if send_to_slack(st.session_state.report_output, slack_webhook):
-                    st.success("Sent to Slack!")
-                else:
-                    st.error("Slack Webhook missing/invalid")
-
-        # Email Tool
         with st.container(border=True):
             st.markdown("**Email Delivery**")
             email_target = st.text_input("Recipient Email", placeholder="manager@company.com")
             if st.button("📧 Send via Email"):
-                if send_to_email(st.session_state.report_output, email_target, sg_key):
+                if send_email_smtp(st.session_state.report_output, email_target, sender_email, app_password):
                     st.success(f"Email sent to {email_target}")
                 else:
-                    st.error("Email failed. Check API Key.")
+                    st.error("Check SMTP settings & App Password.")