positive_magic [en, zh] (#3)

- positive_magic [en, zh] (87337ae4c5527b3ac556876f65f0b727452bd9b6)

app.py

@@ -9,6 +9,7 @@ from typing import Union, List, Optional
 import numpy as np
 import time
 import zipfile
+import re
 
 # Description for the app
 DESCRIPTION = """## Qwen Image Hpc/."""
@@ -60,6 +61,16 @@ def get_duration(
     per_image_duration = 5 + num_inference_steps * step_duration
     return 10 + num_images * per_image_duration
 
+# Positive Magic texts
+positive_magic = {
+    "en": "Ultra HD, 4K, cinematic composition.",
+    "zh": "超清，4K，电影级构图"
+}
+
+# Function to check if the prompt contains Chinese characters
+def contains_chinese(text):
+    return bool(re.search(r'[\u4e00-\u9fff]', text))
+
 # Generation function for Qwen/Qwen-Image
 def generate_qwen(
     prompt: str,
@@ -122,9 +133,15 @@ def generate(
     zip_images: bool,
     progress=gr.Progress(track_tqdm=True),
 ):
+    # Enhance the prompt with Positive Magic based on language
+    if contains_chinese(prompt):
+        final_prompt = prompt + " " + positive_magic["zh"]
+    else:
+        final_prompt = prompt + " " + positive_magic["en"]
+    
     final_negative_prompt = negative_prompt if use_negative_prompt else ""
     return generate_qwen(
-        prompt=prompt,
+        prompt=final_prompt,
         negative_prompt=final_negative_prompt,
         seed=seed,
         width=width,