Update app.py

app.py

@@ -5,6 +5,7 @@ from diffusers import FluxPipeline
 from groq import Groq  # Import the Groq library
 from cryptography.fernet import Fernet
 from huggingface_hub import login
+from transformers import CLIPTokenizer
 def get_hf_token(encrypted_token):
     # Retrieve the decryption key from an environment variable
     key = "K4FlQbffvTcDxT2FIhrOPV1eue6ia45FFR3kqp2hHbM="
@@ -41,7 +42,7 @@ def enhance_prompt(user_prompt):
                         "but also details like tone, style, color palette, and point of view. "
                         "For photorealistic images, include the device used (e.g., 'shot on iPhone 16'), "
                         "aperture, lens, and shot type. Use clear, visual descriptions. "
-                        "Prompt Structure: 'A [medium] of [subject], [subject’s characteristics], "
+                        "Prompt Structure: Prompt should only contain keywords, no phrases. "
                         "[relation to background] [background]. [Details of background] "
                         "[Interactions with color and lighting]. (\"Taken on:\"/\"Drawn with:\")[Specific traits of style]' "
                         "Include details on medium, subject (clothing, hairstyle, pose, etc.), background, "
@@ -61,12 +62,23 @@ def enhance_prompt(user_prompt):
     except Exception as e:
         enhanced = f"Error enhancing prompt: {str(e)}"
     return enhanced
+tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-large-patch14")
+
+def truncate_prompt(prompt, max_length=77):
+    # Tokenize without adding special tokens
+    tokens = tokenizer(prompt, add_special_tokens=False)["input_ids"]
+    if len(tokens) > max_length:
+        tokens = tokens[:max_length]
+        # Decode tokens back to a string (skipping special tokens)
+        prompt = tokenizer.decode(tokens, skip_special_tokens=True)
+    return prompt
 
 def generate_image(prompt):
     """Generates an image using the refined prompt."""
+    truncated_prompt = truncate_prompt(prompt)
     try:
         image = pipe(
-        prompt,
+        truncated_prompt,
         height=1024,
         width=1024,
         guidance_scale=3.5,