Create components/plan_generator.py

components/plan_generator.py

@@ -0,0 +1,37 @@
+# components/plan_generator.py
+
+import streamlit as st
+from langchain_groq import ChatGroq
+from langchain.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+import os
+from utils.prompts import STRATEGY_PROMPT
+from utils.helpers import generate_pdf
+
+def run_plan_generator():
+    st.header("📑 Business Plan & Pitch Generator")
+
+    if "bmc_answers" not in st.session_state:
+        st.warning("⚠️ Please complete the Canvas Assistant first.")
+        return
+
+    st.markdown("Generating plan using your Business Model Canvas answers...")
+
+    with st.spinner("Thinking like a startup advisor..."):
+        llm = ChatGroq(
+            temperature=0.3,
+            model_name="llama3-70b-8192",
+            groq_api_key=os.getenv("GROQ_API_KEY")
+        )
+
+        prompt = ChatPromptTemplate.from_template(STRATEGY_PROMPT)
+        chain = prompt | llm | StrOutputParser()
+
+        full_plan = chain.invoke({"input": st.session_state.bmc_answers})
+
+    st.subheader("📘 Business Plan")
+    st.write(full_plan)
+
+    if st.button("📤 Export as PDF"):
+        generate_pdf("business_plan", full_plan)
+        st.success("✅ PDF Exported Successfully!")