Update app.py

app.py

@@ -9,9 +9,9 @@ from typing import Any
 # 常量定义
 HTTP_STATUS_CENSORED = 451
 HTTP_STATUS_OK = 200
-MAX_SEED = 2147483647
+MAX_SEED = 2147483647 # (2**31 - 1)
 MAX_IMAGE_SIZE = 2048
-MIN_IMAGE_SIZE = 256
+MIN_IMAGE_SIZE = 256 # Smallest dimension for SDXL like models often 512, but API might support smaller. Adjusted to API's limits.
 
 # 调试模式
 DEBUG_MODE = os.environ.get("DEBUG_MODE", "false").lower() == "true"
@@ -24,9 +24,8 @@ MODEL_CONFIGS = {
 
 def validate_dimensions(width: int, height: int) -> tuple[int, int]:
     """验证并调整图片尺寸"""
-    # 确保尺寸是32的倍数
-    width = max(MIN_IMAGE_SIZE, min(width, MAX_IMAGE_SIZE))
-    height = max(MIN_IMAGE_SIZE, min(height, MAX_IMAGE_SIZE))
+    width = max(MIN_IMAGE_SIZE, min(int(width), MAX_IMAGE_SIZE))
+    height = max(MIN_IMAGE_SIZE, min(int(height), MAX_IMAGE_SIZE))
     width = (width // 32) * 32
     height = (height // 32) * 32
     return width, height
@@ -45,108 +44,61 @@ class ImageGenerationConfig:
     width: int = 1024
     height: int = 1024
     seed: int | None = None
-    use_polish: bool = False
+    use_polish: bool = False # This wasn't exposed in UI, assuming false
     is_lumina: bool = True
     lumina_config: LuminaConfig = field(default_factory=LuminaConfig)
 
 class ImageClient:
     """图像生成客户端"""
     def __init__(self) -> None:
-        # 使用环境变量中的API_TOKEN
         self.x_token = os.environ.get("API_TOKEN", "")
         if not self.x_token:
             raise ValueError("环境变量中未设置API_TOKEN")
 
-        # API端点
         self.lumina_api_url = "https://ops.api.talesofai.cn/v3/make_image"
         self.lumina_task_status_url = "https://ops.api.talesofai.cn/v1/artifact/task/{task_uuid}"
-
-        # 轮询配置（增加超时时间以应对服务器负载）
-        self.max_polling_attempts = 100  # 增加到100次
-        self.polling_interval = 3.0      # 保持3秒间隔
-        # 总超时时间：100 × 3.0 = 300秒 = 5分钟
-
-        # 默认请求头
+        self.max_polling_attempts = 100
+        self.polling_interval = 3.0
         self.default_headers = {
             "Content-Type": "application/json",
-            "x-platform": "nieta-app/web",
+            "x-platform": "nieta-app/web", # Or a generic identifier if preferred
             "X-Token": self.x_token,
         }
 
     def _prepare_prompt_data(self, prompt: str, negative_prompt: str = "") -> list[dict[str, Any]]:
-        """准备提示词数据"""
-        prompts = [
-            {
-                "type": "freetext",
-                "value": prompt,
-                "weight": 1.0
-            }
-        ]
-        
+        prompts_data = [{"type": "freetext", "value": prompt, "weight": 1.0}]
         if negative_prompt:
-            prompts.append({
-                "type": "freetext",
-                "value": negative_prompt,
-                "weight": -1.0
-            })
-
-        # 添加Lumina元素
-        prompts.append({
-            "type": "elementum",
-            "value": "b5edccfe-46a2-4a14-a8ff-f4d430343805",
-            "uuid": "b5edccfe-46a2-4a14-a8ff-f4d430343805",
-            "weight": 1.0,
-            "name": "lumina1",
+            prompts_data.append({"type": "freetext", "value": negative_prompt, "weight": -1.0})
+        prompts_data.append({
+            "type": "elementum", "value": "b5edccfe-46a2-4a14-a8ff-f4d430343805",
+            "uuid": "b5edccfe-46a2-4a14-a8ff-f4d430343805", "weight": 1.0, "name": "lumina1",
             "img_url": "https://oss.talesofai.cn/picture_s/1y7f53e6itfn_0.jpeg",
-            "domain": "",
-            "parent": "",
-            "label": None,
-            "sort_index": 0,
-            "status": "IN_USE",
-            "polymorphi_values": {},
-            "sub_type": None,
+            "domain": "", "parent": "", "label": None, "sort_index": 0, "status": "IN_USE",
+            "polymorphi_values": {}, "sub_type": None,
         })
-
-        return prompts
+        return prompts_data
 
     def _build_payload(self, config: ImageGenerationConfig) -> dict[str, Any]:
-        """构建API请求载荷"""
         payload = {
-            "storyId": "",
-            "jobType": "universal",
-            "width": config.width,
-            "height": config.height,
-            "rawPrompt": config.prompts,
-            "seed": config.seed,
-            "meta": {"entrance": "PICTURE,PURE"},
-            "context_model_series": None,
-            "negative_freetext": "",
+            "storyId": "", "jobType": "universal", "width": config.width, "height": config.height,
+            "rawPrompt": config.prompts, "seed": config.seed, "meta": {"entrance": "PICTURE,PURE"},
+            "context_model_series": None, "negative_freetext": "", # Negative handled in rawPrompt
             "advanced_translator": config.use_polish,
         }
-
         if config.is_lumina:
             client_args = {}
-            if config.lumina_config.model_name:
-                client_args["ckpt_name"] = config.lumina_config.model_name
-            if config.lumina_config.cfg is not None:
-                client_args["cfg"] = str(config.lumina_config.cfg)
-            if config.lumina_config.step is not None:
-                client_args["steps"] = str(config.lumina_config.step)
-
-            if client_args:
-                payload["client_args"] = client_args
-
+            if config.lumina_config.model_name: client_args["ckpt_name"] = config.lumina_config.model_name
+            if config.lumina_config.cfg is not None: client_args["cfg"] = str(config.lumina_config.cfg)
+            if config.lumina_config.step is not None: client_args["steps"] = str(config.lumina_config.step)
+            if client_args: payload["client_args"] = client_args
         return payload
 
     async def _poll_task_status(self, task_uuid: str) -> dict[str, Any]:
-        """轮询任务状态"""
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            for _ in range(self.max_polling_attempts):
-                response = await client.get(
-                    self.lumina_task_status_url.format(task_uuid=task_uuid),
-                    headers=self.default_headers
-                )
-                
+        status_url = self.lumina_task_status_url.format(task_uuid=task_uuid)
+        async with httpx.AsyncClient(timeout=30.0) as client: # Timeout for individual poll request
+            for attempt in range(self.max_polling_attempts):
+                response = await client.get(status_url, headers=self.default_headers)
+
                 if response.status_code != HTTP_STATUS_OK:
                     return {
                         "success": False,
@@ -161,12 +113,10 @@ class ImageClient:
                         "success": False,
                         "error": f"任务状态响应解析失败: {response.text[:500]}"
                     }
-                
-                # 使用正确的字段名（根据model_studio的实现）
+
                 task_status = result.get("task_status")
-                
+
                 if task_status == "SUCCESS":
-                    # 从artifacts数组中获取图片URL
                     artifacts = result.get("artifacts", [])
                     if artifacts and len(artifacts) > 0:
                         image_url = artifacts[0].get("url")
@@ -196,7 +146,6 @@ class ImageClient:
                     }
 
                 await asyncio.sleep(self.polling_interval)
-
         return {
             "success": False,
             "error": "⏳ 生图任务超时（5分钟），服务器可能正在处理大量请求，请稍后重试"
@@ -207,7 +156,7 @@ class ImageClient:
         try:
             # 获取模型路径
             model_path = MODEL_CONFIGS.get(model_name, MODEL_CONFIGS["ep3"])
-            
+
             # 准备配置
             config = ImageGenerationConfig(
                 prompts=self._prepare_prompt_data(prompt, negative_prompt),
@@ -221,41 +170,40 @@ class ImageClient:
                     step=steps
                 )
             )
-
             # 发送生成请求
             async with httpx.AsyncClient(timeout=300.0) as client:
                 payload = self._build_payload(config)
                 if DEBUG_MODE:
                     print(f"DEBUG: 发送API请求到 {self.lumina_api_url}")
                     print(f"DEBUG: 请求载荷: {payload}")
-                
+
                 response = await client.post(
                     self.lumina_api_url,
                     json=payload,
                     headers=self.default_headers
                 )
-                
+
                 if DEBUG_MODE:
                     print(f"DEBUG: API响应状态码: {response.status_code}")
                     print(f"DEBUG: API响应内容: {response.text[:1000]}")
 
                 if response.status_code == HTTP_STATUS_CENSORED:
                     return None, "内容不合规"
-                
+
                 # 处理并发限制错误
                 if response.status_code == 433:
                     return None, "⏳ 服务器正忙，同时生成的图片数量已达上限，请稍后重试"
-                
+
                 if response.status_code != HTTP_STATUS_OK:
                     return None, f"API请求失败: {response.status_code} - {response.text}"
 
                 # API直接返回UUID字符串（根据model_studio的实现）
                 content = response.text.strip()
                 task_uuid = content.replace('"', "")
-                
+
                 if DEBUG_MODE:
                     print(f"DEBUG: API返回UUID: {task_uuid}")
-                
+
                 if not task_uuid:
                     return None, f"未获取到任务ID，API响应: {response.text}"
 
@@ -265,388 +213,152 @@ class ImageClient:
                     return result["image_url"], None
                 else:
                     return None, result["error"]
-
         except Exception as e:
             return None, f"生成图片时发生错误: {str(e)}"
 
 # 创建图片生成客户端实例
-image_client = ImageClient()
-# 简短的示例标题（用于显示）
+try:
+    image_client = ImageClient()
+except Exception as e:
+    print(f"Failed to initialize ImageClient: {e}")
+    image_client = None
+
+# Example prompts
 example_titles = [
     "A stylized female demon with red hair and glitch effects",
-    "A young man relaxes on a hazy urban rooftop", 
+    "A young man relaxes on a hazy urban rooftop",
     "A gentle, freckled girl embraces a goat in a meadow"
 ]
-
-# 完整提示词映射
 full_prompts = {
-    "A stylized female demon with red hair and glitch effects": "Stylized anime illustration of a female demon or supernatural character with vibrant red hair in twintails/pigtails and glowing purple eyes. Character has black horns and features bandage-like cross markings on face. Subject wears a black sleeveless top and holds a pink bubblegum or candy sphere near mouth. Digital glitch effects create pixelated elements in her hair and around background. Dramatic lighting with stark white/black contrasting background featuring cracks or lightning patterns. Character has gold/yellow accessories including bracelets and hair decorations. Modern anime art style with sharp contrast and vivid colors. Portrait composition showing three-quarter view of character with confident or playful expression. Color palette dominated by reds, blacks, whites, purple and pink accents. Surreal or otherworldly atmosphere enhanced by particle effects and lighting. Professional digital illustration combining traditional anime aesthetics with contemporary glitch art elements. Character design suggests edgy or alternative styling with possible cyberpunk or modern demon girl influences.",
-    "A young man relaxes on a hazy urban rooftop": "Atmospheric anime illustration of young man with messy brown hair on urban rooftop during overcast day. Character wears white dress shirt and dark trousers, leaning back against railing while holding canned drink. Scene set on building rooftop with industrial elements like water tower, power lines, and metal structures visible. Cityscape background shows apartment buildings and urban architecture through soft hazy lighting. Subject has relaxed pose suggesting brief break or moment of contemplation. Color palette uses muted whites, grays, and industrial tones creating realistic urban atmosphere. Art style combines detailed architectural elements with soft, painterly technique. Composition emphasizes vertical lines of city buildings and metal structures. Professional digital artwork capturing slice-of-life moment in urban setting. Scene suggests peaceful solitude amid busy city environment. Lighting creates gentle, overcast mood with subtle shadows and highlights. Character design and setting reflect contemporary Japanese salary-man or office worker aesthetic.",
-    "A gentle, freckled girl embraces a goat in a meadow": "Enchanting anime illustration of a gentle, freckled girl with long, wavy orange hair and elegant ram horns, tenderly embracing a white baby goat in a sunlit meadow. The composition is a close-up, focusing on the upper body and faces of both the girl and the goat, capturing an intimate and heartwarming moment. She wears a vintage-inspired dress with a high collar, puffed sleeves, and a delicate white headband, adorned with golden ribbons and lace details. The sunlight bathes the scene in warm, golden tones, casting soft shadows and creating a dreamy, pastoral atmosphere. The background is filled with lush green grass and scattered white flowers, enhancing the idyllic countryside setting. The art style is painterly and vibrant, with expressive brushwork and a focus on light and texture, evoking a sense of peace, innocence, and connection with nature."
-}
-
-css = """
-.main-container {
-    max-width: 1400px !important;
-    margin: 0 auto !important;
-    padding: 20px !important;
-}
-
-.left-panel {
-    background: linear-gradient(145deg, #f8f9fa, #e9ecef) !important;
-    border-radius: 16px !important;
-    padding: 24px !important;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.1) !important;
-    border: 1px solid rgba(255,255,255,0.2) !important;
+    example_titles[0]: "Stylized anime illustration of a female demon or supernatural character with vibrant red hair in twintails/pigtails and glowing purple eyes. Character has black horns and features bandage-like cross markings on face. Subject wears a black sleeveless top and holds a pink bubblegum or candy sphere near mouth. Digital glitch effects create pixelated elements in her hair and around background. Dramatic lighting with stark white/black contrasting background featuring cracks or lightning patterns. Character has gold/yellow accessories including bracelets and hair decorations. Modern anime art style with sharp contrast and vivid colors. Portrait composition showing three-quarter view of character with confident or playful expression. Color palette dominated by reds, blacks, whites, purple and pink accents. Surreal or otherworldly atmosphere enhanced by particle effects and lighting. Professional digital illustration combining traditional anime aesthetics with contemporary glitch art elements. Character design suggests edgy or alternative styling with possible cyberpunk or modern demon girl influences.",
+    example_titles[1]: "Atmospheric anime illustration of young man with messy brown hair on urban rooftop during overcast day. Character wears white dress shirt and dark trousers, leaning back against railing while holding canned drink. Scene set on building rooftop with industrial elements like water tower, power lines, and metal structures visible. Cityscape background shows apartment buildings and urban architecture through soft hazy lighting. Subject has relaxed pose suggesting brief break or moment of contemplation. Color palette uses muted whites, grays, and industrial tones creating realistic urban atmosphere. Art style combines detailed architectural elements with soft, painterly technique. Composition emphasizes vertical lines of city buildings and metal structures. Professional digital artwork capturing slice-of-life moment in urban setting. Scene suggests peaceful solitude amid busy city environment. Lighting creates gentle, overcast mood with subtle shadows and highlights. Character design and setting reflect contemporary Japanese salary-man or office worker aesthetic.",
+    example_titles[2]: "Enchanting anime illustration of a gentle, freckled girl with long, wavy orange hair and elegant ram horns, tenderly embracing a white baby goat in a sunlit meadow. The composition is a close-up, focusing on the upper body and faces of both the girl and the goat, capturing an intimate and heartwarming moment. She wears a vintage-inspired dress with a high collar, puffed sleeves, and a delicate white headband, adorned with golden ribbons and lace details. The sunlight bathes the scene in warm, golden tones, casting soft shadows and creating a dreamy, pastoral atmosphere. The background is filled with lush green grass and scattered white flowers, enhancing the idyllic countryside setting. The art style is painterly and vibrant, with expressive brushwork and a focus on light and texture, evoking a sense of peace, innocence, and connection with nature."
 }
 
-.right-panel {
-    background: linear-gradient(145deg, #ffffff, #f8f9fa) !important;
-    border-radius: 16px !important;
-    padding: 24px !important;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.1) !important;
-    border: 1px solid rgba(255,255,255,0.2) !important;
-    display: flex !important;
-    flex-direction: column !important;
-    align-items: center !important;
-    justify-content: flex-start !important;
-    min-height: 600px !important;
-}
-
-#main-prompt textarea {
-    min-height: 180px !important;
-    font-size: 15px !important;
-    line-height: 1.6 !important;
-    padding: 16px !important;
-    border-radius: 12px !important;
-    border: 2px solid #e9ecef !important;
-    transition: all 0.3s ease !important;
-    background: white !important;
-}
-
-#main-prompt textarea:focus {
-    border-color: #4f46e5 !important;
-    box-shadow: 0 0 0 3px rgba(79, 70, 229, 0.1) !important;
-}
-
-.run-button {
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-    border: none !important;
-    border-radius: 12px !important;
-    padding: 12px 32px !important;
-    font-weight: 600 !important;
-    font-size: 16px !important;
-    color: white !important;
-    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4) !important;
-    transition: all 0.3s ease !important;
-}
-
-.run-button:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.6) !important;
-}
-
-.settings-section {
-    background: white !important;
-    border-radius: 12px !important;
-    padding: 16px !important;
-    margin-top: 12px !important;
-    box-shadow: 0 2px 8px rgba(0,0,0,0.05) !important;
-    border: 1px solid #e9ecef !important;
-}
-
-.settings-title {
-    font-size: 18px !important;
-    font-weight: 600 !important;
-    color: #374151 !important;
-    margin-bottom: 12px !important;
-    padding-bottom: 6px !important;
-    border-bottom: 2px solid #e9ecef !important;
-}
-
-.right-panel .settings-title {
-    margin-bottom: 8px !important;
-    padding-bottom: 4px !important;
-    font-size: 16px !important;
-}
-
-.right-panel .block {
-    min-height: unset !important;
-    height: auto !important;
-    flex: none !important;
-}
-
-.right-panel .html-container {
-    padding: 0 !important;
-    margin: 0 !important;
-}
-
-.slider-container .wrap {
-    background: #f8f9fa !important;
-    border-radius: 8px !important;
-    padding: 6px !important;
-    margin: 2px 0 !important;
-}
-
-.settings-section .block {
-    margin: 4px 0 !important;
-}
-
-.settings-section .row {
-    margin: 6px 0 !important;
-}
-
-.settings-section .form {
-    gap: 4px !important;
-}
-
-.settings-section .html-container {
-    padding: 0 !important;
-    margin: 8px 0 4px 0 !important;
-}
-
-.result-image {
-    border-radius: 16px !important;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.15) !important;
-    max-width: 100% !important;
-    height: auto !important;
-    min-height: 400px !important;
-    width: 100% !important;
-}
-
-.result-image img {
-    border-radius: 16px !important;
-    object-fit: contain !important;
-    max-width: 100% !important;
-    max-height: 600px !important;
-    width: auto !important;
-    height: auto !important;
-}
-
-.examples-container {
-    margin-top: 20px !important;
-}
-
-/* 示例按钮样式 - 每个按钮单独一行 */
-.examples-container button {
-    font-size: 13px !important;
-    padding: 10px 16px !important;
-    border-radius: 10px !important;
-    background: #f8f9fa !important;
-    border: 1px solid #e9ecef !important;
-    color: #495057 !important;
-    transition: all 0.2s ease !important;
-    width: 100% !important;
-    margin-bottom: 8px !important;
-    text-align: left !important;
-    font-weight: 500 !important;
-}
-
-.examples-container button:hover {
-    background: #e9ecef !important;
-    border-color: #6c757d !important;
-    transform: translateY(-1px) !important;
-    box-shadow: 0 2px 8px rgba(0,0,0,0.1) !important;
-}
-
-.examples-container button:active {
-    transform: translateY(0px) !important;
-    background: #dee2e6 !important;
-}
-
-
-
-.title-header {
-    text-align: center !important;
-    margin-bottom: 30px !important;
-    padding: 20px !important;
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-    color: white !important;
-    border-radius: 16px !important;
-    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3) !important;
-}
-
-.title-header h1 {
-    font-size: 28px !important;
-    font-weight: 700 !important;
-    margin: 0 !important;
-    text-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
-}
-"""
-
-
-
 async def infer(
-    prompt,
-    seed,
-    randomize_seed,
-    width,
-    height,
-    cfg,
-    steps,
-    model_name,
-    progress=gr.Progress(track_tqdm=True),
+    prompt_text, seed_val, randomize_seed_val, width_val, height_val,
+    cfg_val, steps_val, model_name_val, progress=gr.Progress(track_tqdm=True)
 ):
-    if not prompt.strip():
+    if image_client is None:
+        raise gr.Error("ImageClient 未正确初始化。请检查应用日志和API_TOKEN配置。")
+    if not prompt_text.strip():
         raise gr.Error("提示词不能为空")
 
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
+    current_seed = int(seed_val)
+    if randomize_seed_val:
+        current_seed = random.randint(0, MAX_SEED)
 
     # 验证并调整尺寸
-    width, height = validate_dimensions(width, height)
+    width_val, height_val = validate_dimensions(width_val, height_val)
 
     # 验证其他参数
-    if not 1.0 <= cfg <= 20.0:
+    if not (1.0 <= float(cfg_val) <= 20.0):
         raise gr.Error("CFG Scale 必须在 1.0 到 20.0 之间")
-    if not 1 <= steps <= 50:
+    if not (1 <= int(steps_val) <= 50):
         raise gr.Error("Steps 必须在 1 到 50 之间")
 
     image_url, error = await image_client.generate_image(
-        prompt=prompt,
+        prompt=prompt_text,
         negative_prompt="",
-        seed=seed,
-        width=width,
-        height=height,
-        cfg=cfg,
-        steps=steps,
-        model_name=model_name
+        seed=current_seed,
+        width=width_val,
+        height=height_val,
+        cfg=float(cfg_val),
+        steps=int(steps_val),
+        model_name=model_name_val
     )
 
     if error:
         raise gr.Error(error)
 
-    return image_url, seed
-
-with gr.Blocks(css=css) as demo:
-    with gr.Column(elem_classes=["main-container"]):
-        # 标题区域
-        with gr.Row(elem_classes=["title-header"]):
-            gr.HTML("<h1>🎨 Lumina Text-to-Image Playground</h1>")
-        
-        # 主体左右布局
-        with gr.Row(equal_height=True):
-            # 左侧参数设置面板
-            with gr.Column(scale=1, elem_classes=["left-panel"]):
-                gr.HTML("<div class='settings-title'>✨ Generation Settings</div>")
-                
-                # 主要提示词输入
-                prompt = gr.Text(
-                    label="Prompt",
-                    show_label=True,
-                    max_lines=10,
-                    lines=8,
-                    placeholder="Describe what you want to generate in detail...",
-                    elem_id="main-prompt",
-                )
-                
-                run_button = gr.Button("🚀 Generate Image", elem_classes=["run-button"], variant="primary")
-                
-                # 参数设置区域
-                with gr.Column(elem_classes=["settings-section"]):
-                    gr.HTML("<div class='settings-title'>🎛️ Parameters</div>")
-                    
-                    # 种子设置
-                    with gr.Row():
-                        seed = gr.Slider(
-                            label="Seed",
-                            minimum=0,
-                            maximum=MAX_SEED,
-                            step=1,
-                            value=0,
-                            elem_classes=["slider-container"]
-                        )
-                        randomize_seed = gr.Checkbox(label="Random Seed", value=True)
-                    
-                    # 尺寸设置
-                    gr.HTML("<div style='margin: 16px 0 8px 0; font-weight: 600; color: #6B7280;'>📐 Image Dimensions</div>")
-                    with gr.Row():
-                        width = gr.Slider(
-                            label="Width",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=1024,
-                            elem_classes=["slider-container"]
-                        )
-                        height = gr.Slider(
-                            label="Height",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=1024,
-                            elem_classes=["slider-container"]
-                        )
-                    
-                    # 生成参数
-                    gr.HTML("<div style='margin: 16px 0 8px 0; font-weight: 600; color: #6B7280;'>⚙️ Generation Parameters</div>")
-                    with gr.Row():
-                        cfg = gr.Slider(
-                            label="CFG Scale",
-                            minimum=1.0,
-                            maximum=20.0,
-                            step=0.1,
-                            value=5.5,
-                            elem_classes=["slider-container"]
-                        )
-                        steps = gr.Slider(
-                            label="Steps",
-                            minimum=1,
-                            maximum=50,
-                            step=1,
-                            value=30,
-                            elem_classes=["slider-container"]
-                        )
-                    
-                    # 模型选择
-                    model_name = gr.Dropdown(
-                        label="🤖 Model Selection",
-                        choices=list(MODEL_CONFIGS.keys()),
-                        value="ep3"
-                    )
-                
-                # 示例提示词
-                with gr.Column(elem_classes=["examples-container"]):
-                    gr.HTML("<div class='settings-title'>💡 Example Prompts</div>")
-                    example_btn1 = gr.Button("A stylized female demon with red hair and glitch effects", size="sm")
-                    example_btn2 = gr.Button("A young man relaxes on a hazy urban rooftop", size="sm") 
-                    example_btn3 = gr.Button("A gentle, freckled girl embraces a goat in a meadow", size="sm")
-            
-            # 右侧图像显示面板
-            with gr.Column(scale=1, elem_classes=["right-panel"]):
-                gr.HTML("<div class='settings-title'>🖼️ Generated Image</div>")
-                result = gr.Image(
-                    label="Result", 
-                    show_label=False, 
-                    elem_classes=["result-image"],
-                    height=600,
-                    container=True,
-                    show_download_button=True,
-                    show_share_button=False
+    return image_url, current_seed
+
+
+# Links for HTML header
+DISCORD_LINK = os.environ.get("DISCORD_LINK", "https://discord.com/invite/AtRtbe9W8w") # Example
+APP_INDEX_LINK = os.environ.get("APP_INDEX_LINK", "https://neta.art/") # Example
+APP_INDEX_ICON = "https://cdn-avatars.huggingface.co/v1/production/uploads/62be651a1e22ec8427aa7096/XQEUF5niIZXQbiOOxn8rQ.jpeg" # Using HF logo
+
+
+with gr.Blocks(theme=gr.themes.Soft(), title="Lumina Image Playground") as demo:
+    gr.Markdown("<h1>🎨 Lumina Text-to-Image Playground | Lumina 文本生图工具</h1>")
+    gr.Markdown("Describe your vision and let the AI bring it to life! Uses an external API for image generation. | 描述您的创意愿景，让AI为您创造精美图像！使用外部API进行图像生成。")
+
+    gr.HTML(f"""
+        <div style="display: flex; justify-content: flex-start; align-items: center; gap: 15px; margin-bottom: 20px; padding: 10px;">
+            <a href="{DISCORD_LINK}" target="_blank" style="text-decoration: none; color: #5865F2; font-weight: 500; display: inline-flex; align-items: center; gap: 5px;">
+                <img src="https://assets-global.website-files.com/6257adef93867e50d84d30e2/636e0a69f118df70ad7828d4_icon_clyde_blurple_RGB.svg" alt="Discord" style="height: 20px;">
+                Join Discord | 加入Discord
+            </a>
+            <a href="{APP_INDEX_LINK}" target="_blank" style="text-decoration: none; color: #333; font-weight: 500; display: inline-flex; align-items: center; gap: 5px;">
+                <img src="{APP_INDEX_ICON}" alt="App Index" style="height: 20px; border-radius: 3px;">
+                Nieta Index | 捏她主页
+            </a>
+        </div>
+    """)
+
+    with gr.Row(variant="panel"):
+        with gr.Column(scale=2): # Controls Panel
+            gr.Markdown("## ⚙️ Generation Controls | 生成控制")
+            prompt = gr.Textbox(
+                label="Prompt | 提示词", lines=5,
+                placeholder="e.g., A majestic dragon soaring through a cyberpunk city skyline, neon lights reflecting off its scales, intricate details. | 例如：一条威武的巨龙翱翔在赛博朋克城市天际线，霓虹灯映照在它的鳞片上，细节精美。",
+                info="Describe the image you want to create. | 描述您想要创建的图像。"
+            )
+
+            with gr.Accordion("🔧 Advanced Settings | 高级设置", open=True):
+                model_name = gr.Dropdown(
+                    label="Model Version | 模型版本", choices=list(MODEL_CONFIGS.keys()), value="ep3",
+                    info="Select the generation model. | 选择生成模型。"
                 )
+                with gr.Row():
+                    cfg = gr.Slider(label="CFG Scale | CFG缩放", minimum=1.0, maximum=20.0, step=0.1, value=5.5, info="Guidance strength. Higher values adhere more to prompt. | 引导强度，更高的值更贴近提示词。")
+                    steps = gr.Slider(label="Sampling Steps | 采样步数", minimum=1, maximum=50, step=1, value=30, info="Number of steps. More steps can improve quality but take longer. | 步数，更多步数可提高质量但耗时更长。")
+
+                with gr.Row():
+                    width = gr.Slider(label="Width | 宽度", minimum=MIN_IMAGE_SIZE, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
+                    height = gr.Slider(label="Height | 高度", minimum=MIN_IMAGE_SIZE, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
+
+                with gr.Row():
+                    seed = gr.Slider(label="Seed | 种子", minimum=0, maximum=MAX_SEED, step=1, value=random.randint(0, MAX_SEED))
+                    randomize_seed = gr.Checkbox(label="Randomize Seed | 随机种子", value=True, info="Use a new random seed for each generation if checked. | 勾选后每次生成使用新的随机种子。")
+
+            run_button = gr.Button("🚀 Generate Image | 生成图像", variant="primary", scale=0)
+
+            with gr.Group():
+                 gr.Markdown("### ✨ Example Prompts | 示例提示词")
+                 for i, title in enumerate(example_titles):
+                     btn = gr.Button(title)
+                     btn.click(lambda t=title: full_prompts[t], outputs=[prompt])
+
+
+        with gr.Column(scale=3): # Output Panel
+            gr.Markdown("## 🖼️ Generated Image | 生成图像")
+            result_image = gr.Image(
+                label="Output Image | 输出图像", show_label=False, type="filepath",
+                height=600, # Max display height
+                show_download_button=True, interactive=False,
+                elem_id="result_image_display" # for potential CSS targeting if needed
+            )
+            generated_seed_info = gr.Textbox(label="Seed Used | 使用的种子", interactive=False, placeholder="The seed for the generated image will appear here. | 生成图像所使用的种子值将显示在此处。")
 
-    gr.on(
-        triggers=[run_button.click, prompt.submit],
-        fn=infer,
-        inputs=[
-            prompt,
-            seed,
-            randomize_seed,
-            width,
-            height,
-            cfg,
-            steps,
-            model_name,
-        ],
-        outputs=[result, seed],
-    )
-    
-    # 示例按钮点击事件
-    example_btn1.click(lambda: full_prompts["A stylized female demon with red hair and glitch effects"], outputs=[prompt])
-    example_btn2.click(lambda: full_prompts["A young man relaxes on a hazy urban rooftop"], outputs=[prompt])
-    example_btn3.click(lambda: full_prompts["A gentle, freckled girl embraces a goat in a meadow"], outputs=[prompt])
-    
+    # Event Handlers
+    inputs_list = [prompt, seed, randomize_seed, width, height, cfg, steps, model_name]
+    outputs_list = [result_image, generated_seed_info]
+
+    run_button.click(fn=infer, inputs=inputs_list, outputs=outputs_list, api_name="generate_image")
+    prompt.submit(fn=infer, inputs=inputs_list, outputs=outputs_list, api_name="generate_image_submit")
 
 
 if __name__ == "__main__":
-    demo.launch()
+    if DEBUG_MODE:
+        print("DEBUG_MODE is enabled.")
+    if not os.environ.get("API_TOKEN"):
+        print("**************************************************************************************")
+        print("WARNING: API_TOKEN environment variable is not set locally.")
+        print("The application will run, but image generation will fail until API_TOKEN is provided.")
+        print("You can set it by running: export API_TOKEN='your_actual_token_here'")
+        print("Or if using a .env file, ensure it's loaded or API_TOKEN is set in your run config.")
+        print("**************************************************************************************")
+
+    demo.launch(debug=DEBUG_MODE, show_error=True)