Upload app.py with huggingface_hub

app.py

@@ -369,12 +369,14 @@ def polish_prompt(original_prompt: str, mode: str = "generate") -> str:
 TASK: Rewrite the user's input into a precise, technical prompt describing the target visual result.
 
 STRICT RULES:
-- MAXIMUM 600 TOKENS (strict limit)
+- MAXIMUM 600 TOKENS (strict limit). You MUST write the new prompt in maximum of 600 tokens.
 - Focus on: artistic style, color palette, lighting, texture, rendering technique, mood
 - Describe HOW the image should look, not what to change
 - No action words like "transform", "convert", "change"
 - Present tense, as if describing the final image
 
+You MUST respect the maximum of 600 TOKENS in your response.
+
 OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no preamble, no word count."""
     else:
         system_prompt = """ROLE: Expert prompt engineer for AI image generation.
@@ -382,11 +384,13 @@ OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no pream
 TASK: Expand the user's input into a detailed, expressive prompt for stunning image generation.
 
 STRICT RULES:
-- MAXIMUM 600 TOKENS (strict limit)
+- MAXIMUM 600 TOKENS (strict limit). You MUST write the new prompt in maximum of 600 tokens.
 - Be descriptive about: subject, lighting, atmosphere, style, composition, details
 - Use vivid, specific language
 - Include artistic style references when appropriate
 
+You MUST respect the maximum of 600 TOKENS in your response.
+
 OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no preamble, no word count."""
 
     try: