Update app.py

app.py

@@ -206,19 +206,19 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
     energy_y = 472  
 
     # Capitalize only the first letter of the first word while keeping the rest as is
-    def capitalize_first_word(text):
-        words = text.split()
-        if words:
-            words[0] = words[0].capitalize()
-        return " ".join(words)
-    
-    provider_text = capitalize_first_word(str(model_data['provider']))
-    model_text = capitalize_first_word(str(model_data['model']))
+    def smart_capitalize(text):
+    """Capitalizes the first letter of a string only if it's not already capitalized."""
+    if not text:
+        return text  # Return unchanged if empty
+    return text if text[0].isupper() else text[0].upper() + text[1:]
+
+    # Apply smart capitalization
+    provider_text = smart_capitalize(str(model_data['provider']))
+    model_text = smart_capitalize(str(model_data['model']))
     
     draw.text((title_x, title_y), provider_text, font=title_font, fill="black")
     draw.text((title_x, title_y + 38), model_text, font=title_font, fill="black")
 
-
     details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
     for i, line in enumerate(details_lines):
         bbox = draw.textbbox((0, 0), line, font=details_font)