Update app.py

app.py

@@ -37,7 +37,9 @@ def get_default_font(size=20):
         font_paths = [
             "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
             "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf",
-            "/usr/share/fonts/truetype/freefont/FreeSans.ttf"
+            "/usr/share/fonts/truetype/freefont/FreeSans.ttf",
+            "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
+            "/usr/share/fonts/truetype/noto/NotoSans-Regular.ttf"
         ]
         
         for path in font_paths:
@@ -132,23 +134,51 @@ def process_image(image, target_lang, overlay_opacity=0.7, progress=gr.Progress(
                 translated = translate_text(text, target_lang)
                 translations.append(f"{text} → {translated}")
                 
-                # Draw translation with background
-                x, y = bbox[0][0], max(0, bbox[0][1] - 30)
+                # Extract bounding box coordinates
+                top_left, top_right, bottom_right, bottom_left = bbox
                 
-                # Create semi-transparent background
-                text_size = draw.textbbox((0, 0), translated, font=font)
-                text_width = text_size[2] - text_size[0]
-                text_height = text_size[3] - text_size[1]
+                # Calculate text dimensions and position
+                # Use the original text bounding box size and position
+                x, y = top_left[0], top_left[1]
+                width = top_right[0] - top_left[0]
+                height = bottom_left[1] - top_left[1]
+                
+                # Create a rectangle to cover the original text completely
+                # Add a bit of padding around the original text area
+                padding = 4
                 
-                # Draw background rectangle
-                bg_opacity = int(255 * overlay_opacity)
+                # Create solid background to cover original text
                 draw.rectangle(
-                    [x - 5, y - 5, x + text_width + 5, y + text_height + 5],
-                    fill=(0, 0, 0, bg_opacity)
+                    [
+                        x - padding, 
+                        y - padding, 
+                        x + width + padding, 
+                        y + height + padding
+                    ],
+                    fill=(0, 0, 0, 255)  # Solid black background to cover original text
                 )
                 
-                # Draw text
-                draw.text((x, y), translated, fill=(255, 0, 0), font=font)
+                # Calculate font size to fit within the bounding box
+                # Start with a reasonable default size and adjust if needed
+                fontsize = min(20, int(width / (len(translated) * 0.5))) if translated else 20
+                fontsize = max(fontsize, 12)  # Ensure minimum readability
+                
+                # Get adjusted font
+                adjusted_font = get_default_font(size=fontsize)
+                if adjusted_font is None:
+                    adjusted_font = font
+                
+                # Get text size for centering
+                text_size = draw.textbbox((0, 0), translated, font=adjusted_font)
+                text_width = text_size[2] - text_size[0]
+                text_height = text_size[3] - text_size[1]
+                
+                # Center text in the bounding box
+                text_x = x + (width - text_width) / 2
+                text_y = y + (height - text_height) / 2
+                
+                # Draw text with contrasting color
+                draw.text((text_x, text_y), translated, fill=(255, 0, 0), font=adjusted_font)
         
         # Join all translations
         all_translations = "\n".join(translations)