Update updated_components.py

updated_components.py

@@ -253,7 +253,7 @@ def render_ideation_stage():
                     st.rerun()
 
 def render_storyboard_stage():
-    """Render the storyboard review stage UI"""
+    """Render the storyboard review stage UI (simplified version without AI features)"""
     display_navigation_bar()
     st.header("📋 Step 2: Review and Edit Storyboard")
     
@@ -263,24 +263,9 @@ def render_storyboard_stage():
     # Display storyboard for review with edit options
     edited_storyboard = []
     
-    # Add a "regenerate" button and AI options
-    col1, col2 = st.columns([3, 1])
-    with col1:
-        # Get AI models for regeneration
-        ai_manager = get_ai_manager()
-        available_models = ai_manager.get_available_models()
-            
-        selected_model = st.selectbox(
-            "AI Model",
-            options=list(available_models.keys()),
-            format_func=lambda x: available_models.get(x, x),
-            index=0,
-            key="storyboard_regen_model"
-        )
-    
-    with col2:
-        if st.button("🔄 Regenerate All", help="Create a new storyboard with current title/purpose"):
-            st.info("Storyboard regeneration feature coming soon")
+    # Simple regenerate button 
+    if st.button("🔄 Regenerate All", help="Create a new storyboard with current title/purpose"):
+        st.info("Storyboard regeneration feature coming soon")
     
     # Add slide button with options
     col1, col2 = st.columns([3, 1])
@@ -367,89 +352,6 @@ def render_storyboard_stage():
                 
             visual_elements = st.text_area(f"Visual Elements (one per line) ###{i}", value=visual_elements_text)
             
-            # AI enhancement options - USING COLUMNS INSTEAD OF NESTED EXPANDER
-            st.write("##### 🧠 AI Enhancement")
-            ai_cols = st.columns(2)
-            
-            with ai_cols[0]:
-                # Get AI manager
-                ai_manager = get_ai_manager()
-                available_models = ai_manager.get_available_models()
-                
-                enhancement_model = st.selectbox(
-                    "AI Model",
-                    options=list(available_models.keys()),
-                    format_func=lambda x: available_models.get(x, x),
-                    index=0,
-                    key=f"enhance_model_{i}"
-                )
-            
-            with ai_cols[1]:
-                enhancement_type = st.selectbox(
-                    "Enhancement Type",
-                    ["Content", "Title", "Visual Suggestions", "All"],
-                    key=f"enhance_type_{i}"
-                )
-            
-            if st.button("✨ Enhance with AI", key=f"enhance_slide_{i}"):
-                with st.spinner("🧠 Enhancing slide content..."):
-                    # Update slide with current edits first
-                    updated_slide = {
-                        'title': slide_title,
-                        'purpose': slide_purpose,
-                        'key_points': key_points.split("\n") if "\n" in key_points else [key_points] if key_points else [],
-                        'visual_elements': visual_elements.split("\n") if "\n" in visual_elements else [visual_elements] if visual_elements else []
-                    }
-                    
-                    try:
-                        from agents import enhance_slide_component
-                        
-                        # Apply enhancement based on selected type
-                        if enhancement_type in ["Content", "All"]:
-                            updated_slide, result = enhance_slide_component(updated_slide, "content")
-                        
-                        if enhancement_type in ["Title", "All"]:
-                            updated_slide, result = enhance_slide_component(updated_slide, "title")
-                        
-                        if enhancement_type in ["Visual Suggestions", "All"]:
-                            updated_slide, result = enhance_slide_component(updated_slide, "visuals")
-                        
-                        # Update the storyboard with enhanced slide
-                        st.session_state.storyboard[i] = updated_slide
-                        st.success("Slide enhanced with AI!")
-                        st.rerun()
-                    except Exception as e:
-                        st.error(f"Error enhancing slide: {str(e)}")
-            
-            # Web search enhancement - MOVED OUT OF NESTED EXPANDER
-            if perplexity_key:
-                st.write("##### 🔍 Web Search Enhancement")
-                search_query = st.text_input(
-                    "Search Query",
-                    value=f"latest information about {slide_title}",
-                    key=f"search_query_{i}"
-                )
-                
-                if st.button("🔍 Search & Enhance", key=f"search_enhance_{i}"):
-                    # Update slide with current edits first
-                    updated_slide = {
-                        'title': slide_title,
-                        'purpose': slide_purpose,
-                        'key_points': key_points.split("\n") if "\n" in key_points else [key_points] if key_points else [],
-                        'visual_elements': visual_elements.split("\n") if "\n" in visual_elements else [visual_elements] if visual_elements else []
-                    }
-                    
-                    try:
-                        # Enhance with web search
-                        enhanced_slide = ai_manager.enhance_with_web_search(updated_slide, search_query)
-                        
-                        # Update the storyboard
-                        st.session_state.storyboard[i] = enhanced_slide
-                        st.success("Slide enhanced with web search results!")
-                        st.rerun()
-                    except Exception as e:
-                        st.error(f"Error enhancing with web search: {str(e)}")
-            
             # Update storyboard with edits
             edited_slide = {
                 'title': slide_title,