Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,242 +1,209 @@
 import streamlit as st
 import requests
 import base64
-API_URL = "https://subashdvorak-brainstroming-fast-api.hf.space"  # Replace with your FastAPI URL
+from streamlit import switch_page
 
-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"):
-    with st.spinner("Analyzing your context...."):
-        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("---")
+API_URL = "http://127.0.0.1:8848"  # Replace with your FastAPI URL
 
-
-# ---------------------- IDEATION ----------------------
-st.subheader("Step 2: Generate Ideas")
-if st.button("Generate Creative Ideas"):
-    with st.spinner("Implementing the ideation. Please wait...."):
-        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.")
-
-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.')
-st.markdown("---")
-
-
-# ---------------------- 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"):
-    with st.spinner("Making changes. Please wait...."):
-        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
+st.set_page_config(page_title="AI Business Ideation Platform", layout="wide")
+st.header("Welcome to RT-GENAI")
+
+# Initialize session state
+session_defaults = {
+    'business_details': {},
+    'details_completed': False,
+    'final_ideation': [],
+    'human_interactions': [],
+    'brainstorm_response': {},
+    'final_story': "",
+    'generated_image': "",
+    'selected_topics': [],
+    'selected_from_brainstorm': [],
+    'story': "",
+    'brainstorming_topics': [],
+    'topics_to_pass': [],
+    'base64_images': [],
+    'history': [],
+    'history_index': -1,
+    'trigger_brainstorm': False
 }
-
-for key, val in defaults.items():
+for key, val in session_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)
+# Step 1: Context Analysis or Orchestration
+st.subheader("How can I help you?")
+msg = st.text_input("")
+
+
+uploaded_files = st.file_uploader("📂 Upload reference images (optional)", type=['png', 'jpg', 'jpeg'], accept_multiple_files=True, key="initial_image_uploader")
 
 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
+    print('Base 64 image:',base64_images)
 
-def call_brainstorming_api():
-    print('Selected topics:', st.session_state.topics_to_pass)
-    response = requests.post(
-        f"{API_URL}/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"):
-    with st.spinner("Brainstorming. Please wait..."):
-    
-        # 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"):
-    with st.spinner("Going 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"):
-    with st.spinner("Going 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"):
-    with st.spinner("Generating final story. Please wait..."):
-        resp = requests.post(f"{API_URL}/generate-final-story")
+if st.button("Submit"):
+    # if not st.session_state.details_completed:
+    if st.session_state.details_completed == True:
+        resp = requests.post(f"{API_URL}/context-analysis", json={"message": msg})
         if resp.ok:
-            st.session_state.final_story = resp.json()['response']
-            # st.markdown(st.session_state.final_story)
+            response = resp.json()
+            st.markdown(response.get('response'))
+            if response.get('complete'):
+                st.session_state.details_completed = True
+                st.session_state.business_details = response.get("business_details", {})
+                st.success("Context analysis completed. Now you can interact freely.")
         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 "generated_image" not in st.session_state:
-    st.session_state.generated_image = None
-
-if st.button("Generate Image from Story"):
-    with st.spinner("Generating image. Please wait..."):
-        resp = requests.post(f"{API_URL}/generate-image")
+            st.error("Context analysis failed.")
+    else:
+        resp = requests.post(f"{API_URL}/orchestration", json={"message": msg,"image_base64": st.session_state.base64_images})
         if resp.ok:
-            img_base64 = resp.json()['response']
-            st.session_state.generated_image=img_base64
-            st.success("Image generated successfully.")
-        else:
-            st.error("Failed to generate image.")
-
-# Display image if it exists
-if st.session_state.generated_image != "":
-    st.markdown(
-    f'<img src="data:image/png;base64,{st.session_state.generated_image}" width="500"/>',
-    unsafe_allow_html=True
-)
+            response = resp.json()
+            tool = response.get("tool_response")
+            message = response.get("message_response")
+            image_caption = response.get("image_caption")
+            st.write('Tool response:', tool)
+
+            # ========== Tool Handling ==========
+            if "ideation" in tool:
+                resp = requests.post(f"{API_URL}/ideation")
+                if resp.ok:
+                    st.session_state.final_ideation = resp.json()["response"]["improver_response"][-1]
+
+                    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 generated successfully.")
+
+            elif "human-idea-refining" in tool:
+                resp = requests.post(f"{API_URL}/human-idea-refining", json={"query": message})
+                if resp.ok:
+                    refined = resp.json()['response']
+                    st.session_state.human_interactions.append(refined)
+                    st.markdown(st.session_state.human_interactions[-1])
+                    st.success("Ideas refined.")
+
+            elif "generate-story" in tool:
+                switch_page("pages/all_in_one.py")
+                # def call_brainstorming_api():
+                #     response = requests.post(
+                #         f"{API_URL}/brainstorm",
+                #         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)
+
+                # call_brainstorming_api()
+
+                # uploaded_files = st.file_uploader("📂 Upload reference images (optional)", type=['png', 'jpg', 'jpeg'], accept_multiple_files=True, key="brainstorm_image_uploader")
+                # if uploaded_files:
+                #     base64_images = [base64.b64encode(f.read()).decode('utf-8') for f in uploaded_files]
+                #     st.session_state.base64_images = base64_images
+
+
+
+                # # Triggered Brainstorming
+                # if st.session_state.trigger_brainstorm:
+                #     st.session_state.trigger_brainstorm = False  # reset flag
+                #     # Save history
+                #     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]
+                #     st.session_state.history.append({
+                #         "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_index += 1
+
+                #     # Update selection
+                #     st.session_state.selected_topics.extend(st.session_state.selected_from_brainstorm)
+                #     st.session_state.selected_from_brainstorm = []
+
+                #     st.rerun()
+
+                # 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 'Brainstorm' to start.")
+
+                # if st.button("🚀 Brainstorm"):
+                #     st.session_state.trigger_brainstorm = True
+                #     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.")
+
+            elif "generate-ultimate-story" in tool:
+                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)
+                    st.success("Final story generated.")
+
+            elif "generate-image" in tool:
+                resp = requests.post(f"{API_URL}/generate-image")
+                if resp.ok:
+                    st.session_state.generated_image = resp.json()["response"]
+                    st.image(f"data:image/png;base64,{st.session_state.generated_image}", use_column_width=True)
+                    st.success("Image generated.")
 
+            else:
+                st.markdown(f"🤖 {message}")
+                if "no any information" not in image_caption.lower():
+                    st.markdown(f"🤖 {image_caption}")
+        else:
+            st.error("Orchestration failed.")