Update app.py

app.py

@@ -57,9 +57,16 @@ async def gen(prompt, basemodel, w, h, scales, steps, seed, upscale_factor, proc
     return [str(image_path), str(prompt_file_path)]
 
 async def improve_prompt(prompt, language):
-    instruction = "With this idea, describe in English a detailed txt2img prompt in 500 characters at most..." if language == "en" else "Con esta idea, describe en español un prompt detallado de txt2img..."
-    response = llm_client.text_generation(f"{prompt}: {instruction}", max_new_tokens=500)
-    return response.get('generated_text', '').strip()[:500]
+    try:
+        instruction_en = "With this idea, describe in English a detailed txt2img prompt in 500 characters at most, add illumination, atmosphere, cinematic elements, and characters if need it..."
+        instruction_es = "Con esta idea, describe en español un prompt detallado de txt2img en un máximo de 500 caracteres, con iluminación, atmósfera, elementos cinematográficos y en su caso personajes..."
+        instruction = instruction_en if language == "en" else instruction_es
+        formatted_prompt = f"{prompt}: {instruction}"
+        response = llm_client.text_generation(formatted_prompt, max_new_tokens=500)
+        improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
+        return improved_text[:500] if len(improved_text) > 500 else improved_text
+    except Exception as e:
+        return f"Error mejorando el prompt: {e}"
 
 def save_image(image, seed):
     if image.mode == 'RGBA': image = image.convert('RGB')