Create utils.py

utils.py

@@ -4,20 +4,7 @@ from io import BytesIO
 import json
 import anthropic
 import os
-
-# Initialize the Claude client with API key from environment (Huggingface secrets)
-api_key = os.getenv("ANTHROPIC_API_KEY")
-if not api_key:
-    st.sidebar.error("ANTHROPIC_API_KEY not found in environment variables")
-    client = None
-else:
-    # Create the client
-    try:
-        client = anthropic.Anthropic(api_key=api_key)
-        st.sidebar.success("Anthropic API client initialized successfully")
-    except Exception as e:
-        st.sidebar.error(f"Error initializing Anthropic API: {str(e)}")
-        client = None
+from multi_llm_provider import get_ai_manager
 
 # Define available templates
 TEMPLATES = {
@@ -79,12 +66,32 @@ TEMPLATES = {
     }
 }
 
-def generate_storyboard(title, purpose, audience):
-    """Generate a presentation storyboard using Claude"""
-    if not client:
-        st.error("Anthropic API client not available")
-        return default_storyboard(title)
+def generate_storyboard(title, purpose, audience, model=None):
+    """Generate a presentation storyboard using the specified AI model"""
+    
+    # Get AI manager
+    ai_manager = get_ai_manager()
+    
+    # Use default model if none specified
+    if not model:
+        model = st.session_state.get("default_model", "claude-3-sonnet-20250219")
     
+    # Check for web research data
+    web_research = ""
+    if hasattr(st.session_state, 'web_research') and st.session_state.web_research:
+        research_data = st.session_state.web_research
+        web_research = "Recent information from web research:\n\n"
+        
+        for i, item in enumerate(research_data[:3]):  # Limit to 3 results
+            title = item.get("title", "No title")
+            snippet = item.get("snippet", "No snippet")
+            url = item.get("url", "No URL")
+            
+            web_research += f"{i+1}. {title}\n"
+            web_research += f"   {snippet}\n"
+            web_research += f"   Source: {url}\n\n"
+    
+    # Create the prompt
     prompt = f"""
     You are an expert presentation designer with deep expertise in creating compelling PowerPoint presentations.
     
@@ -93,6 +100,8 @@ def generate_storyboard(title, purpose, audience):
     - Purpose: {purpose}
     - Target Audience: {audience}
     
+    {web_research if web_research else ""}
+    
     For each slide, include:
     1. Slide title
     2. Slide purpose
@@ -116,21 +125,21 @@ def generate_storyboard(title, purpose, audience):
     ]
     """
     
+    system_prompt = "You are an expert presentation designer specializing in creating logical, compelling PowerPoint storyboards. Always output valid JSON without explanations or extra text."
+    
     with st.sidebar:
         progress_text = st.empty()
-        progress_text.write("Generating storyboard...")
+        progress_text.write(f"Generating storyboard using {model}...")
     
     try:
-        response = client.messages.create(
-            model="claude-3-7-sonnet-20250219",
-            max_tokens=4000,
-            temperature=0.7,
-            system="You are an expert presentation designer specializing in creating logical, compelling PowerPoint storyboards. Always output valid JSON without explanations or extra text.",
-            messages=[
-                {"role": "user", "content": prompt}
-            ]
+        # Generate storyboard using the selected model
+        response_text = ai_manager.generate_text(
+            prompt=prompt,
+            model=model,
+            system_prompt=system_prompt,
+            temperature=st.session_state.get("ai_temperature", 0.7),
+            max_tokens=4000
         )
-        response_text = response.content[0].text
         
         # For debugging, show raw response
         st.sidebar.write("Raw response: " + response_text[:200] + "...")
@@ -143,6 +152,13 @@ def generate_storyboard(title, purpose, audience):
             json_str = response_text[json_start:json_end]
             parsed_json = json.loads(json_str)
             progress_text.write(f"Generated storyboard with {len(parsed_json)} slides")
+            
+            # Add AI model used to each slide for future reference
+            for slide in parsed_json:
+                if "ai_settings" not in slide:
+                    slide["ai_settings"] = {}
+                slide["ai_settings"]["model"] = model
+            
             return parsed_json
         else:
             progress_text.error("JSON not found in response")
@@ -185,11 +201,15 @@ def default_storyboard(title):
         }
     ]
 
-def generate_slide_content(slide_info, template_name):
-    """Generate detailed content for a slide using Claude"""
-    if not client:
-        st.error("Anthropic API client not available")
-        return default_slide_content(slide_info)
+def generate_slide_content(slide_info, template_name, model=None):
+    """Generate detailed content for a slide using the specified AI model"""
+    
+    # Get AI manager
+    ai_manager = get_ai_manager()
+    
+    # Use default model if none specified
+    if not model:
+        model = slide_info.get("ai_settings", {}).get("model", st.session_state.get("default_model", "claude-3-sonnet-20250219"))
     
     prompt = f"""
     Create detailed content for a PowerPoint slide with the following information:
@@ -218,17 +238,17 @@ def generate_slide_content(slide_info, template_name):
     Focus on clarity, impact, and alignment with the slide's purpose.
     """
     
+    system_prompt = "You are an expert presentation content creator. Always output valid JSON without explanations or extra text."
+    
     try:
-        response = client.messages.create(
-            model="claude-3-7-sonnet-20250219",
-            max_tokens=2000,
-            temperature=0.5,
-            system="You are an expert presentation content creator. Always output valid JSON without explanations or extra text.",
-            messages=[
-                {"role": "user", "content": prompt}
-            ]
+        # Generate content using the selected model
+        response_text = ai_manager.generate_text(
+            prompt=prompt,
+            model=model,
+            system_prompt=system_prompt,
+            temperature=st.session_state.get("ai_temperature", 0.5),
+            max_tokens=2000
         )
-        response_text = response.content[0].text
         
         # Extract the JSON from the response
         json_start = response_text.find("{")
@@ -236,7 +256,14 @@ def generate_slide_content(slide_info, template_name):
         
         if json_start >= 0 and json_end > 0:
             json_str = response_text[json_start:json_end]
-            return json.loads(json_str)
+            content = json.loads(json_str)
+            
+            # Store AI model used
+            if "ai_settings" not in content:
+                content["ai_settings"] = {}
+            content["ai_settings"]["model"] = model
+            
+            return content
         else:
             # Fallback: create a basic slide
             st.sidebar.error(f"JSON not found in response: {response_text}")
@@ -264,6 +291,12 @@ def create_slide_preview(slide, template_name):
     colors = template["colors"]
     fonts = template["fonts"]
     
+    # Apply custom colors if available
+    if "custom_colors" in st.session_state and template_name in st.session_state.custom_colors:
+        custom_colors = st.session_state.custom_colors[template_name]
+        colors["primary"] = custom_colors.get("primary", colors["primary"])
+        colors["accent"] = custom_colors.get("accent", colors["accent"])
+    
     # Prepare content
     title = slide.get('title', 'Untitled Slide')
     
@@ -337,87 +370,6 @@ def create_slide_preview(slide, template_name):
     
     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:
@@ -427,7 +379,38 @@ def add_custom_template(template_file):
         
         # Store in session state
         st.session_state.custom_template = file_content
+        st.session_state.custom_template_name = template_file.name
+        
+        # Extract template information if possible
+        try:
+            # Try to extract color scheme from the template
+            background_color = "#FFFFFF"  # Default
+            accent_color = "#0073E6"  # Default
+            
+            # Look for theme colors in the first slide if available
+            if len(prs.slides) > 0:
+                slide = prs.slides[0]
+                
+                # Look for background color
+                for shape in slide.shapes:
+                    if hasattr(shape, 'fill') and shape.fill.type != 0:  # 0 is None
+                        if hasattr(shape.fill, 'fore_color') and hasattr(shape.fill.fore_color, 'rgb'):
+                            rgb = shape.fill.fore_color.rgb
+                            if rgb:
+                                background_color = f"#{rgb.r:02x}{rgb.g:02x}{rgb.b:02x}"
+                                break
+            
+            # Update custom template with extracted colors
+            TEMPLATES["custom"]["colors"]["primary"] = accent_color
+            TEMPLATES["custom"]["colors"]["secondary"] = background_color
+            
+            # Get the number of slide layouts available
+            num_layouts = len(prs.slide_layouts)
+            st.session_state.custom_template_layouts = num_layouts
+            
+        except Exception as extract_error:
+            st.warning(f"Could not extract all template details: {str(extract_error)}")
         
-        return True, f"Custom template '{template_file.name}' uploaded successfully!"
+        return True, f"Custom template '{template_file.name}' uploaded successfully with {len(prs.slides)} master layouts!"
     except Exception as e:
         return False, f"Error with template file: {str(e)}"