Update app.py

app.py

@@ -77,7 +77,7 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
             upscale_image = Image.open(upscale_image_path)
             upscale_image.save(DATA_PATH / f"upscale_image_{seed}.jpg", format="JPEG")
             progress_bar.progress(100)
-            image_path.unlink()  # Borrar la imagen original si se escaló
+            image_path.unlink()  
             return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path)]
         else:
             progress_bar.empty()
@@ -88,11 +88,11 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
 
 async def improve_prompt(prompt):
     try:
-        instruction = ("With this idea, describe in English a detailed txt2img prompt in 200 character at most...")
-        formatted_prompt = f"{prompt}: {instruction}"
-        response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
+        instruction = ("With this idea, describe in English a detailed txt2img prompt in 300 character at most...")
+        formatted_prompt = f"{instruction}"
+        response = llm_client.text_generation(formatted_prompt, max_new_tokens=300)
         improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
-        return improved_text[:200] if len(improved_text) > 200 else improved_text
+        return improved_text[:300] if len(improved_text) > 300 else improved_text
     except Exception as e:
         return f"Error mejorando el prompt: {e}"
 
@@ -152,12 +152,12 @@ else:
     width = 1280
     height = 720
 
-if st.sidebar.button("Mejorar prompt"):
-    improved_prompt = asyncio.run(improve_prompt(prompt))
-    st.session_state.improved_prompt = improved_prompt
-    st.write(f"Prompt mejorado: {improved_prompt}")
-
 if st.sidebar.button("Generar Imagen"):
+    with st.spinner("Mejorando prompt..."):
+        improved_prompt = asyncio.run(improve_prompt(prompt))
+        st.session_state.improved_prompt = improved_prompt
+        st.write(f"Prompt mejorado: {improved_prompt}")
+    
     with st.spinner("Generando imagen..."):
         result = run_gen()
         image_paths = result[0]