Update app.py

app.py

@@ -11,17 +11,14 @@ except ImportError:
     InferenceClient = None
 
 
-# =============================================================================
-# CONFIGURATION
-# =============================================================================
-
 APP_TITLE = "Free Image Prompt Generator"
 
-DEFAULT_MODEL = os.getenv(
+MODEL = os.getenv(
     "PROMPT_MODEL",
     "Qwen/Qwen2.5-7B-Instruct",
 )
 
+
 STYLES = [
     "Cinematic",
     "Photorealistic",
@@ -78,7 +75,8 @@ CAMERAS = [
     "Dynamic action perspective",
 ]
 
-DEFAULT_NEGATIVE = (
+
+NEGATIVE_PROMPT = (
     "text, captions, subtitles, logos, watermarks, signatures, "
     "malformed anatomy, extra limbs, duplicated objects, distorted faces, "
     "bad hands, inconsistent perspective, floating objects, visual artifacts, "
@@ -86,33 +84,25 @@ DEFAULT_NEGATIVE = (
 )
 
 
-# =============================================================================
-# TEXT HELPERS
-# =============================================================================
-
 def clean_text(text: str) -> str:
-    """Normalize whitespace while preserving paragraph breaks."""
     if not text:
         return ""
 
     text = str(text)
     text = text.replace("\r\n", "\n")
     text = text.replace("\r", "\n")
-
     text = re.sub(r"[ \t]+", " ", text)
     text = re.sub(r"\n{3,}", "\n\n", text)
 
     return text.strip()
 
 
-def extract_prompt_text(text: str) -> str:
-    """Clean common formatting accidentally returned by language models."""
+def clean_ai_response(text: str) -> str:
     if not text:
         return ""
 
     text = str(text).strip()
 
-    # Remove Markdown code fences.
     text = re.sub(
         r"^```(?:text|prompt|markdown)?\s*",
         "",
@@ -122,25 +112,17 @@ def extract_prompt_text(text: str) -> str:
 
     text = re.sub(r"\s*```$", "", text)
 
-    # Remove accidental surrounding quotation marks.
     if len(text) >= 2:
-        if (
-            (text.startswith('"') and text.endswith('"'))
-            or
-            (text.startswith("'") and text.endswith("'"))
-        ):
-            text = text[1:-1].strip()
+        if text[0] == '"' and text[-1] == '"':
+            text = text[1:-1]
 
-    return clean_text(text)
+        elif text[0] == "'" and text[-1] == "'":
+            text = text[1:-1]
 
+    return clean_text(text)
 
-# =============================================================================
-# HUGGING FACE CLIENT
-# =============================================================================
 
 def get_client() -> Optional["InferenceClient"]:
-    """Return a configured Hugging Face client if available."""
-
     token = os.getenv("HF_TOKEN")
 
     if not token:
@@ -151,20 +133,18 @@ def get_client() -> Optional["InferenceClient"]:
 
     try:
         return InferenceClient(
-            model=DEFAULT_MODEL,
+            model=MODEL,
             token=token,
         )
     except Exception as exc:
         print(
-            f"[HF CLIENT ERROR] {type(exc).__name__}: {exc}"
+            "[CLIENT ERROR]",
+            type(exc).__name__,
+            str(exc),
         )
         return None
 
 
-# =============================================================================
-# LOCAL PROMPT ENGINE
-# =============================================================================
-
 def build_local_prompt(
     idea: str,
     style: str,
@@ -172,7 +152,7 @@ def build_local_prompt(
     mood: str,
     detail: str,
     camera: str,
-) -> Tuple[str, str]:
+) -> str:
 
     idea = clean_text(idea)
 
@@ -184,77 +164,54 @@ def build_local_prompt(
 
     prompt = (
         f"{style} image of {idea}. "
-        f"Use a {camera.lower()} with intentional framing, strong visual "
-        f"hierarchy, convincing depth, natural perspective, and balanced "
-        f"negative space. "
+        f"Use a {camera.lower()} with intentional framing, "
+        f"strong visual hierarchy, convincing depth, natural perspective, "
+        f"and balanced negative space. "
         f"Create a coherent environment with meaningful foreground, "
-        f"midground, and background elements that support the subject. "
-        f"Use {mood.lower()} lighting and atmosphere with believable light "
-        f"falloff, dimensional shadows, subtle highlights, and atmospheric "
-        f"depth. "
+        f"midground, and background elements supporting the subject. "
+        f"Use {mood.lower()} lighting and atmosphere with believable "
+        f"light falloff, dimensional shadows, subtle highlights, "
+        f"and atmospheric depth. "
         f"Apply a consistent {style.lower()} aesthetic with "
         f"{detail.lower()}, refined textures, coherent materials, "
         f"realistic surface response, and controlled visual complexity. "
-        f"Use a deliberate color palette appropriate to the subject, "
-        f"environment, lighting, and mood. "
-        f"High visual fidelity, crisp focal detail, clean edges, nuanced "
-        f"textures, realistic spatial relationships, polished professional "
-        f"finish, and cinematic visual storytelling. "
+        f"Use a deliberate color palette appropriate to the scene. "
+        f"High visual fidelity, crisp focal detail, clean edges, "
+        f"nuanced textures, realistic spatial relationships, "
+        f"polished professional finish, cinematic visual storytelling. "
         f"Aspect ratio {aspect}."
     )
 
-    return clean_text(prompt), DEFAULT_NEGATIVE
-
+    return clean_text(prompt)
 
-# =============================================================================
-# AI PROMPT ENGINE
-# =============================================================================
 
-AI_SYSTEM_PROMPT = """
+AI_SYSTEM = """
 You are an expert image-prompt architect.
 
-Transform a user's image idea into a production-ready prompt for an
-image-generation model.
-
-CORE PRINCIPLES:
-
-1. Preserve the user's core idea, subject, intent, and important details.
-2. Never radically alter the user's concept.
-3. Add useful visual specificity rather than random decoration.
-4. Resolve reasonable ambiguity intelligently.
-5. Prefer concrete visual information over generic adjectives.
-6. Avoid unnecessary repetition.
-7. Every added detail must support the user's concept.
-8. Do not invent named people, brands, locations, products, or story elements
-   unless requested or clearly implied.
-9. Avoid contradictory attributes.
-10. Use useful camera, composition, lighting, spatial, material, texture,
-    environmental, color, and depth information.
-11. Avoid generic AI filler such as "masterpiece", "stunning", "breathtaking",
-    and repeated "ultra-realistic" language unless it genuinely adds meaning.
-12. Do not over-engineer simple concepts.
-13. Make the result directly usable in an image-generation model.
-14. Do not explain your choices.
-15. Do not mention that you are an AI.
-16. Return ONLY the finished image prompt.
-17. Do not wrap the prompt in quotation marks.
-
-A strong prompt should naturally communicate:
-
-- subject
-- environment
-- composition
-- camera language where useful
-- lighting
-- atmosphere
-- materials and textures where useful
-- visual style
-- color
-- depth
-- image quality
-- aspect ratio
-
-Use natural prose rather than unnecessary section headings.
+Transform the user's idea into a production-ready image-generation prompt.
+
+Preserve the user's core concept, subject and intent.
+
+Add useful visual specificity involving:
+subject, environment, composition, camera, lighting, atmosphere, materials,
+textures, color, depth and visual style.
+
+Resolve reasonable ambiguity intelligently.
+
+Do not invent unrelated story elements, people, brands or locations.
+
+Avoid generic filler such as:
+masterpiece, stunning, breathtaking, amazing, incredible.
+
+Avoid repetition and contradictory attributes.
+
+Do not radically change the user's concept.
+
+Return ONLY the finished image prompt.
+
+Do not explain your choices.
+Do not mention being an AI.
+Do not use quotation marks around the answer.
 """
 
 
@@ -272,10 +229,10 @@ def build_ai_prompt(
     if client is None:
         return (
             "",
-            "Smart AI is unavailable; using the built-in prompt engine.",
+            "Hugging Face AI unavailable.",
         )
 
-    user_prompt = f"""
+    request = f"""
 USER IDEA:
 {idea}
 
@@ -285,7 +242,7 @@ STYLE:
 ASPECT RATIO:
 {aspect}
 
-MOOD / ATMOSPHERE:
+MOOD:
 {mood}
 
 DETAIL LEVEL:
@@ -300,11 +257,11 @@ CAMERA / COMPOSITION:
             messages=[
                 {
                     "role": "system",
-                    "content": AI_SYSTEM_PROMPT,
+                    "content": AI_SYSTEM,
                 },
                 {
                     "role": "user",
-                    "content": user_prompt,
+                    "content": request,
                 },
             ],
             max_tokens=1200,
@@ -312,43 +269,42 @@ CAMERA / COMPOSITION:
         )
 
         if not response:
-            return "", "Smart AI returned no response."
+            return "", "AI returned no response."
 
         if not response.choices:
-            return "", "Smart AI returned no usable response."
+            return "", "AI returned no choices."
 
         message = response.choices[0].message
 
         if not message:
-            return "", "Smart AI returned an empty message."
+            return "", "AI returned no message."
 
         content = message.content
 
         if not content:
-            return "", "Smart AI returned an empty prompt."
+            return "", "AI returned empty content."
 
-        content = extract_prompt_text(content)
+        content = clean_ai_response(content)
 
         if not content:
-            return "", "Smart AI returned no usable prompt."
+            return "", "AI returned an empty prompt."
 
         return content, None
 
     except Exception as exc:
+
         print(
-            f"[HF AI ERROR] {type(exc).__name__}: {exc}"
+            "[HF ERROR]",
+            type(exc).__name__,
+            str(exc),
         )
 
         return (
             "",
-            "Smart AI was unavailable; using the built-in prompt engine.",
+            "Hugging Face request failed.",
         )
 
 
-# =============================================================================
-# MAIN GENERATION
-# =============================================================================
-
 def generate_prompt(
     idea: str,
     style: str,
@@ -357,187 +313,113 @@ def generate_prompt(
     detail: str,
     camera: str,
     use_ai: bool,
-) -> Tuple[str, str, str]:
+):
 
     if not idea or not idea.strip():
+
         return (
             "",
-            DEFAULT_NEGATIVE,
+            NEGATIVE_PROMPT,
             "⚠️ Enter an image idea first.",
         )
 
     idea = clean_text(idea)
 
     if use_ai:
+
         ai_prompt, error = build_ai_prompt(
-            idea=idea,
-            style=style,
-            aspect=aspect,
-            mood=mood,
-            detail=detail,
-            camera=camera,
+            idea,
+            style,
+            aspect,
+            mood,
+            detail,
+            camera,
         )
 
         if ai_prompt:
+
             return (
                 ai_prompt,
-                DEFAULT_NEGATIVE,
-                "✓ Prompt created with Smart AI",
+                NEGATIVE_PROMPT,
+                f"✓ Generated with {MODEL}",
             )
 
-        local_prompt, negative = build_local_prompt(
-            idea=idea,
-            style=style,
-            aspect=aspect,
-            mood=mood,
-            detail=detail,
-            camera=camera,
+        local_prompt = build_local_prompt(
+            idea,
+            style,
+            aspect,
+            mood,
+            detail,
+            camera,
         )
 
         return (
             local_prompt,
-            negative,
-            f"✓ Prompt created with the built-in engine • {error}",
+            NEGATIVE_PROMPT,
+            f"✓ Built-in engine used • {error}",
         )
 
-    local_prompt, negative = build_local_prompt(
-        idea=idea,
-        style=style,
-        aspect=aspect,
-        mood=mood,
-        detail=detail,
-        camera=camera,
+    local_prompt = build_local_prompt(
+        idea,
+        style,
+        aspect,
+        mood,
+        detail,
+        camera,
     )
 
     return (
         local_prompt,
-        negative,
-        "✓ Prompt created with the built-in engine",
+        NEGATIVE_PROMPT,
+        "✓ Generated with built-in engine",
     )
 
 
-# =============================================================================
-# PROMPT TRANSFORMATION
-# =============================================================================
+TRANSFORM_INSTRUCTIONS = {
 
-TRANSFORMATION_INSTRUCTIONS = {
     "Improve": """
-Improve this image prompt.
+Improve this image-generation prompt.
 
 Preserve the exact core subject and concept.
 
-Increase useful visual specificity where it genuinely improves the image.
-Strengthen composition, lighting, depth, materials, atmosphere, spatial
-relationships, and camera language where appropriate.
+Increase useful visual specificity.
+Strengthen composition, lighting, depth, atmosphere, materials and camera
+language where appropriate.
 
-Remove generic filler and repetition.
+Remove repetition and generic filler.
 
 Do not radically change the concept.
 
-Return ONLY the improved image prompt.
+Return ONLY the improved prompt.
 """,
 
     "Remix": """
-Remix this image prompt.
+Remix this image-generation prompt.
 
 Preserve the core subject and concept.
 
-Create a fresh creative interpretation by changing appropriate elements such
-as environment, composition, lighting, atmosphere, color, camera perspective,
-or visual treatment.
+Create a fresh interpretation by intelligently changing elements such as
+environment, composition, lighting, atmosphere, color or camera perspective.
 
 Do not replace the main subject.
 
-Do not turn the concept into something unrelated.
-
-Return ONLY the remixed image prompt.
+Return ONLY the remixed prompt.
 """,
 
     "Simplify": """
-Simplify this image prompt.
+Simplify this image-generation prompt.
 
-Preserve the important subject, environment, style, mood, and visual intent.
+Preserve the important subject, environment, style, mood and visual intent.
 
-Remove unnecessary adjectives, repetition, generic AI language, and
-over-engineered descriptions.
+Remove unnecessary adjectives, repetition and generic AI language.
 
-Make the prompt concise while retaining the information that matters.
+Make the prompt concise and effective.
 
-Return ONLY the simplified image prompt.
+Return ONLY the simplified prompt.
 """,
 }
 
 
-def transform_prompt(
-    prompt: str,
-    action: str,
-) -> Tuple[str, str]:
-
-    if not prompt or not prompt.strip():
-        return (
-            "",
-            "⚠️ Generate a prompt first.",
-        )
-
-    prompt = clean_text(prompt)
-
-    instruction = TRANSFORMATION_INSTRUCTIONS.get(action)
-
-    if not instruction:
-        return (
-            prompt,
-            "✓ Prompt ready.",
-        )
-
-    client = get_client()
-
-    if client is not None:
-        try:
-            response = client.chat_completion(
-                messages=[
-                    {
-                        "role": "system",
-                        "content": (
-                            "You are an expert image prompt editor. "
-                            "Return only the requested image prompt."
-                        ),
-                    },
-                    {
-                        "role": "user",
-                        "content": (
-                            f"{instruction}\n\n"
-                            f"CURRENT PROMPT:\n{prompt}"
-                        ),
-                    },
-                ],
-                max_tokens=1200,
-                temperature=(
-                    0.8
-                    if action == "Remix"
-                    else 0.5
-                ),
-            )
-
-            if response and response.choices:
-                content = response.choices[0].message.content
-
-                if content:
-                    content = extract_prompt_text(content)
-
-                    if content:
-                        return (
-                            content,
-                            f"✓ Prompt {action.lower()}ed with Smart AI",
-                        )
-
-        except Exception as exc:
-            print(
-                f"[HF TRANSFORM ERROR] {type(exc).__name__}: {exc}"
-            )
-
-    return local_transform(prompt, action)
-
-
 def local_transform(
     prompt: str,
     action: str,
@@ -547,17 +429,15 @@ def local_transform(
 
     if action == "Improve":
 
-        prompt = prompt.rstrip(". ")
-
         addition = (
-            " Enhance the scene with coherent depth, intentional composition, "
+            " Enhance coherent depth, intentional composition, "
             "natural lighting, refined textures, believable materials, "
             "and strong focal detail."
         )
 
         return (
-            f"{prompt}.{addition}",
-            "✓ Prompt improved with the built-in engine",
+            prompt.rstrip(".") + "." + addition,
+            "✓ Improved locally",
         )
 
     if action == "Simplify":
@@ -572,225 +452,123 @@ def local_transform(
 
         return (
             clean_text(prompt),
-            "✓ Prompt simplified with the built-in engine",
+            "✓ Simplified locally",
         )
 
     if action == "Remix":
 
-        remix_phrases = [
+        choices = [
             "with dramatic atmospheric depth",
             "under cinematic environmental lighting",
             "with a striking editorial composition",
             "with subtle surreal visual elements",
             "during a visually dramatic transitional moment",
-            "with strong foreground-to-background depth",
             "using an unconventional cinematic perspective",
         ]
 
-        addition = random.choice(remix_phrases)
-
-        prompt = prompt.rstrip(". ")
+        addition = random.choice(choices)
 
         return (
-            f"{prompt}, {addition}.",
-            "✓ Prompt remixed with the built-in engine",
+            prompt.rstrip(".") + ", " + addition + ".",
+            "✓ Remixed locally",
         )
 
     return (
         prompt,
-        "✓ Prompt ready.",
-    )
-
-
-def improve_prompt(prompt: str):
-    return transform_prompt(prompt, "Improve")
-
-
-def remix_prompt(prompt: str):
-    return transform_prompt(prompt, "Remix")
-
-
-def simplify_prompt(prompt: str):
-    return transform_prompt(prompt, "Simplify")
-
-
-# =============================================================================
-# SURPRISE ME
-# =============================================================================
-
-SURPRISE_IDEAS = [
-    "a tiny astronaut exploring an enormous abandoned greenhouse",
-    "a fox sitting inside an ancient observatory beneath a meteor shower",
-    "a futuristic Brisbane street transformed by bioluminescent plants",
-    "a mechanical whale swimming through a flooded cathedral",
-    "a lone explorer discovering a hidden city beneath the ocean",
-    "a tiny robot repairing a gigantic antique clock",
-    "a medieval library built inside the hollow trunk of an enormous tree",
-    "an abandoned space station slowly being reclaimed by vegetation",
-    "a cyberpunk night market during a tropical storm",
-    "a mysterious train crossing an impossible landscape at sunrise",
-    "an ancient temple hidden beneath a frozen ocean",
-    "a giant botanical garden growing inside an abandoned spacecraft",
-]
-
-
-def surprise_me():
-    idea = random.choice(SURPRISE_IDEAS)
-    style = random.choice(STYLES)
-    aspect = random.choice(ASPECT_RATIOS)
-    mood = random.choice(MOODS)
-    detail = random.choice(DETAIL_LEVELS)
-    camera = random.choice(CAMERAS)
-
-    return (
-        idea,
-        style,
-        aspect,
-        mood,
-        detail,
-        camera,
-        "🎲 Surprise idea created — hit Generate Prompt.",
+        "✓ Prompt ready",
     )
 
 
-# =============================================================================
-# EXAMPLES
-# =============================================================================
-
-EXAMPLES = [
-    [
-        "A futuristic Brisbane skyline after a tropical storm",
-        "Cinematic",
-        "16:9",
-        "Moody",
-        "Highly detailed",
-        "Wide establishing shot",
-    ],
-    [
-        "A tiny robot repairing an antique mechanical clock",
-        "3D Render",
-        "1:1",
-        "Whimsical",
-        "Extremely detailed",
-        "Macro composition",
-    ],
-    [
-        "A medieval library hidden inside a giant tree",
-        "Fantasy",
-        "4:5",
-        "Serene",
-        "Highly detailed",
-        "Symmetrical composition",
-    ],
-    [
-        "A lone astronaut exploring an abandoned underwater city",
-        "Concept Art",
-        "21:9",
-        "Mysterious",
-        "Extremely detailed",
-        "Wide establishing shot",
-    ],
-]
-
-
-# =============================================================================
-# CUSTOM CSS
-# =============================================================================
-
-CUSTOM_CSS = """
-.gradio-container {
-    max-width: 1100px !important;
-    margin: auto !important;
-}
+def transform_prompt(
+    prompt: str,
+    action: str,
+) -> Tuple[str, str]:
 
-.hero {
-    text-align: center;
-    padding: 18px 10px 8px 10px;
-}
+    if not prompt or not prompt.strip():
 
-.hero h1 {
-    font-size: 2.4rem;
-    margin-bottom: 8px;
-}
+        return (
+            "",
+            "⚠️ Generate a prompt first.",
+        )
 
-.hero p {
-    font-size: 1.05rem;
-    opacity: 0.8;
-}
+    prompt = clean_text(prompt)
 
-.output-actions {
-    margin-top: 8px;
-}
+    client = get_client()
+    instruction = TRANSFORM_INSTRUCTIONS.get(action)
 
-.status {
-    text-align: center;
-    font-size: 0.9rem;
-}
+    if client is not None and instruction:
 
-.primary-button {
-    min-height: 52px;
-    font-size: 1.05rem !important;
-    font-weight: 600 !important;
-}
+        try:
 
-footer {
-    display: none !important;
-}
-"""
+            response = client.chat_completion(
+                messages=[
+                    {
+                        "role": "system",
+                        "content": (
+                            "You are an expert image prompt editor. "
+                            "Return only the requested prompt."
+                        ),
+                    },
+                    {
+                        "role": "user",
+                        "content": (
+                            instruction
+                            + "\n\nCURRENT PROMPT:\n"
+                            + prompt
+                        ),
+                    },
+                ],
+                max_tokens=1200,
+                temperature=(
+                    0.8
+                    if action == "Remix"
+                    else 0.5
+                ),
+            )
 
+            if response and response.choices:
 
-# =============================================================================
-# GRADIO APPLICATION
-# =============================================================================
+                content = response.choices[0].message.content
 
-with gr.Blocks(
-    title=APP_TITLE,
-    css=CUSTOM_CSS,
-) as demo:
+                if content:
 
-    # -------------------------------------------------------------------------
-    # HERO
-    # -------------------------------------------------------------------------
+                    content = clean_ai_response(content)
 
-    gr.Markdown(
-        """
-<div class="hero">
+                    if content:
 
-# 🫂 Free Image Prompt Generator
+                        return (
+                            content,
+                            f"✓ {action} completed with Smart AI",
+                        )
 
-### Turn what's in your head into a production-ready image prompt.
+        except Exception as exc:
 
-Describe your idea. I'll handle the visual details.
+            print(
+                "[TRANSFORM ERROR]",
+                type(exc).__name__,
+                str(exc),
+            )
 
-</div>
-"""
+    return local_transform(
+        prompt,
+        action,
     )
 
-    # -------------------------------------------------------------------------
-    # MAIN WORKSPACE
-    # -------------------------------------------------------------------------
 
-    with gr.Row():
+def improve_prompt(prompt):
+    return transform_prompt(prompt, "Improve")
 
-        # =====================================================================
-        # INPUT PANEL
-        # =====================================================================
 
-        with gr.Column(scale=1):
+def remix_prompt(prompt):
+    return transform_prompt(prompt, "Remix")
 
-            idea = gr.Textbox(
-                label="What do you want to create?",
-                placeholder=(
-                    "Example: A lone astronaut exploring an abandoned "
-                    "underwater city"
-                ),
-                lines=5,
-                max_lines=10,
-            )
 
-            surprise_button = gr.Button(
-                "🎲 Surprise Me",
-                variant="secondary",
-            )
+def simplify_prompt(prompt):
+    return transform_prompt(prompt, "Simplify")
 
-            with gr.Accor
+
+SURPRISE_IDEAS = [
+    "a tiny astronaut exploring an enormous abandoned greenhouse",
+    "a fox sitting inside an ancient observatory beneath a meteor shower",
+    "a futuristic Brisbane street transformed by bioluminescent plants",
+    "a mechanical