Update app.py

app.py

@@ -135,6 +135,46 @@ except ImportError:
     FONTTOOLS_AVAILABLE = False
     print("⚠️ fontTools not available - font analysis limited")
 
+# =============================================
+# ENHANCED COLOR MAP
+# =============================================
+COLOR_MAP = {
+    # Basic colors
+    "white": "FFFFFF",
+    "black": "000000",
+    "red": "FF0000",
+    "green": "00FF00", 
+    "blue": "0000FF",
+    "yellow": "FFFF00",
+    
+    # Purple variations
+    "purple": "800080",      # Medium purple
+    "darkpurple": "4B0082",  # Indigo
+    "lightpurple": "9370DB", # Light purple
+    "violet": "EE82EE",      # Violet
+    "magenta": "FF00FF",     # Magenta
+    
+    # Gold variations
+    "gold": "FFD700",        # Gold
+    "orange": "FFA500",      # Orange
+    "darkgold": "B8860B",    # Dark goldenrod
+    
+    # Other colors
+    "cyan": "00FFFF",
+    "pink": "FFC0CB",
+    "brown": "A52A2A",
+    "gray": "808080",
+    "navy": "000080",
+    "teal": "008080",
+    "maroon": "800000",
+    "olive": "808000",
+    "coral": "FF7F50",
+    "lavender": "E6E6FA",
+    "turquoise": "40E0D0",
+    "indigo": "4B0082",
+    "crimson": "DC143C"
+}
+
 # =============================================
 # CREATE FASTAPI APP
 # =============================================
@@ -425,30 +465,23 @@ def wrap_text_for_ass(text: str, max_width: int, font_family: str, font_size: in
     return "\\N".join(lines)
 
 def create_caption_ass(captions: List[CaptionSegment], style: CaptionStyle, work_dir: str, font_path: str) -> str:
-    """Create ASS subtitle file for captions"""
+    """Create ASS subtitle file for captions with enhanced colors"""
     
     # Get actual font family name
     font_family_name = get_font_family_name(font_path)
     print(f"📝 Caption font family: {font_family_name}")
     
-    # Color mapping
-    color_map = {
-        "white": "FFFFFF", "black": "000000", "red": "FF0000",
-        "green": "00FF00", "blue": "0000FF", "yellow": "FFFF00",
-        "gold": "FFD700", "purple": "800080", "cyan": "00FFFF"
-    }
-    
     # Get font color and convert from RRGGBB to BBGGRR for ASS
-    font_color_rgb = color_map.get(style.color.lower(), "FFFFFF")
+    font_color_rgb = COLOR_MAP.get(style.color.lower(), "FFFFFF")
     # Convert RGB to BGR (swap first two and last two characters)
     font_color_bgr = font_color_rgb[4:6] + font_color_rgb[2:4] + font_color_rgb[0:2]
     print(f"🎨 Font color: {style.color} -> RGB={font_color_rgb} -> BGR={font_color_bgr}")
     
     # Parse background color
     bg_parts = style.bg_color.split('@')
-    bg_color_name = bg_parts[0]
+    bg_color_name = bg_parts[0].lower()
     bg_opacity = float(bg_parts[1]) if len(bg_parts) > 1 else 0.5
-    bg_color_rgb = color_map.get(bg_color_name.lower(), "000000")
+    bg_color_rgb = COLOR_MAP.get(bg_color_name, "000000")
     # Convert RGB to BGR for background
     bg_color_bgr = bg_color_rgb[4:6] + bg_color_rgb[2:4] + bg_color_rgb[0:2]
     bg_alpha = int((1 - bg_opacity) * 255)
@@ -468,8 +501,7 @@ def create_caption_ass(captions: List[CaptionSegment], style: CaptionStyle, work
     margin_v = style.margin
     
     # Create ASS header with proper border settings
-    # BorderStyle=1 means outline + shadow, Outline=1 gives thin outline, Shadow=0 for no shadow
-    # We set BackColour to the background color, and it will be used as box fill
+    # BorderStyle=3 means opaque box, Outline controls box padding
     ass_header = f"""[Script Info]
 ; Script generated by Video Styling Space - Auto Caption
 ScriptType: v4.00+
@@ -510,7 +542,7 @@ Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
     return ass_file
 
 # =============================================
-# SRT PARSING FUNCTIONS (ADDED)
+# SRT PARSING FUNCTIONS
 # =============================================
 
 def parse_srt_file(srt_content: str) -> List[CaptionSegment]:
@@ -570,11 +602,11 @@ def srt_time_to_seconds(time_str: str) -> float:
     return 0
 
 # =============================================
-# TEXT OVERLAY FUNCTION (EXISTING)
+# TEXT OVERLAY FUNCTION
 # =============================================
 
 def create_text_overlay(input_video, output_video, text_style):
-    """Add text overlay using ASS subtitles"""
+    """Add text overlay using ASS subtitles with enhanced colors"""
     font_path = get_font_path(text_style.font_family)
     if not font_path:
         print(f"⚠️ Font not found: {text_style.font_family}")
@@ -590,24 +622,17 @@ def create_text_overlay(input_video, output_video, text_style):
     work_dir = os.path.dirname(output_video)
     ass_file = os.path.join(work_dir, "subtitle.ass")
     
-    # Color mapping
-    color_map = {
-        "white": "FFFFFF", "black": "000000", "red": "FF0000",
-        "green": "00FF00", "blue": "0000FF", "yellow": "FFFF00",
-        "gold": "FFD700", "purple": "800080", "cyan": "00FFFF"
-    }
-    
     # Get font color and convert from RRGGBB to BBGGRR for ASS
-    font_color_rgb = color_map.get(text_style.color.lower(), "FFFFFF")
+    font_color_rgb = COLOR_MAP.get(text_style.color.lower(), "FFFFFF")
     # Convert RGB to BGR (swap first two and last two characters)
     font_color_bgr = font_color_rgb[4:6] + font_color_rgb[2:4] + font_color_rgb[0:2]
     print(f"🎨 Font color: {text_style.color} -> RGB={font_color_rgb} -> BGR={font_color_bgr}")
     
     # Parse background color
     bg_parts = text_style.bg_color.split('@')
-    bg_color_name = bg_parts[0]
+    bg_color_name = bg_parts[0].lower()
     bg_opacity = float(bg_parts[1]) if len(bg_parts) > 1 else 0.5
-    bg_color_rgb = color_map.get(bg_color_name.lower(), "000000")
+    bg_color_rgb = COLOR_MAP.get(bg_color_name, "000000")
     # Convert RGB to BGR for background
     bg_color_bgr = bg_color_rgb[4:6] + bg_color_rgb[2:4] + bg_color_rgb[0:2]
     bg_alpha = int((1 - bg_opacity) * 255)
@@ -712,7 +737,6 @@ Dialogue: 0,0:00:00.00,0:00:10.00,Default,,0,0,0,,{text_style.text}"""
     print(f"✅ Video styled successfully with ASS subtitles")
     return True
 
-
 # =============================================
 # DEBUG ENDPOINTS
 # =============================================
@@ -1001,6 +1025,19 @@ async def test_srt_parsing(project_id: str, srt_filename: str):
     except Exception as e:
         return {"error": str(e)}
 
+@app.get("/debug/colors")
+async def list_colors():
+    """List all available colors"""
+    return {
+        "colors": list(COLOR_MAP.keys()),
+        "examples": {
+            "purple_variations": ["purple", "darkpurple", "lightpurple", "violet", "magenta"],
+            "gold_variations": ["gold", "orange", "darkgold"],
+            "basic": ["white", "black", "red", "green", "blue", "yellow"],
+            "others": ["cyan", "pink", "brown", "gray", "navy", "teal", "maroon", "olive", "coral", "lavender", "turquoise", "indigo", "crimson"]
+        }
+    }
+
 # =============================================
 # MAIN API ENDPOINTS
 # =============================================
@@ -1010,6 +1047,7 @@ async def health():
     return {
         "status": "healthy", 
         "fonts_loaded": len(FONTS),
+        "colors_available": len(COLOR_MAP),
         "pil_available": PIL_AVAILABLE,
         "fonttools_available": FONTTOOLS_AVAILABLE,
         "whisper_available": WHISPER_AVAILABLE
@@ -1471,12 +1509,13 @@ async def add_captions_from_srt(request: SrtCaptionRequest):
 async def root():
     return {
         "name": "Text Styling API with Auto Caption",
-        "version": "3.1.0",
+        "version": "3.2.0",
         "features": {
             "title_overlay": "✅",
             "manual_captions": "✅",
             "auto_transcription": "✅" if WHISPER_AVAILABLE else "❌",
-            "srt_support": "✅"
+            "srt_support": "✅",
+            "enhanced_colors": f"✅ ({len(COLOR_MAP)} colors)"
         },
         "endpoints": {
             "style": "POST /api/style - Add title overlay",
@@ -1484,6 +1523,7 @@ async def root():
             "caption_from_srt": "POST /api/caption-from-srt - Add captions from SRT URL",
             "transcribe": "POST /api/transcribe-and-caption - Auto-transcribe from audio",
             "fonts": "GET /fonts - List available fonts",
+            "colors": "GET /debug/colors - List available colors",
             "health": "GET /health - Check status"
         },
         "debug": {
@@ -1491,9 +1531,11 @@ async def root():
             "font_info": "GET /debug/font-info/{font_name}",
             "characters": "GET /debug/characters/{font_name}",
             "test_transcription": "POST /debug/test-transcription?audio_url=...&language=zh",
-            "test_srt": "POST /debug/test-srt - Test SRT parsing"
+            "test_srt": "POST /debug/test-srt - Test SRT parsing",
+            "colors": "GET /debug/colors - List all colors"
         },
-        "fonts_loaded": len(FONTS)
+        "fonts_loaded": len(FONTS),
+        "colors_available": len(COLOR_MAP)
     }
 
 # =============================================