Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,16 +25,16 @@ if 'generated_image' not in st.session_state:
 st.subheader("Step 1: Context Analysis")
 msg = st.text_input("Describe your business here until i say to move forward:")
 if st.button("Submit"):
-    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.")
+    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)
@@ -44,18 +44,17 @@ st.markdown("---")
 # ---------------------- IDEATION ----------------------
 st.subheader("Step 2: Generate Ideas")
 if st.button("Generate Creative Ideas"):
-    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.")
+    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.")
 
-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]}")
@@ -63,20 +62,21 @@ if len(st.session_state.final_ideation)>0:
     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"):
-    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.")
+    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])
@@ -112,7 +112,6 @@ if uploaded_files:
 
 def call_brainstorming_api():
     print('Selected topics:', st.session_state.topics_to_pass)
-    st.spinner("Brainstorming. Please wait...")
     response = requests.post(
         f"{API_URL}/brainstrom",  # Replace with your actual endpoint
         params={"thread_id": "my-session"},
@@ -152,52 +151,54 @@ with col2:
         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()
+    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"):
-    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.")
+    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"):
-    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.")
+    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("---")
 
@@ -205,14 +206,14 @@ st.markdown("---")
 # ---------------------- FINAL STORY ----------------------
 st.subheader("📖 Step 5: Final Story")
 if st.button("Generate Final Story"):
-    st.spinner("Generating final story. Please wait...")
-    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)
+    with st.spinner("Generating final story. Please wait..."):
+        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("---")
 
 
@@ -223,14 +224,14 @@ if "generated_image" not in st.session_state:
     st.session_state.generated_image = None
 
 if st.button("Generate Image from Story"):
-    st.spinner("Generating image. Please wait...")
-    resp = requests.post(f"{API_URL}/generate-image")
-    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.")
+    with st.spinner("Generating image. Please wait..."):
+        resp = requests.post(f"{API_URL}/generate-image")
+        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 != "":