Upload app.py with huggingface_hub

app.py

@@ -430,14 +430,16 @@ 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.
 
+GLOBAL RULE: You must generate the improved prompt with a maximum size of 800 tokens. This is a Global rule.
+
 STRICT RULES:
-- MAXIMUM 600 TOKENS (strict limit). You MUST write the new prompt in maximum of 600 tokens.
+- MAXIMUM 800 TOKENS (strict limit). You MUST write the new prompt in maximum of 800 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.
+You MUST respect the maximum of 800 TOKENS in your response. This is a Global rule.
 
 OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no preamble, no word count."""
     else:
@@ -445,20 +447,22 @@ 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.
 
+GLOBAL RULE: You must generate the improved prompt with a maximum size of 800 tokens. This is a Global rule.
+
 STRICT RULES:
-- MAXIMUM 600 TOKENS (strict limit). You MUST write the new prompt in maximum of 600 tokens.
+- MAXIMUM 800 TOKENS (strict limit). You MUST write the new prompt in maximum of 800 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.
+You MUST respect the maximum of 800 TOKENS in your response. This is a Global rule.
 
 OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no preamble, no word count."""
 
     try:
         response = client.chat.completions.create(
             model="deepseek-reasoner",
-            max_tokens=600,
+            max_tokens=800,
             messages=[
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": original_prompt}
@@ -484,7 +488,7 @@ OUTPUT FORMAT: Only the final prompt text. No thinking, no explanation, no pream
                 content = content.split("</think>")[-1].strip()
             if content.startswith('"') and content.endswith('"'):
                 content = content[1:-1]
-            max_words = 600  # 600 tokens limit for all modes
+            max_words = 800  # 800 tokens limit for all modes (Global rule)
             words = content.split()
             if len(words) > max_words:
                 content = " ".join(words[:max_words])