Update app.py

app.py

@@ -397,121 +397,134 @@ st.set_page_config(layout="wide")
 st.markdown("<h1 style='text-align: center;'>Clarifai Story Teller</h1>", unsafe_allow_html=True)
 st.markdown("<h2 style='text-align: center;'>Branching Reading Adventure</h2>", unsafe_allow_html=True)
 
+tabs = ["Create Your Story Script", "Build Your Image-Audio Book", "Interact with Your Characters"]
+
+# Initialize the current tab in session state
+if "current_tab" not in st.session_state:
+    st.session_state.current_tab = tabs[0]
+
+# Function to switch tabs
+def switch_tab(tab_name):
+    st.session_state.current_tab = tab_name
 
 # Create tabs
-tab1, tab2, tab3 = st.tabs(["Create Your Story Script", "Build Your Image-Audio Book", "Interact with Your Characters"])
+tab1, tab2, tab3 = st.tabs(tabs)
 
 # Tab 1: Introduction
-with tab1:
-    # Set up the Streamlit interface
-    col1, col2, col3 = st.columns([1, 4, 1])
-
-    with col1:
-        st.image('crewai/resources/whale.jpg')     
-
-    with col2:
-        # Input for the user
-        input_topic = st.text_area("What Exciting Adventures Awaits Us", height=100, placeholder="Start Our Story...")
-        st.session_state['on_topic'] = input_topic
-        # Button to run the process
-        if st.button("Create a Story"):
-            # Run the crewai process
-            with st.spinner('Generating Content...'):
-                result = crewai_process(input_topic)
-                # Display the result
-                st.session_state['text_block']  = result
-                st.text_area("Output", value=result , height=300)
-
-    with col3:
-        st.image('crewai/resources/clarifai.png')
+if st.session_state.current_tab == tabs[0]:
+    with tab1:
+        # Set up the Streamlit interface
+        col1, col2, col3 = st.columns([1, 4, 1])
+    
+        with col1:
+            st.image('crewai/resources/whale.jpg')     
+    
+        with col2:
+            # Input for the user
+            input_topic = st.text_area("What Exciting Adventures Awaits Us", height=100, placeholder="Start Our Story...")
+            st.session_state['on_topic'] = input_topic
+            # Button to run the process
+            if st.button("Create a Story"):
+                # Run the crewai process
+                with st.spinner('Generating Content...'):
+                    result = crewai_process(input_topic)
+                    # Display the result
+                    st.session_state['text_block']  = result
+                    st.text_area("Output", value=result , height=300)
+    
+        with col3:
+            st.image('crewai/resources/clarifai.png')
 
 # Tab 2: Data Visualization
-with tab2:
+elif st.session_state.current_tab == tabs[1]:
+    with tab2:
+    
+        
+        # Streamlit main page
+        
+        if st.button("Generate Images and Audio"):
+            
+            sentence_chunks = split_text_into_sentences_and_chunks(st.session_state.text_block , 8)
+            prompts = [' '.join(chunk) for chunk in sentence_chunks]
+            cols = st.columns(4)
+            with st.spinner('Generating Content...'):
+                for i, prompt in enumerate(prompts):
+                    image_path, img_error = generate_image(prompt+st.session_state.on_topic )
+                    audio, audio_error = generate_audio(prompt)
+        
+                    with cols[i % 4]:
+                        if img_error:
+                            st.error(img_error)
+                        else:
+                            st.session_state['image_paths'].append(image_path)
+                            st.image(image_path, prompt, use_column_width=True)
+        
+                        if audio_error:
+                            st.error(audio_error)
+                        else:
+                            st.audio(audio, format='audio/wav')
 
+# Tab 3: User Input and Results
+elif st.session_state.current_tab == tabs[2]:
+    with tab3:
     
-    # Streamlit main page
     
-    if st.button("Generate Images and Audio"):
-        
-        sentence_chunks = split_text_into_sentences_and_chunks(st.session_state.text_block , 8)
-        prompts = [' '.join(chunk) for chunk in sentence_chunks]
-        cols = st.columns(4)
-        with st.spinner('Generating Content...'):
-            for i, prompt in enumerate(prompts):
-                image_path, img_error = generate_image(prompt+st.session_state.on_topic )
-                audio, audio_error = generate_audio(prompt)
+        if 'image_paths' in st.session_state and st.session_state['image_paths']:
+                # Create a slider for image selection
     
-                with cols[i % 4]:
-                    if img_error:
-                        st.error(img_error)
-                    else:
-                        st.session_state['image_paths'].append(image_path)
-                        st.image(image_path, prompt, use_column_width=True)
+                col1, col2 = st.columns([2, 3])
     
-                    if audio_error:
-                        st.error(audio_error)
-                    else:
-                        st.audio(audio, format='audio/wav')
-
-# Tab 3: User Input and Results
-with tab3:
-
-
-    if 'image_paths' in st.session_state and st.session_state['image_paths']:
-            # Create a slider for image selection
-
-            col1, col2 = st.columns([2, 3])
-
-
-            with col2:
-                # Display the selected image
-                
-                image_index = st.radio(
-                    "Choose an image",
-                    options=list(range(len(st.session_state['image_paths']))),
-                    format_func=lambda x: f"Image {x + 1}"
-                )
-
-                
-                st.image(st.session_state['image_paths'][image_index])
-                
-            with col1:
-
-                st.header("Image Details")
-                st.divider()
-                st.subheader("Image Components")
+    
+                with col2:
+                    # Display the selected image
+                    
+                    image_index = st.radio(
+                        "Choose an image",
+                        options=list(range(len(st.session_state['image_paths']))),
+                        format_func=lambda x: f"Image {x + 1}"
+                    )
+    
+                    
+                    st.image(st.session_state['image_paths'][image_index])
+                    
+                with col1:
+    
+                    st.header("Image Details")
+                    st.divider()
+                    st.subheader("Image Components")
+                    
+                    image_conepts = get_image_concepts(st.session_state['image_paths'][image_index])
+    
+                    unique_names = set()
+                    for region in image_conepts:
+                        for concept in region.data.concepts:
+                            name = concept.name
+                            # Add unique names to the set
+                            unique_names.add(name)
                 
-                image_conepts = get_image_concepts(st.session_state['image_paths'][image_index])
-
-                unique_names = set()
-                for region in image_conepts:
-                    for concept in region.data.concepts:
-                        name = concept.name
-                        # Add unique names to the set
-                        unique_names.add(name)
+                    # Display unique names
+                    
+                    if unique_names:
+                        st.write(', '.join(unique_names))
+                    else:
+                        st.write("No unique items detected.")
+                    
+                    st.divider()
+                    st.subheader("Description of Our Image")
             
-                # Display unique names
-                
-                if unique_names:
-                    st.write(', '.join(unique_names))
-                else:
-                    st.write("No unique items detected.")
-                
-                st.divider()
-                st.subheader("Description of Our Image")
-        
-                image_text_output = analyze_image(st.session_state['image_paths'][image_index])
-                
-                st.write(image_text_output)
-                st.divider()
-
-                st.header("Create a Story About This Image")
-
-        
-                # Button for actions related to the selected image
-                if st.button("Create a New Story"):
-                    st.session_state['text_block'] = image_text_output
-                    #crewai_process(st.session_state['text_block'])
-                    #switch_tab(1)
+                    image_text_output = analyze_image(st.session_state['image_paths'][image_index])
+                    
+                    st.write(image_text_output)
+                    st.divider()
+    
+                    st.header("Create a Story About This Image")
+    
+            
+                    # Button for actions related to the selected image
+                    if st.button("Create a New Story"):
+                        st.session_state['text_block'] = image_text_output
+                        crewai_process(st.session_state['text_block'])
+                        switch_tab(tabs[0])
+