Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,222 @@
 import streamlit as st
+import requests
+import base64
+API_URL = "http://127.0.0.1:8848"  # Replace with your FastAPI URL
 
-st.set_page_config(page_title="Business Assistant", layout="wide")
-st.title("🧠 Business Storytelling Assistant")
-st.markdown("Use the sidebar to navigate through different steps.")
+st.set_page_config(page_title="AI Business Ideation Platform", layout="wide")
+st.header("Business Ideation & Story Generation Platform")
+
+# Store responses across user interactions
+if 'business_details' not in st.session_state:
+    st.session_state.business_details = {}
+if 'final_ideation' not in st.session_state:
+    st.session_state.final_ideation = []
+if 'human_interactions' not in st.session_state:
+    st.session_state.human_interactions = []
+if 'brainstorm_response' not in st.session_state:
+    st.session_state.brainstorm_response = {}
+if 'final_story' not in st.session_state:
+    st.session_state.final_story = ""
+if 'generated_image' not in st.session_state:
+    st.session_state.generated_image = ""
+
+# ---------------------- CONTEXT ANALYSIS ----------------------
+
+st.subheader("Step 1: Context Analysis")
+msg = st.text_input("Describe your business here until i say to move forward:")
+if st.button("Submit"):
+    resp = requests.post(f"{API_URL}/context-analysis", json={"message": msg})
+    if resp.ok:
+        response = resp.json()
+        st.markdown(response.get('response'))
+        if response.get('complete') == True:
+            st.session_state.business_details = response.get("business_details", {})
+            st.success('Context analysis completed. Scroll Down to step 2 now.')
+    else:
+        st.error("Failed to analyze context.")
+
+if st.session_state.business_details != {}:
+    st.json(st.session_state.business_details)
+st.markdown("---")
+
+
+# ---------------------- IDEATION ----------------------
+st.subheader("Step 2: Generate Ideas")
+if st.button("Generate Creative Ideas"):
+    resp = requests.post(f"{API_URL}/ideation")
+    if resp.ok:
+        result = resp.json()
+        st.session_state.final_ideation = result["response"]["improver_response"][-1]
+        # st.write("Final Ideas:")
+        # for idea in eval(st.session_state.final_ideation):
+        #     st.markdown(f"- {idea}")
+    else:
+        st.error("Failed to generate ideas.")
+
+st.markdown("---")
+if len(st.session_state.final_ideation)>0:
+    # st.subheader("Final Ideations")
+    st.markdown(f"*Idea:1* - {eval(st.session_state.final_ideation)[0]}")
+    st.markdown(f"*Idea:2* - {eval(st.session_state.final_ideation)[1]}")
+    st.markdown(f"*Idea:3* - {eval(st.session_state.final_ideation)[2]}")
+    st.markdown(f"*Idea:4* - {eval(st.session_state.final_ideation)[3]}")
+    st.success('Ideas are generated. Go down to refine your ideas.')
+
+
+# ---------------------- HUMAN REFINEMENT ----------------------
+st.subheader("Step 3: Human Feedback Loop")
+query = st.text_input("Suggest a change or improvement to the ideas:")
+if st.button("Refine with Feedback"):
+    resp = requests.post(f"{API_URL}/human-idea-refining", json={"query": query})
+    if resp.ok:
+        refined = resp.json()['response']
+        st.session_state.human_interactions.append(refined)
+        # st.markdown(refined)
+    else:
+        st.error("Failed to refine ideas.")
+
+if len(st.session_state.human_interactions)>0:
+    st.markdown(st.session_state.human_interactions[-1])
+
+st.markdown("---")
+
+
+# ---------------------- BRAINSTORM ----------------------
+st.subheader("Step 4: Story Boarding with brainstorming")
+
+defaults = {
+    "selected_topics": [],
+    "selected_from_brainstorm": [],
+    "story": "",
+    "brainstorming_topics": [],
+    "topics_to_pass":[],
+    "base64_images": [],
+    "history": [],
+    "history_index": -1,  # -1 means no history yet
+}
+
+for key, val in defaults.items():
+    if key not in st.session_state:
+        st.session_state[key] = val
+
+uploaded_files = st.file_uploader("📂 Upload reference images (optional)", type=['png', 'jpg', 'jpeg'], accept_multiple_files=True)
+
+if uploaded_files:
+    base64_images = []
+    for file in uploaded_files:
+        base64_images.append(base64.b64encode(file.read()).decode('utf-8'))
+    st.session_state.base64_images = base64_images
+
+def call_brainstorming_api():
+    print('Selected topics:', st.session_state.topics_to_pass)
+    response = requests.post(
+        "http://127.0.0.1:8848/brainstrom",  # Replace with your actual endpoint
+        params={"thread_id": "my-session"},
+        json={
+            "preferred_topics": st.session_state.topics_to_pass,
+            "image_base64_list": st.session_state.base64_images
+        }
+    )
+    if response.ok:
+        result_json = response.json()
+        data = result_json.get("response", {})
+        st.session_state.story = data.get("stories", [""])[-1]
+        st.session_state.brainstorming_topics = data.get("brainstroming_topics", [])
+    else:
+        st.error("❌ API call failed.")
+        st.write(response.text)
+
+col1, col2 = st.columns(2)
+
+with col1:
+    st.subheader("📖 Story")
+    st.text_area("Generated Story", st.session_state.story, height=300)
+
+with col2:
+    st.subheader("Brainstorming Topics (Click to Select)")
+    if st.session_state.brainstorming_topics:
+        topics_dict = st.session_state.brainstorming_topics[-1]
+        selected = set(st.session_state.selected_from_brainstorm)
+        for label, topic in topics_dict.items():
+            if st.checkbox(topic, key=topic, value=topic in selected):
+                selected.add(topic)
+            else:
+                selected.discard(topic)
+        st.session_state.topics_to_pass = list(selected)
+        st.session_state.selected_from_brainstorm = list(selected)
+    else:
+        st.info("No brainstorming topics yet. Click 'Generate Brainstorm' to start.")
+
+if st.button("🚀 Brainstorm"):
+    # Truncate future if we're not at the end
+    if st.session_state.history_index < len(st.session_state.history) - 1:
+        st.session_state.history = st.session_state.history[:st.session_state.history_index + 1]
+
+    # Save current state to history
+    current_state = {
+        "selected_topics": st.session_state.selected_topics.copy(),
+        "selected_from_brainstorm": st.session_state.selected_from_brainstorm.copy(),
+        "story": st.session_state.story,
+        "brainstorming_topics": st.session_state.brainstorming_topics.copy()
+    }
+    st.session_state.history.append(current_state)
+    st.session_state.history_index += 1
+
+    # Update current state
+    st.session_state.selected_topics.extend(st.session_state.selected_from_brainstorm)
+    st.session_state.selected_from_brainstorm = []
+
+    call_brainstorming_api()
+    st.rerun()
+
+if st.button("🔙 Back"):
+    if st.session_state.history_index > 0:
+        st.session_state.history_index -= 1
+        state = st.session_state.history[st.session_state.history_index]
+        st.session_state.selected_topics = state["selected_topics"]
+        st.session_state.selected_from_brainstorm = state["selected_from_brainstorm"]
+        st.session_state.story = state["story"]
+        st.session_state.brainstorming_topics = state["brainstorming_topics"]
+        st.rerun()
+    else:
+        st.warning("You're at the first step.")
+
+if st.button("🔜 Forward"):
+    if st.session_state.history_index < len(st.session_state.history) - 1:
+        st.session_state.history_index += 1
+        state = st.session_state.history[st.session_state.history_index]
+        st.session_state.selected_topics = state["selected_topics"]
+        st.session_state.selected_from_brainstorm = state["selected_from_brainstorm"]
+        st.session_state.story = state["story"]
+        st.session_state.brainstorming_topics = state["brainstorming_topics"]
+        st.rerun()
+    else:
+        st.warning("You're at the most recent step.")
+
+st.markdown("---")
+
+
+# ---------------------- FINAL STORY ----------------------
+st.subheader("📖 Step 5: Final Story")
+if st.button("Generate Final Story"):
+    resp = requests.post(f"{API_URL}/generate-final-story")
+    if resp.ok:
+        st.session_state.final_story = resp.json()['response']
+        # st.markdown(st.session_state.final_story)
+    else:
+        st.error("Failed to generate final story.")
+st.markdown(st.session_state.final_story)
+st.markdown("---")
+
+
+# ---------------------- GENERATE IMAGE ----------------------
+st.subheader("🖼️ Step 6: Generate Image")
+if st.button("Generate Image from Story"):
+    resp = requests.post(f"{API_URL}/generate-image")
+    if resp.ok:
+        img_url = resp.json()['response']
+        # st.image("image.png", caption="Generated Image from Story")
+        st.session_state.generated_image = "image.png"
+    else:
+        st.error("Failed to generate image.")
+st.image("image.png",caption="Generated Image from Story")