Update updated_components.py

updated_components.py

@@ -263,20 +263,12 @@ def render_storyboard_stage():
     # Display storyboard for review with edit options
     edited_storyboard = []
     
-    # Add controls for storyboard operations
-    col1, col2, col3 = st.columns([2, 1, 1])
+    # Add a "regenerate" button and AI options
+    col1, col2 = st.columns([3, 1])
     with col1:
-        # Get available models for regeneration
-        try:
-            from multi_llm_provider import get_ai_manager
-            ai_manager = get_ai_manager()
-            available_models = ai_manager.get_available_models()
-        except (ImportError, Exception):
-            available_models = {
-                "claude-3-sonnet-20250219": "Claude 3 Sonnet",
-                "claude-3-haiku-20250319": "Claude 3 Haiku",
-                "claude-3-opus-20250229": "Claude 3 Opus"
-            }
+        # Get AI models for regeneration
+        ai_manager = get_ai_manager()
+        available_models = ai_manager.get_available_models()
             
         selected_model = st.selectbox(
             "AI Model",
@@ -288,46 +280,7 @@ def render_storyboard_stage():
     
     with col2:
         if st.button("🔄 Regenerate All", help="Create a new storyboard with current title/purpose"):
-            from utils import generate_storyboard
-            with st.spinner("🐊 SlideGator.AI is rethinking your presentation structure..."):
-                storyboard = generate_storyboard(
-                    st.session_state.presentation_title,
-                    st.session_state.presentation_purpose,
-                    st.session_state.target_audience,
-                    model=selected_model
-                )
-                if storyboard:
-                    st.session_state.storyboard = storyboard
-                    st.success("Storyboard regenerated!")
-                    st.rerun()
-    
-    with col3:
-        if st.button("🔍 Analyze Flow", help="Analyze the narrative flow of your presentation"):
-            with st.spinner("🧠 Analyzing presentation flow..."):
-                # Convert storyboard to text for analysis
-                storyboard_text = ""
-                for i, slide in enumerate(st.session_state.storyboard):
-                    storyboard_text += f"Slide {i+1}: {slide.get('title', 'Untitled')}\n"
-                    storyboard_text += f"Purpose: {slide.get('purpose', '')}\n"
-                    key_points = slide.get('key_points', [])
-                    if isinstance(key_points, list):
-                        for point in key_points:
-                            storyboard_text += f"- {point}\n"
-                    else:
-                        storyboard_text += f"- {key_points}\n"
-                    storyboard_text += "\n"
-                
-                # Analyze the flow using available AI
-                try:
-                    from agents import enhance_slide_component
-                    from utils import generate_storyboard
-                    
-                    # Just display a simple analysis for now
-                    analysis = "Flow Analysis: Your presentation has a logical structure with a clear beginning, middle, and end. Consider adding more transition slides between major sections to improve the narrative flow."
-                    st.info("### Presentation Flow Analysis")
-                    st.write(analysis)
-                except Exception as e:
-                    st.error(f"Error analyzing presentation flow: {str(e)}")
+            st.info("Storyboard regeneration feature coming soon")
     
     # Add slide button with options
     col1, col2 = st.columns([3, 1])
@@ -361,59 +314,6 @@ def render_storyboard_stage():
             
             st.rerun()
     
-    # Add slide templates section
-    with st.expander("📑 Slide Templates"):
-        st.write("Add pre-designed slide templates to your presentation:")
-        
-        template_cols = st.columns(3)
-        
-        slide_templates = [
-            {"name": "Section Divider", "icon": "🏷️", "title": "Section Title", 
-             "purpose": "Introduce a new section", 
-             "key_points": ["A clean slide to mark a new section of your presentation"]},
-            
-            {"name": "Comparison", "icon": "⚖️", "title": "Comparison", 
-             "purpose": "Compare two options or approaches",
-             "key_points": ["Option A benefits", "Option A challenges", "Option B benefits", "Option B challenges"],
-             "visual_elements": ["Two-column layout", "Comparison table or chart"]},
-            
-            {"name": "Quote", "icon": "💬", "title": "Key Quote", 
-             "purpose": "Highlight an important quote",
-             "key_points": ["The quote goes here", "- Attribution"],
-             "visual_elements": ["Quotation marks graphic", "Simple background"]},
-            
-            {"name": "Data Visualization", "icon": "📊", "title": "Key Metrics", 
-             "purpose": "Present important data",
-             "key_points": ["Metric 1 and its significance", "Metric 2 and its significance", "Trends and patterns"],
-             "visual_elements": ["Chart or graph", "Data visualization"]},
-            
-            {"name": "Call to Action", "icon": "🎯", "title": "Next Steps", 
-             "purpose": "Define clear action items",
-             "key_points": ["Specific action item 1", "Specific action item 2", "Timeline and ownership"],
-             "visual_elements": ["Arrow or path graphic", "Timeline visualization"]},
-            
-            {"name": "Team Slide", "icon": "👥", "title": "Our Team", 
-             "purpose": "Introduce key team members",
-             "key_points": ["Team member 1 with role", "Team member 2 with role", "Team member 3 with role"],
-             "visual_elements": ["Team photos or icons", "Organizational chart"]}
-        ]
-        
-        for i, template in enumerate(slide_templates):
-            with template_cols[i % 3]:
-                if st.button(f"{template['icon']} {template['name']}", key=f"template_{template['name']}"):
-                    # Create a new slide based on the template
-                    new_slide = {
-                        'title': template['title'],
-                        'purpose': template['purpose'],
-                        'key_points': template['key_points'],
-                        'visual_elements': template.get('visual_elements', [])
-                    }
-                    
-                    # Add to storyboard
-                    st.session_state.storyboard.append(new_slide)
-                    st.success(f"Added {template['name']} slide!")
-                    st.rerun()
-    
     # Display storyboard slides
     for i, slide in enumerate(st.session_state.storyboard):
         # Set current slide in session state when expander is opened
@@ -467,50 +367,88 @@ def render_storyboard_stage():
                 
             visual_elements = st.text_area(f"Visual Elements (one per line) ###{i}", value=visual_elements_text)
             
-            # AI enhancement options
-            with st.expander("🧠 AI Enhancement"):
-                ai_cols = st.columns(2)
-                
-                with ai_cols[0]:
-                    # Get AI models for enhancement
-                    enhancement_model = "claude-3-sonnet-20250219"  # Default model
+            # 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()
                 
-                with ai_cols[1]:
-                    enhancement_type = st.selectbox(
-                        "Enhancement Type",
-                        ["Content", "Title", "Visual Suggestions", "All"],
-                        key=f"enhance_type_{i}"
-                    )
+                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("✨ 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 []
-                        }
+                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)
                         
-                        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)}")
+                        # 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 = {