Create components/canvas_chat.py

components/canvas_chat.py

@@ -1,47 +1,33 @@
-import os
 import streamlit as st
-from langchain.prompts import ChatPromptTemplate
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.runnables import RunnableLambda, RunnablePassthrough
 from langchain_core.output_parsers import StrOutputParser
 from langchain_groq import ChatGroq
-from utils.prompts import CANVAS_PROMPT
-from utils.session import load_json_state, save_json_state
-
+from utils.prompts import STRATEGY_PROMPT
+import time
 
 def run_canvas_chat():
-    st.header("🧠 Canvas Assistant")
+    st.title("🧠 Canvas Assistant")
 
-    canvas_data = load_json_state("canvas_data.json", default={})
+    if "canvas_data" not in st.session_state:
+        st.session_state.canvas_data = {}
 
-    if "canvas_stage" not in st.session_state:
-        st.session_state.canvas_stage = "ask_idea"
+    with st.form("canvas_form"):
+        st.header("Complete the Business Model Canvas")
 
-    if st.session_state.canvas_stage == "ask_idea":
-        user_idea = st.text_input("What's your business idea?")
-        if st.button("Submit Idea"):
-            if user_idea.strip() != "":
-                st.session_state.canvas_stage = "generating_canvas"
-                st.session_state.business_idea = user_idea.strip()
-                save_json_state("canvas_data.json", {"business_idea": user_idea.strip()})
-                st.rerun()
+        st.session_state.canvas_data["customer_segments"] = st.text_area("Customer Segments")
+        st.session_state.canvas_data["value_propositions"] = st.text_area("Value Propositions")
+        st.session_state.canvas_data["channels"] = st.text_area("Channels")
+        st.session_state.canvas_data["customer_relationships"] = st.text_area("Customer Relationships")
+        st.session_state.canvas_data["revenue_streams"] = st.text_area("Revenue Streams")
+        st.session_state.canvas_data["key_resources"] = st.text_area("Key Resources")
+        st.session_state.canvas_data["key_activities"] = st.text_area("Key Activities")
+        st.session_state.canvas_data["key_partnerships"] = st.text_area("Key Partnerships")
+        st.session_state.canvas_data["cost_structure"] = st.text_area("Cost Structure")
 
-    elif st.session_state.canvas_stage == "generating_canvas":
-        with st.spinner("Generating canvas based on your idea..."):
-            prompt = ChatPromptTemplate.from_messages([
-                ("system", CANVAS_PROMPT),
-                ("user", f"My business idea is: {st.session_state.business_idea}")
-            ])
-            chain = prompt | ChatGroq(model_name="llama3-8b-8192", temperature=0.4)
-            response = chain.invoke({"input": st.session_state.business_idea})
-            canvas_output = response.content.strip()
-            canvas_data["canvas_output"] = canvas_output
-            save_json_state("canvas_data.json", canvas_data)
-            st.session_state.canvas_stage = "display_canvas"
-            st.rerun()
+        submitted = st.form_submit_button("Submit Canvas")
 
-    elif st.session_state.canvas_stage == "display_canvas":
-        st.subheader("📄 Business Model Canvas")
-        canvas_output = canvas_data.get("canvas_output", "No canvas generated.")
-        st.markdown(canvas_output)
-        if st.button("Edit or Regenerate"):
-            st.session_state.canvas_stage = "ask_idea"
-            st.rerun()
+    if submitted:
+        st.success("✅ Canvas information saved!")
+        time.sleep(1)
+        st.rerun()