Update app.py

app.py

@@ -15,6 +15,7 @@ st.set_page_config(page_title="PPTX Smart Enhancer", layout="wide")
 # Configuration
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 MODEL_NAME = "gemma2-9b-it"
+
 # Style Options
 FONT_CHOICES = {
     "Arial": "Arial",
@@ -41,19 +42,21 @@ COLOR_PALETTES = {
 }
 
 def get_custom_colors():
-    """Get custom colors from user input"""
-    col1, col2, col3 = st.columns(3)
+    """Get custom colors from user input with background option"""
+    col1, col2, col3, col4 = st.columns(4)
     with col1:
         primary = st.color_picker("Primary Color", "#2B579A")
     with col2:
         secondary = st.color_picker("Secondary Color", "#5B9BD5")
     with col3:
         text = st.color_picker("Text Color", "#000000")
+    with col4:
+        background = st.color_picker("Background Color", "#FFFFFF")
     return {
         "primary": primary,
         "secondary": secondary,
         "text": text,
-        "background": "#FFFFFF"  # Default background for custom
+        "background": background
     }
 
 def get_ai_response(prompt: str) -> str:
@@ -87,10 +90,10 @@ def extract_slide_text(slide) -> str:
 def split_formatted_runs(text):
     """Split text into formatted runs removing markdown."""
     formatted_segments = []
-    # Split bold first
     bold_parts = re.split(r'(\*\*)', text)
     bold = False
     current_text = []
+    
     for part in bold_parts:
         if part == '**':
             if current_text:
@@ -98,25 +101,19 @@ def split_formatted_runs(text):
                 current_text = []
             bold = not bold
         else:
-            # Split italic within non-bold sections
-            if not bold:
-                italic_parts = re.split(r'(\*)', part)
-                italic = False
-                for italic_part in italic_parts:
-                    if italic_part == '*':
-                        if current_text:
-                            formatted_segments.append({'text': ''.join(current_text), 'bold': False, 'italic': italic})
-                            current_text = []
-                        italic = not italic
-                    else:
-                        current_text.append(italic_part)
-                if current_text:
-                    formatted_segments.append({'text': ''.join(current_text), 'bold': False, 'italic': italic})
-                    current_text = []
-            else:
-                current_text.append(part)
-    if current_text:
-        formatted_segments.append({'text': ''.join(current_text), 'bold': bold, 'italic': False})
+            italic_parts = re.split(r'(\*)', part)
+            italic = False
+            for italic_part in italic_parts:
+                if italic_part == '*':
+                    if current_text:
+                        formatted_segments.append({'text': ''.join(current_text), 'bold': bold, 'italic': italic})
+                        current_text = []
+                    italic = not italic
+                else:
+                    current_text.append(italic_part)
+            if current_text:
+                formatted_segments.append({'text': ''.join(current_text), 'bold': bold, 'italic': italic})
+                current_text = []
     return formatted_segments
 
 def apply_formatting(text_frame, content, design_settings):
@@ -125,10 +122,8 @@ def apply_formatting(text_frame, content, design_settings):
     paragraphs = [p for p in content.split('\n') if p.strip()]
     
     for para in paragraphs:
-        # Remove any markdown headers completely
         clean_para = re.sub(r'^#+\s*', '', para).strip()
         
-        # Handle section headers (original markdown headers become bold titles)
         if re.match(r'^(What is|Understanding|What\'s|Drive Theory|Incentive Theory)', clean_para, re.IGNORECASE):
             p = text_frame.add_paragraph()
             p.text = clean_para
@@ -137,7 +132,6 @@ def apply_formatting(text_frame, content, design_settings):
             p.space_after = Pt(12)
             continue
             
-        # Handle bullet points
         if re.match(r'^[\•\-\*] ', clean_para):
             p = text_frame.add_paragraph()
             p.level = 0
@@ -151,7 +145,6 @@ def apply_formatting(text_frame, content, design_settings):
             p.font.size = Pt(design_settings["body_size"])
             continue
             
-        # Regular paragraph with clean formatting
         p = text_frame.add_paragraph()
         segments = split_formatted_runs(clean_para)
         for seg in segments:
@@ -161,18 +154,16 @@ def apply_formatting(text_frame, content, design_settings):
             run.font.italic = seg['italic']
         p.font.size = Pt(design_settings["body_size"])
         p.space_after = Pt(6)
-        
+
 def enhance_slide(slide, user_prompt, design_settings):
     """Enhance slide content with proper formatting."""
     original_text = extract_slide_text(slide)
     if not original_text.strip():
         return
     
-    # Get enhanced content
     prompt = f"Improve this slide content:\n{original_text}\n\nInstructions: {user_prompt}"
     enhanced_content = get_ai_response(prompt)
     
-    # Clear existing content (keep title if exists)
     title_shape = getattr(slide.shapes, 'title', None)
     shapes_to_remove = [s for s in slide.shapes if s != title_shape]
     
@@ -180,7 +171,6 @@ def enhance_slide(slide, user_prompt, design_settings):
         sp = shape._element
         sp.getparent().remove(sp)
     
-    # Add formatted content
     left = Inches(1) if title_shape else Inches(0.5)
     top = Inches(1.5) if title_shape else Inches(0.5)
     textbox = slide.shapes.add_textbox(left, top, Inches(8), Inches(5))
@@ -192,7 +182,7 @@ def enhance_slide(slide, user_prompt, design_settings):
 
 def apply_design(slide, design_settings):
     """Apply visual design to slide."""
-    colors = COLOR_PALETTES[design_settings["color_palette"]]
+    colors = design_settings["colors"]
     
     for shape in slide.shapes:
         if hasattr(shape, "text_frame"):
@@ -201,7 +191,6 @@ def apply_design(slide, design_settings):
                     run.font.name = design_settings["font"]
                     run.font.color.rgb = RGBColor.from_string(colors["text"][1:])
                     
-                    # Title styling
                     if shape == getattr(slide.shapes, 'title', None):
                         run.font.color.rgb = RGBColor.from_string(colors["primary"][1:])
                         run.font.size = Pt(design_settings["title_size"])
@@ -209,7 +198,7 @@ def apply_design(slide, design_settings):
     
     if design_settings["set_background"]:
         slide.background.fill.solid()
-        slide.background.fill.fore_color.rgb = RGBColor.from_string(colors["secondary"][1:])
+        slide.background.fill.fore_color.rgb = RGBColor.from_string(colors["background"][1:])
 
 def process_presentation(uploaded_file, user_prompt, design_settings):
     """Process and enhance the entire presentation."""
@@ -220,7 +209,7 @@ def process_presentation(uploaded_file, user_prompt, design_settings):
     prs = Presentation(tmp_path)
     
     with st.spinner("Enhancing slides..."):
-        for i, slide in enumerate(prs.slides):
+        for slide in prs.slides:
             enhance_slide(slide, user_prompt, design_settings)
     
     os.unlink(tmp_path)
@@ -229,17 +218,13 @@ def process_presentation(uploaded_file, user_prompt, design_settings):
 def main():
     st.title("Professional PPTX Enhancer")
     
-    # File upload
     uploaded_file = st.file_uploader("Upload PowerPoint", type=["pptx"])
-    
-    # Content settings
     user_prompt = st.text_area(
         "Enhancement Instructions",
         placeholder="Example: 'Make headings clear and add bullet points for key items'",
         height=100
     )
     
-# Enhanced Design settings
     with st.expander("🎨 Design Options", expanded=True):
         tab1, tab2 = st.tabs(["Preset Themes", "Custom Colors"])
         
@@ -259,17 +244,10 @@ def main():
             custom_colors = get_custom_colors()
             use_custom = st.checkbox("Use Custom Colors")
     
-    # Prepare design settings
-    if use_custom:
-        colors = custom_colors
-        color_palette = "Custom"
-    else:
-        colors = COLOR_PALETTES[color_palette]
-    
     design_settings = {
         "font": FONT_CHOICES[font],
-        "colors": colors,
-        "color_palette": color_palette,
+        "colors": custom_colors if use_custom else COLOR_PALETTES[color_palette],
+        "color_palette": "Custom" if use_custom else color_palette,
         "title_size": title_size,
         "heading_size": heading_size,
         "body_size": body_size,
@@ -278,9 +256,12 @@ def main():
     
     if st.button("Enhance Presentation") and uploaded_file and user_prompt:
         try:
+            if not GROQ_API_KEY:
+                st.error("GROQ API key not found! Please set it in Space settings.")
+                st.stop()
+                
             enhanced_prs = process_presentation(uploaded_file, user_prompt, design_settings)
             
-            # Create download
             buffer = io.BytesIO()
             enhanced_prs.save(buffer)
             buffer.seek(0)