Update app.py

app.py

@@ -15,9 +15,26 @@ api_key = os.getenv("HF_API_KEY")
 image_api_url = os.getenv("IMAGE_API_URL")
 headers = {"Authorization": f"Bearer {api_key}"}
 
+# Danh sách màu sắc và mã màu hex tương ứng
+color_map = {
+    "Đỏ": "#FF0000",
+    "Cam": "#FF7F00",
+    "Vàng": "#FFFF00",
+    "Lục": "#00FF00",
+    "Lam": "#0000FF",
+    "Chàm": "#4B0082",
+    "Tím": "#800080",
+    "Trắng": "#FFFFFF",
+    "Đen": "#000000",
+    "Xám": "#808080",
+    "Cầu vồng": "#FF69B4",  # Ví dụ màu cầu vồng, có thể thay bằng màu sắc khác nếu cần
+}
+
 # Hàm gọi API Hugging Face để tạo hình ảnh
 def query(payload):
     response = requests.post(image_api_url, headers=headers, json=payload)
+    if response.status_code != 200:
+        return f"Error: {response.status_code}, {response.text}"
     return response.content
 
 # Hàm dịch tiếng Việt sang tiếng Anh
@@ -29,8 +46,13 @@ def generate_image(prompt, view, style, hair_color, eye_color, lip_color):
     # Dịch prompt từ tiếng Việt sang tiếng Anh
     prompt_en = translate_to_english(prompt)
     
+    # Lấy mã màu hex tương ứng từ tên màu đã chọn
+    hair_color_hex = color_map.get(hair_color, "#FF0000")
+    eye_color_hex = color_map.get(eye_color, "#FF0000")
+    lip_color_hex = color_map.get(lip_color, "#FF0000")
+    
     # Thêm các yếu tố lựa chọn vào prompt
-    prompt_with_choices = f"{prompt_en}, View: {view}, Style: {style}, Hair color: {hair_color}, Eye color: {eye_color}, Lip color: {lip_color}"
+    prompt_with_choices = f"{prompt_en}, View: {view}, Style: {style}, Hair color: {hair_color_hex}, Eye color: {eye_color_hex}, Lip color: {lip_color_hex}"
 
     # Thêm yếu tố ngẫu nhiên vào prompt
     random_factor = random.randint(1, 1000)
@@ -50,9 +72,18 @@ iface = gr.Interface(
         "text",  # Prompt
         gr.Dropdown(["Cận cảnh", "Nửa người", "Toàn thân"], label="View"),
         gr.Dropdown(["Art", "Anime", "Realistic"], label="Style"),
-        gr.ColorPicker(label="Màu tóc"),
-        gr.ColorPicker(label="Màu mắt"),
-        gr.ColorPicker(label="Màu môi")
+        gr.Dropdown(
+            ["Đỏ", "Cam", "Vàng", "Lục", "Lam", "Chàm", "Tím", "Trắng", "Đen", "Xám", "Cầu vồng"],
+            label="Màu tóc"
+        ),
+        gr.Dropdown(
+            ["Đỏ", "Cam", "Vàng", "Lục", "Lam", "Chàm", "Tím", "Trắng", "Đen", "Xám", "Cầu vồng"],
+            label="Màu mắt"
+        ),
+        gr.Dropdown(
+            ["Đỏ", "Cam", "Vàng", "Lục", "Lam", "Chàm", "Tím", "Trắng", "Đen", "Xám", "Cầu vồng"],
+            label="Màu môi"
+        ),
     ],
     outputs="image",
     title="Image Generator - Trần Như Tuấn",