Update text_integration.py

text_integration.py

@@ -2,7 +2,7 @@ from PIL import Image, ImageDraw, ImageFont
 import gradio as gr
 import numpy as np
 
-def add_text_to_image(image, text, text_x, text_y, font_size="90", font_family="Arial", font_color="Weiß"):
+def add_text_to_image(image, text, text_x, text_y, font_size="90", font_family="Arial", font_color="#FFFFFF"):
     """Fügt Text mit PIL zum Bild hinzu mit Formatierungsoptionen"""
     if image is None or not text:
         return image
@@ -15,18 +15,12 @@ def add_text_to_image(image, text, text_x, text_y, font_size="90", font_family="
         image_with_text = image.copy()
         draw = ImageDraw.Draw(image_with_text)
         
-        # Schriftgröße direkt verwenden (kein Mapping)
+        # Schriftgröße anpassen - direkte Verwendung der ausgewählten Größe
+        # Für 512px Bilder sind diese Größen angemessen
         try:
-            pixel_size = int(font_size)
+            pixel_size = int(font_size)  # Direkt die ausgewählte Größe verwenden
         except:
-            pixel_size = 90
-        
-        # Farbnamen in Hex-Codes umwandeln
-        color_mapping = {
-            "Weiß": "#FFFFFF",
-            "Schwarz": "#000000"
-        }
-        hex_color = color_mapping.get(font_color, "#FFFFFF")
+            pixel_size = 90  # Fallback
         
         # Schriftart basierend auf Auswahl
         font_paths = {
@@ -38,7 +32,7 @@ def add_text_to_image(image, text, text_x, text_y, font_size="90", font_family="
         try:
             font_path = font_paths.get(font_family, "arial.ttf")
             font = ImageFont.truetype(font_path, pixel_size)
-            print(f"Schriftart geladen: {font_path} mit Größe {pixel_size}px")
+            print(f"Schriftart geladen: {font_path} mit Größe {pixel_size}")
         except Exception as font_error:
             print(f"Fehler beim Laden der Schriftart: {font_error}")
             # Fallback auf Default Font
@@ -48,15 +42,15 @@ def add_text_to_image(image, text, text_x, text_y, font_size="90", font_family="
                 font = ImageFont.load_default()
                 print("Fallback auf Default Font")
         
-        draw.text((text_x, text_y), text, fill=hex_color, font=font)
-        print(f"Text '{text}' an Position ({text_x}, {text_y}) hinzugefügt - Größe: {pixel_size}px, Schrift: {font_family}, Farbe: {font_color} ({hex_color})")
+        draw.text((text_x, text_y), text, fill=font_color, font=font)
+        print(f"Text '{text}' an Position ({text_x}, {text_y}) hinzugefügt - Größe: {pixel_size}px, Schrift: {font_family}, Farbe: {font_color}")
         return image_with_text
         
     except Exception as e:
         print(f"Fehler beim Text hinzufügen: {e}")
         return image
 
-def create_text_preview(image, text, text_x, text_y, font_size="90", font_family="Arial", font_color="Weiß"):
+def create_text_preview(image, text, text_x, text_y, font_size="90", font_family="Arial", font_color="#FFFFFF"):
     """Erstellt eine Vorschau mit Text und rotem Marker - nur EINEN Punkt"""
     if image is None:
         return image
@@ -84,19 +78,12 @@ def create_text_preview(image, text, text_x, text_y, font_size="90", font_family
         
         # Text zeichnen wenn vorhanden
         if text:
-            # Schriftgröße direkt verwenden (kein Mapping)
+            # Schriftgröße direkt verwenden
             try:
                 pixel_size = int(font_size)
             except:
                 pixel_size = 90
             
-            # Farbnamen in Hex-Codes umwandeln
-            color_mapping = {
-                "Weiß": "#FFFFFF", 
-                "Schwarz": "#000000"
-            }
-            hex_color = color_mapping.get(font_color, "#FFFFFF")
-            
             # Schriftart basierend auf Auswahl
             font_paths = {
                 "Arial": "arial.ttf",
@@ -115,8 +102,8 @@ def create_text_preview(image, text, text_x, text_y, font_size="90", font_family
                 except:
                     font = ImageFont.load_default()
             
-            draw.text((text_x, text_y), text, fill=hex_color, font=font)
-            print(f"Vorschau: Text mit Größe {pixel_size}px, Farbe {font_color} ({hex_color}) gezeichnet")
+            draw.text((text_x, text_y), text, fill=font_color, font=font)
+            print(f"Vorschau: Text mit Größe {pixel_size}px gezeichnet")
         
         return preview
         
@@ -135,7 +122,7 @@ def update_text_preview_i2i(original_image, generated_image, text, text_x, text_
 
 def update_text_preview_t2i(image, text, text_x, text_y, font_size, font_family, font_color):
     """Aktualisiert die Text-Vorschau für Text-zu-Bild mit Formatierung"""
-    print(f"Update Vorschau - Größe: {font_size}px, Schrift: {font_family}, Farbe: {font_color}")
+    print(f"Update Vorschau - Größe: {font_size}, Schrift: {font_family}, Farbe: {font_color}")
     return create_text_preview(image, text, text_x, text_y, font_size, font_family, font_color)
 
 def capture_click(event: gr.SelectData):