Update app.py

app.py

@@ -40,6 +40,22 @@ COLOR_PALETTES = {
     "Minimal": {"primary": "#000000", "secondary": "#E0E0E0", "text": "#212121", "background": "#FFFFFF"}
 }
 
+def get_custom_colors():
+    """Get custom colors from user input"""
+    col1, col2, col3 = st.columns(3)
+    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")
+    return {
+        "primary": primary,
+        "secondary": secondary,
+        "text": text,
+        "background": "#FFFFFF"  # Default background for custom
+    }
+
 def get_ai_response(prompt: str) -> str:
     """Get formatted content from Groq API."""
     headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
@@ -223,22 +239,36 @@ def main():
         height=100
     )
     
-    # Design settings
-    with st.expander("Design Options"):
-        col1, col2 = st.columns(2)
+# Enhanced Design settings
+    with st.expander("🎨 Design Options", expanded=True):
+        tab1, tab2 = st.tabs(["Preset Themes", "Custom Colors"])
         
-        with col1:
-            font = st.selectbox("Font", list(FONT_CHOICES.keys()), index=1)
-            color_palette = st.selectbox("Color Scheme", list(COLOR_PALETTES.keys()))
+        with tab1:
+            col1, col2 = st.columns(2)
+            with col1:
+                font = st.selectbox("Font", list(FONT_CHOICES.keys()), index=1)
+                color_palette = st.selectbox("Color Scheme", list(COLOR_PALETTES.keys()))
+            with col2:
+                title_size = st.slider("Title Size", 12, 44, 32)
+                heading_size = st.slider("Heading Size", 12, 36, 24)
+                body_size = st.slider("Body Text Size", 8, 28, 18)
+                set_background = st.checkbox("Apply Background", True)
         
-        with col2:
-            title_size = st.slider("Title Size", 12, 44, 32)
-            heading_size = st.slider("Heading Size", 12, 36, 24)
-            body_size = st.slider("Body Text Size", 8, 28, 18)
-            set_background = st.checkbox("Apply Background", True)
+        with tab2:
+            st.write("Create your own color scheme:")
+            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,
         "title_size": title_size,
         "heading_size": heading_size,