Update app.py

app.py

@@ -11,9 +11,11 @@ from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 from huggingface_hub import HfApi, hf_hub_download
 from typing import Optional, List, Dict, Any
+import base64
+from io import BytesIO
 
 print("=" * 60)
-print("🚀 TEXT STYLING API - ASS SUBTITLE METHOD")
+print("🚀 TEXT STYLING API - ASS SUBTITLE METHOD WITH DEBUG")
 print("=" * 60)
 
 # =============================================
@@ -46,7 +48,7 @@ def download_all_fonts():
         fonts = {}
         
         for file in files:
-            if file.endswith(('.ttf', '.otf')):
+            if file.endswith(('.ttf', '.otf', '.ttc')):
                 font_name = os.path.basename(file)
                 local_path = os.path.join(FONTS_DIR, font_name)
                 
@@ -66,7 +68,7 @@ def download_all_fonts():
                 fonts[font_key] = {
                     "path": local_path,
                     "name": font_name,
-                    "display_name": font_name.replace('.ttf', '').replace('.otf', '').replace('-', ' ')
+                    "display_name": font_name.replace('.ttf', '').replace('.otf', '').replace('.ttc', '').replace('-', ' ')
                 }
         
         print(f"✅ Loaded {len(fonts)} fonts")
@@ -78,6 +80,25 @@ def download_all_fonts():
 # Download fonts at startup
 FONTS = download_all_fonts()
 
+# =============================================
+# TRY TO IMPORT PIL (for debug endpoints)
+# =============================================
+try:
+    from PIL import Image, ImageFont, ImageDraw
+    PIL_AVAILABLE = True
+    print("✅ PIL available for font testing")
+except ImportError:
+    PIL_AVAILABLE = False
+    print("⚠️ PIL not available - font testing limited")
+
+try:
+    from fontTools import ttLib
+    FONTTOOLS_AVAILABLE = True
+    print("✅ fontTools available for font analysis")
+except ImportError:
+    FONTTOOLS_AVAILABLE = False
+    print("⚠️ fontTools not available - font analysis limited")
+
 # =============================================
 # CREATE FASTAPI APP
 # =============================================
@@ -162,6 +183,7 @@ def create_text_overlay(input_video, output_video, text_style):
         return False
     
     print(f"✅ Using font: {font_path}")
+    print(f"📝 Text to render: {text_style.text}")
     
     # Create working directory for ASS file
     work_dir = os.path.dirname(output_video)
@@ -171,7 +193,8 @@ def create_text_overlay(input_video, output_video, text_style):
     color_map = {
         "white": "FFFFFF", "black": "000000", "red": "FF0000",
         "green": "00FF00", "blue": "0000FF", "yellow": "FFFF00",
-        "gold": "FFD700", "purple": "800080", "cyan": "00FFFF"
+        "gold": "FFD700", "purple": "800080", "cyan": "00FFFF",
+        "pink": "FFC0CB", "orange": "FFA500", "brown": "A52A2A"
     }
     
     # Get font color
@@ -248,12 +271,174 @@ Dialogue: 0,0:00:00.00,0:00:10.00,Default,,0,0,0,,{text_style.text}"""
     return True
 
 # =============================================
-# API ENDPOINTS
+# DEBUG ENDPOINTS
+# =============================================
+
+@app.get("/debug/characters/{font_name}")
+async def debug_characters(font_name: str):
+    """Test if font supports specific Chinese characters"""
+    if not PIL_AVAILABLE:
+        return {"error": "PIL not installed - cannot test characters"}
+    
+    try:
+        font_path = None
+        for key, font_info in FONTS.items():
+            if font_name in key or font_name in font_info["name"]:
+                font_path = font_info["path"]
+                break
+        
+        if not font_path:
+            return {"error": f"Font {font_name} not found"}
+        
+        # Test different character sets
+        test_texts = [
+            ("English", "ABCDEFGHIJKLMNOPQRSTUVWXYZ"),
+            ("Numbers", "0123456789"),
+            ("Common Chinese", "的一是在不了有和人这中大为上个国我以要他时来用们生到作地于出就分对成会可主发年动同工也能下过子说产种面而方后多定行学法所民得经十三之进等着部度家电力里如水化高自二理起小物现实加量都两体制机当使点从业本去把性好应开它合还因由其些然前外天政四日那社义事平形相全表间样与关各重新线内数正心反你明看原又么利比或但质气第向道命此变条只没结解问意建月公无系军很情者最立代想已通并提直题党程展五果料象员革位入常文总次品式活设及管特件长求老头基资边流路级少图山统接知较将组见计别她手角期根论运农指几九区强放决西被干做必战先回则任取据处队南给色光门即保治北造百规热领七海口东导器压志世金增争济阶油思术极交受联什认六共权收证改清己美再采转更单风切打白教速花带安场身车例真务具万每目至达走积示议声报斗完类八离华名确才科张信马节话米整空元况今集温传土许步群广石记需段研界拉林律叫且究观越织装影算低持音众书布复容儿须际商非验连断深难近矿千周委素技备半办青省列习响约支般史感劳便团往酸历市克何除消构府称太准精值号率族维划选标写存候毛亲快效斯院查江型眼王按格养易置派层片始却专状育厂京识适属圆包火住调满县局照参红细引听该铁价严龙飞"),
+            ("Test Phrase", "荆南麦圆体测试"),
+            ("Your Text", font_name)
+        ]
+        
+        images = []
+        for label, text in test_texts:
+            # Create image
+            img = Image.new('RGB', (1200, 400), color='white')
+            d = ImageDraw.Draw(img)
+            
+            try:
+                # Try to load font with size 24
+                font = ImageFont.truetype(font_path, 24)
+                # Draw label
+                d.text((10, 10), f"{label}:", fill='black', font=font)
+                # Draw text (wrap if too long)
+                d.text((10, 50), str(text)[:200], fill='black', font=font)
+            except Exception as e:
+                d.text((10, 10), f"Error: {str(e)}", fill='red', font=ImageFont.load_default())
+            
+            # Convert to base64
+            buffered = BytesIO()
+            img.save(buffered, format="PNG")
+            img_base64 = base64.b64encode(buffered.getvalue()).decode()
+            images.append(img_base64)
+        
+        return {
+            "font_name": font_name,
+            "font_path": font_path,
+            "font_size_bytes": os.path.getsize(font_path),
+            "test_images": images,
+            "message": "If Chinese characters appear as boxes, the font lacks those glyphs"
+        }
+        
+    except Exception as e:
+        return {"error": str(e)}
+
+@app.get("/debug/font-info/{font_name}")
+async def font_info(font_name: str):
+    """Get detailed font information"""
+    try:
+        font_path = None
+        for key, font_info in FONTS.items():
+            if font_name in key or font_name in font_info["name"]:
+                font_path = font_info["path"]
+                break
+        
+        if not font_path:
+            return {"error": f"Font {font_name} not found"}
+        
+        info = {
+            "font_name": font_name,
+            "path": font_path,
+            "size_bytes": os.path.getsize(font_path),
+            "pil_available": PIL_AVAILABLE,
+            "fonttools_available": FONTTOOLS_AVAILABLE
+        }
+        
+        # Try to get basic info with PIL
+        if PIL_AVAILABLE:
+            try:
+                font = ImageFont.truetype(font_path, 20)
+                info["pil_loads"] = True
+            except Exception as e:
+                info["pil_loads"] = False
+                info["pil_error"] = str(e)
+        
+        # Try to get detailed info with fontTools
+        if FONTTOOLS_AVAILABLE:
+            try:
+                from fontTools import ttLib
+                font = ttLib.TTFont(font_path)
+                
+                # Get font name
+                name_records = {}
+                for record in font['name'].names:
+                    try:
+                        if record.nameID == 1:
+                            name_records['family'] = record.string.decode('utf-16-be')
+                        elif record.nameID == 2:
+                            name_records['subfamily'] = record.string.decode('utf-16-be')
+                        elif record.nameID == 4:
+                            name_records['full'] = record.string.decode('utf-16-be')
+                        elif record.nameID == 6:
+                            name_records['postscript'] = record.string.decode('utf-16-be')
+                    except:
+                        pass
+                
+                info["names"] = name_records
+                
+                # Check for Chinese glyphs
+                cmap = font.getBestCmap()
+                chinese_ranges = [
+                    (0x4E00, 0x9FFF),  # CJK Unified Ideographs
+                    (0x3400, 0x4DBF),  # Extension A
+                    (0x20000, 0x2A6DF), # Extension B
+                ]
+                
+                has_chinese = False
+                chinese_count = 0
+                for start, end in chinese_ranges:
+                    for code in range(start, min(start+1000, end), 100):
+                        if code in cmap:
+                            has_chinese = True
+                            chinese_count += 1
+                
+                info["has_chinese_glyphs"] = has_chinese
+                info["approx_chinese_glyphs"] = chinese_count * 100
+                info["total_glyphs"] = len(font.getGlyphOrder())
+                
+            except Exception as e:
+                info["fonttools_error"] = str(e)
+        
+        return info
+        
+    except Exception as e:
+        return {"error": str(e)}
+
+@app.get("/debug/font-list")
+async def debug_font_list():
+    """List all fonts with basic info"""
+    font_list = []
+    for key, font_info in FONTS.items():
+        font_list.append({
+            "id": key,
+            "name": font_info["display_name"],
+            "file": font_info["name"],
+            "path": font_info["path"],
+            "size_kb": round(os.path.getsize(font_info["path"]) / 1024, 1)
+        })
+    return {"fonts": font_list}
+
+# =============================================
+# MAIN API ENDPOINTS
 # =============================================
 
 @app.get("/health")
 async def health():
-    return {"status": "healthy", "fonts_loaded": len(FONTS)}
+    return {
+        "status": "healthy", 
+        "fonts_loaded": len(FONTS),
+        "pil_available": PIL_AVAILABLE,
+        "fonttools_available": FONTTOOLS_AVAILABLE
+    }
 
 @app.get("/fonts")
 async def list_fonts():
@@ -278,6 +463,12 @@ async def style_video(request: StylingRequest):
             titled_path = os.path.join(work_dir, "titled.mp4")
             if create_text_overlay(current_video, titled_path, request.title_overlay):
                 current_video = titled_path
+            else:
+                return StylingResponse(
+                    status="error",
+                    project_id=request.project_id,
+                    error="Failed to add text overlay"
+                )
         
         # Upload styled video
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -312,11 +503,16 @@ async def style_video(request: StylingRequest):
 async def root():
     return {
         "name": "Text Styling API",
-        "version": "1.0.0",
+        "version": "2.0.0",
         "endpoints": {
             "style": "POST /api/style",
             "fonts": "GET /fonts",
-            "health": "GET /health"
+            "health": "GET /health",
+            "debug": {
+                "font_list": "GET /debug/font-list",
+                "font_info": "GET /debug/font-info/{font_name}",
+                "characters": "GET /debug/characters/{font_name}"
+            }
         },
         "fonts_loaded": len(FONTS)
     }