Update utils.py

utils.py

@@ -21,4 +21,385 @@ TEMPLATES = {
             "body": "Calibri, sans-serif"
         }
     },
-    "creative": {
+    "creative": {
+        "name": "Creative",
+        "description": "Vibrant design with modern elements and bold colors",
+        "colors": {
+            "primary": "#FF5757",    # Coral red
+            "secondary": "#FFDE59",  # Yellow
+            "accent": "#5CE1E6",     # Turquoise
+            "text": "#333333"        # Dark gray
+        },
+        "fonts": {
+            "title": "Montserrat, sans-serif",
+            "body": "Roboto, sans-serif"
+        }
+    },
+    "minimalist": {
+        "name": "Minimalist",
+        "description": "Simple, elegant design with lots of whitespace",
+        "colors": {
+            "primary": "#FFFFFF",    # White
+            "secondary": "#F5F5F5",  # Off-white
+            "accent": "#000000",     # Black
+            "text": "#333333"        # Dark gray
+        },
+        "fonts": {
+            "title": "Helvetica, Arial, sans-serif",
+            "body": "Helvetica, Arial, sans-serif"
+        }
+    },
+    "custom": {
+        "name": "Custom",
+        "description": "Your custom uploaded template",
+        "colors": {
+            "primary": "#0073E6",    # Blue
+            "secondary": "#FFFFFF",  # White
+            "accent": "#FF6D00",     # Orange
+            "text": "#333333"        # Dark gray
+        },
+        "fonts": {
+            "title": "Arial, sans-serif",
+            "body": "Calibri, sans-serif"
+        }
+    }
+}
+
+# LLM helper function
+def get_llm():
+    """Get the LLM client with API key from environment"""
+    return ChatAnthropic(
+        model="claude-3-5-sonnet-20240620",
+        anthropic_api_key=st.secrets["ANTHROPIC_API_KEY"],
+        temperature=0.7
+    )
+
+def generate_storyboard(title, purpose, audience):
+    """Generate a presentation storyboard using Claude"""
+    llm = get_llm()
+    
+    prompt = f"""
+    You are an expert presentation designer with deep expertise in creating compelling PowerPoint presentations.
+    
+    I need you to create a storyboard for a presentation with the following details:
+    - Title: {title}
+    - Purpose: {purpose}
+    - Target Audience: {audience}
+    
+    For each slide, include:
+    1. Slide title
+    2. Slide purpose
+    3. Key content points
+    4. Suggested visual elements
+    
+    Think deeply about the narrative flow, ensuring each slide builds on the previous one.
+    Format your response as a JSON list where each item is a slide with the above elements.
+    
+    Start with an engaging title slide and end with a clear call-to-action or summary slide.
+    
+    The response should be a valid JSON array like this:
+    [
+      {{
+        "title": "Slide Title",
+        "purpose": "Purpose of this slide",
+        "key_points": ["Key point 1", "Key point 2"],
+        "visual_elements": ["Visual element 1", "Visual element 2"]
+      }},
+      ...
+    ]
+    """
+    
+    st.sidebar.write("Generating storyboard...")
+    
+    try:
+        messages = [
+            SystemMessage(content="You are an expert presentation designer specializing in creating logical, compelling PowerPoint storyboards. Always output valid JSON without explanations or extra text."),
+            HumanMessage(content=prompt)
+        ]
+        response = llm.invoke(messages)
+        response_text = response.content
+        
+        # Extract the JSON from the response
+        json_start = response_text.find("[")
+        json_end = response_text.rfind("]") + 1
+        
+        if json_start >= 0 and json_end > 0:
+            json_str = response_text[json_start:json_end]
+            parsed_json = json.loads(json_str)
+            st.sidebar.write(f"Generated storyboard with {len(parsed_json)} slides")
+            return parsed_json
+        else:
+            st.sidebar.error("JSON not found in response")
+            # Create a default storyboard as fallback
+            return default_storyboard(title)
+    except Exception as e:
+        st.sidebar.error(f"Error generating storyboard: {str(e)}")
+        return default_storyboard(title)
+
+def default_storyboard(title):
+    """Create a default storyboard when API calls fail"""
+    return [
+        {
+            "title": title,
+            "purpose": "Title slide",
+            "key_points": ["Introduction"],
+            "visual_elements": ["Company logo", "Presenter name"]
+        },
+        {
+            "title": "Agenda",
+            "purpose": "Outline the presentation contents",
+            "key_points": ["Overview of key topics", "Agenda items", "Presentation goals"],
+            "visual_elements": ["Bullet list", "Simple timeline graphic"]
+        },
+        {
+            "title": "Key Points",
+            "purpose": "Present main content",
+            "key_points": ["Main point 1", "Main point 2", "Main point 3"],
+            "visual_elements": ["Relevant icons", "Supporting image"]
+        },
+        {
+            "title": "Conclusion",
+            "purpose": "Summarize key points and next steps",
+            "key_points": ["Summary of main points", "Call to action", "Next steps"],
+            "visual_elements": ["Summary graphic", "Contact information"]
+        }
+    ]
+
+def generate_slide_content(slide_info, template_name):
+    """Generate detailed content for a slide using Claude"""
+    llm = get_llm()
+    
+    prompt = f"""
+    Create detailed content for a PowerPoint slide with the following information:
+    
+    Slide Title: {slide_info.get('title', 'Untitled')}
+    Slide Purpose: {slide_info.get('purpose', 'No purpose provided')}
+    Key Points: {slide_info.get('key_points', [])}
+    Visual Elements: {slide_info.get('visual_elements', [])}
+    
+    The presentation uses the {template_name} template style.
+    
+    Provide the following in your response as JSON:
+    1. Final slide title (refined if needed)
+    2. Main content text (bullet points or paragraphs as appropriate)
+    3. Notes for the presenter
+    4. Specific visual instructions (charts, images, etc.)
+    
+    Your response must be a valid JSON object like this:
+    {{
+      "title": "Refined Title",
+      "content": ["Point 1", "Point 2", "Point 3"],
+      "notes": "Notes for the presenter",
+      "visual_instructions": "Details about visuals"
+    }}
+    
+    Focus on clarity, impact, and alignment with the slide's purpose.
+    """
+    
+    try:
+        messages = [
+            SystemMessage(content="You are an expert presentation content creator. Always output valid JSON without explanations or extra text."),
+            HumanMessage(content=prompt)
+        ]
+        response = llm.invoke(messages)
+        response_text = response.content
+        
+        # Extract the JSON from the response
+        json_start = response_text.find("{")
+        json_end = response_text.rfind("}") + 1
+        
+        if json_start >= 0 and json_end > 0:
+            json_str = response_text[json_start:json_end]
+            return json.loads(json_str)
+        else:
+            # Fallback: create a basic slide
+            return {
+                "title": slide_info.get('title', 'Untitled'),
+                "content": slide_info.get('key_points', ["Add content here"]),
+                "notes": f"This slide covers: {slide_info.get('purpose', 'No purpose specified')}",
+                "visual_instructions": str(slide_info.get('visual_elements', ["No visual elements specified"]))
+            }
+    except Exception as e:
+        st.sidebar.error(f"Error generating slide content: {str(e)}")
+        # Return a default structure
+        return {
+            "title": slide_info.get('title', 'Untitled'),
+            "content": slide_info.get('key_points', ["Content to be added"]),
+            "notes": "Add presenter notes here",
+            "visual_instructions": "Add visual elements here"
+        }
+
+def create_slide_preview(slide, template_name):
+    """Create an HTML preview of a slide"""
+    # Get template info
+    template = TEMPLATES.get(template_name, TEMPLATES["professional"])
+    colors = template["colors"]
+    fonts = template["fonts"]
+    
+    # Prepare content
+    title = slide.get('title', 'Untitled Slide')
+    
+    if isinstance(slide.get('content', []), list):
+        content = slide.get('content', [])
+    else:
+        content_text = slide.get('content', '')
+        content = content_text.split('\n') if content_text else []
+    
+    # Determine layout based on slide design or content
+    layout_type = "standard"
+    if "design" in slide and "layout" in slide["design"]:
+        layout_type = slide["design"]["layout"].lower()
+    
+    # Generate HTML based on layout type
+    if "two column" in layout_type or "comparison" in layout_type:
+        # Two-column layout
+        mid_point = len(content) // 2
+        left_content = content[:mid_point]
+        right_content = content[mid_point:]
+        
+        content_html = f"""
+        <div class="row" style="display: flex; flex-direction: row;">
+            <div class="col" style="flex: 1; padding-right: 10px;">
+                <ul>
+                    {"".join(f'<li style="margin-bottom: 8px;">{item}</li>' for item in left_content)}
+                </ul>
+            </div>
+            <div class="col" style="flex: 1; padding-left: 10px;">
+                <ul>
+                    {"".join(f'<li style="margin-bottom: 8px;">{item}</li>' for item in right_content)}
+                </ul>
+            </div>
+        </div>
+        """
+    elif "quote" in layout_type:
+        # Quote layout
+        if content:
+            quote_text = content[0] if isinstance(content, list) and len(content) > 0 else "Quote text"
+            author = content[1] if isinstance(content, list) and len(content) > 1 else ""
+            
+            content_html = f"""
+            <div class="quote-container" style="text-align: center; padding: 20px;">
+                <blockquote style="font-size: 24px; font-style: italic; color: {colors['accent']};">
+                    "{quote_text}"
+                </blockquote>
+                {f'<div style="text-align: right; font-size: 18px;">— {author}</div>' if author else ''}
+            </div>
+            """
+        else:
+            content_html = '<div class="quote-container" style="text-align: center;"><p>No quote content</p></div>'
+    else:
+        # Standard layout with bullet points
+        content_html = f"""
+        <ul>
+            {"".join(f'<li style="margin-bottom: 8px;">{item}</li>' for item in content)}
+        </ul>
+        """
+    
+    # Create the full HTML preview
+    html = f"""
+    <div class="slide-preview" style="background-color: {colors['secondary']}; border: 1px solid #dee2e6; border-radius: 4px; padding: 20px; height: 300px; overflow: hidden; font-family: {fonts['body']};">
+        <div class="slide-title" style="color: {colors['text']}; font-family: {fonts['title']}; font-size: 24px; margin-bottom: 20px; border-bottom: 2px solid {colors['primary']}; padding-bottom: 10px;">
+            {title}
+        </div>
+        <div class="slide-content" style="color: {colors['text']}; font-family: {fonts['body']}; font-size: 16px;">
+            {content_html}
+        </div>
+    </div>
+    """
+    
+    return html
+
+def create_ppt(slides_content, template_name):
+    """Create a PowerPoint presentation from the slides content"""
+    # Create a basic presentation
+    prs = Presentation()
+    
+    # Get template info
+    template = TEMPLATES.get(template_name, TEMPLATES["professional"])
+    
+    # Title slide
+    title_slide_layout = prs.slide_layouts[0]
+    slide = prs.slides.add_slide(title_slide_layout)
+    title = slide.shapes.title
+    subtitle = slide.placeholders[1]
+    
+    title.text = slides_content[0]["title"]
+    subtitle.text = "Created with AI PowerPoint Creator"
+    
+    # Content slides
+    for slide_content in slides_content[1:]:
+        # Determine best layout based on content or explicit design
+        layout_type = "standard"
+        if "design" in slide_content and "layout" in slide_content["design"]:
+            layout_type = slide_content["design"]["layout"].lower()
+        
+        # Select appropriate slide layout
+        if "two column" in layout_type or "comparison" in layout_type:
+            content_slide_layout = prs.slide_layouts[3] if len(prs.slide_layouts) > 3 else prs.slide_layouts[1]
+        elif "title only" in layout_type or "quote" in layout_type:
+            content_slide_layout = prs.slide_layouts[5] if len(prs.slide_layouts) > 5 else prs.slide_layouts[0]
+        elif "picture" in layout_type:
+            content_slide_layout = prs.slide_layouts[6] if len(prs.slide_layouts) > 6 else prs.slide_layouts[1]
+        else:
+            content_slide_layout = prs.slide_layouts[1]  # Default: Title and Content
+            
+        slide = prs.slides.add_slide(content_slide_layout)
+        
+        # Add title
+        if slide.shapes.title:
+            slide.shapes.title.text = slide_content["title"]
+        
+        # Add content based on layout and available placeholders
+        content_placeholders = [shape for shape in slide.placeholders 
+                               if shape.placeholder_format.type != 1]  # 1 is title
+        
+        if len(content_placeholders) > 0:
+            content_placeholder = content_placeholders[0]
+            
+            # Format content based on what's available and layout type
+            if "content" in slide_content:
+                if isinstance(slide_content["content"], list):
+                    if "two column" in layout_type and len(content_placeholders) > 1:
+                        # Split content for two columns
+                        left_placeholder = content_placeholders[0]
+                        right_placeholder = content_placeholders[1]
+                        
+                        mid_point = len(slide_content["content"]) // 2
+                        left_content = slide_content["content"][:mid_point]
+                        right_content = slide_content["content"][mid_point:]
+                        
+                        left_placeholder.text = "\n".join(f"• {point}" for point in left_content)
+                        right_placeholder.text = "\n".join(f"• {point}" for point in right_content)
+                    else:
+                        # Standard bullet points
+                        content_placeholder.text = "\n".join(f"• {point}" for point in slide_content["content"])
+                else:
+                    # Plain text
+                    content_placeholder.text = slide_content["content"]
+            else:
+                content_placeholder.text = "Content to be added"
+        
+        # Add notes if available
+        if "notes" in slide_content and slide_content["notes"]:
+            notes_slide = slide.notes_slide
+            notes_slide.notes_text_frame.text = slide_content["notes"]
+    
+    # Save to BytesIO
+    output = BytesIO()
+    prs.save(output)
+    output.seek(0)
+    return output
+
+def add_custom_template(template_file):
+    """Add a custom template from an uploaded file"""
+    try:
+        # Validate it's a proper PowerPoint file
+        file_content = template_file.getvalue()
+        prs = Presentation(BytesIO(file_content))
+        
+        # Store in session state
+        st.session_state.custom_template = file_content
+        
+        return True, f"Custom template '{template_file.name}' uploaded successfully!"
+    except Exception as e:
+        return False, f"Error with template file: {str(e)}"