Create plan_generator.py

components/plan_generator.py

@@ -1,29 +1,41 @@
+# components/plan_generator.py
+
 import streamlit as st
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.runnables import Runnable
 from langchain_groq import ChatGroq
-from langchain.prompts import ChatPromptTemplate
-from utils.prompts import STRATEGY_PROMPT
 from utils.session import get_canvas_data
+from utils.prompts import STRATEGY_PROMPT
 from utils.helpers import generate_pdf
 
+
 def run_plan_generator():
     st.header("📑 Auto Plan Generator")
 
-    canvas_data = get_canvas_data()
-    if not canvas_data or any(not v.strip() for v in canvas_data.values()):
+    canvas = get_canvas_data()
+    if not canvas:
         st.warning("Please complete the Canvas Assistant first.")
         return
 
-    if st.button("Generate Plan"):
-        with st.spinner("Generating business plan..."):
-            prompt = ChatPromptTemplate.from_template(STRATEGY_PROMPT)
-            chain = prompt | ChatGroq(model="llama3-8b-8192", temperature=0.4)
-            full_canvas = "\n".join([f"{k}: {v}" for k, v in canvas_data.items()])
-            result = chain.invoke({"input": full_canvas})
+    # Display Canvas Summary
+    st.subheader("🧾 Your Canvas Summary")
+    for section, content in canvas.items():
+        st.markdown(f"**{section}**")
+        st.info(content)
+
+    with st.spinner("Generating business plan..."):
+        prompt = ChatPromptTemplate.from_template(
+            STRATEGY_PROMPT + "\n\nCanvas Data:\n{input}"
+        )
+        chain: Runnable = prompt | ChatGroq(model="llama3-8b-8192", temperature=0.4)
+        full_canvas_text = "\n".join([f"{k}: {v}" for k, v in canvas.items()])
+        result = chain.invoke({"input": full_canvas_text})
 
-            st.session_state.generated_plan = result.content
-            st.subheader("Your Business Plan")
-            st.markdown(result.content)
+        st.subheader("📄 Generated Business Plan")
+        st.success(result.content)
 
-            pdf_path = generate_pdf("business_plan", result.content)
-            with open(pdf_path, "rb") as f:
-                st.download_button("Download PDF", f, file_name="business_plan.pdf", mime="application/pdf")
+        # Allow download
+        pdf_path = generate_pdf(result.content , content)
+        with open(pdf_path, "rb") as f:
+            st.download_button("📥 Download as PDF", f, file_name="Business_Plan.pdf")
+