Update app.py

app.py

@@ -411,120 +411,120 @@ def switch_tab(tab_name):
 tab1, tab2, tab3 = st.tabs(tabs)
 
 # Tab 1: Introduction
-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
-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 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...'):
-                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')
+                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:
 
-# Tab 3: User Input and Results
-elif st.session_state.current_tab == tabs[2]:
-    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:
+    # Streamlit main page
     
-                    st.header("Image Details")
-                    st.divider()
-                    st.subheader("Image Components")
-                    
-                    image_conepts = get_image_concepts(st.session_state['image_paths'][image_index])
+    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)
     
-                    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))
+                with cols[i % 4]:
+                    if img_error:
+                        st.error(img_error)
                     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")
+                        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
+
+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")
+                
+                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)
             
-                    # 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])
-                    
-                 
+                # 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(tabs[0])
+                
+