Create app.py

app.py

@@ -0,0 +1,145 @@
+import os
+import sqlite3
+import streamlit as st
+from typing import List, TypedDict
+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
+
+# --- 1. PAGE SETUP & CUSTOM CSS ---
+st.set_page_config(page_title="Strategic Intel-Agent", layout="wide", page_icon="🛡️")
+
+st.markdown("""
+    <style>
+    .main { background-color: #f8f9fa; }
+    .stButton>button { width: 100%; border-radius: 8px; height: 3em; font-weight: bold; transition: 0.3s; }
+    .stDownloadButton>button { background-color: #007bff; color: white; border-radius: 8px; }
+    .report-container { background-color: white; padding: 25px; border-radius: 12px; border: 1px solid #dee2e6; box-shadow: 0 4px 6px rgba(0,0,0,0.05); }
+    .tool-card { padding: 15px; border-radius: 10px; border-left: 5px solid #007bff; background: #ffffff; margin-bottom: 10px; }
+    </style>
+    """, unsafe_allow_html=True)
+
+# --- 2. SIDEBAR (CONFIG) ---
+with st.sidebar:
+    st.image("https://cdn-icons-png.flaticon.com/512/2103/2103633.png", width=80)
+    st.title("Settings")
+    
+    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")
+
+    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
+    
+    st.divider()
+    st.info("The agent will research and allow you to critique the data before generating a final report.")
+
+# --- 3. AGENT LOGIC ---
+DB_PATH = "/tmp/agent_memory.db"
+
+class AgentState(TypedDict):
+    company_name: str
+    research_data: List[dict]
+    iteration: int
+    report: str
+
+@st.cache_resource
+def initialize_agent():
+    llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+    search_tool = TavilySearchResults(max_results=3)
+
+    def researcher(state: AgentState):
+        iter_count = state.get("iteration", 0)
+        query = f"Financial SWOT and strategy for {state['company_name']} 2025"
+        if iter_count > 0:
+            query = f"Additional deep-dive business analysis for {state['company_name']} 2025"
+        results = search_tool.invoke({"query": query})
+        return {"research_data": results, "iteration": iter_count + 1}
+
+    def writer(state: AgentState):
+        context = "\n".join([f"Source: {r['url']}\nContent: {r['content']}" for r in state['research_data']])
+        prompt = f"Write a professional SWOT for {state['company_name']}. Cite [1][2].\n\nDATA:\n{context}"
+        response = llm.invoke(prompt)
+        return {"report": response.content}
+
+    builder = StateGraph(AgentState)
+    builder.add_node("researcher", researcher)
+    builder.add_node("writer", writer)
+    builder.add_edge(START, "researcher")
+    builder.add_edge("researcher", "writer")
+    builder.add_edge("writer", END)
+    
+    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    return builder.compile(checkpointer=SqliteSaver(conn), interrupt_before=["writer"])
+
+# --- 4. UI STATE MANAGEMENT ---
+if "report_output" not in st.session_state:
+    st.session_state.report_output = None
+
+agent_app = initialize_agent()
+config = {"configurable": {"thread_id": thread_id}}
+
+# --- 5. MAIN GUI ---
+st.title("🛡️ Strategic Intel-Agent")
+st.caption("Agentic Intelligence: Research & SWOT Generation")
+
+# 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")
+    with col_btn:
+        if st.button("🔍 Start Research"):
+            st.session_state.report_output = None
+            with st.status("Initializing Agent...", expanded=True) as status:
+                agent_app.invoke({"company_name": company, "iteration": 0}, config)
+                status.update(label="Research Phase Complete!", state="complete", expanded=False)
+            st.rerun()
+
+current_state = agent_app.get_state(config)
+
+if current_state.values.get("research_data"):
+    st.divider()
+    
+    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))
+
+    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()
+
+    c1, c2 = st.columns(2)
+    with c1:
+        if st.button("✅ Approve & Generate SWOT", type="primary"):
+            with st.spinner("Synthesizing Report..."):
+                final_res = agent_app.invoke(None, config)
+                st.session_state.report_output = final_res.get("report")
+                st.rerun()
+    with c2:
+        if st.button("🔄 Data Insufficient: Search Again"):
+            agent_app.update_state(config, {"research_data": []}, as_node="researcher")
+            with st.spinner("Refining search..."):
+                agent_app.invoke(None, config)
+            st.rerun()
+
+# --- 6. REPORT SECTION ---
+if st.session_state.report_output:
+    st.markdown("---")
+    st.markdown("### 🏁 Final SWOT Report")
+    st.markdown(f'<div class="report-container">{st.session_state.report_output}</div>', unsafe_allow_html=True)
+    
+    st.divider()
+    st.download_button(
+        label="📄 Download Report (.md)",
+        data=st.session_state.report_output,
+        file_name=f"{company}_SWOT.md",
+        mime="text/markdown"
+    )