Update app.py

app.py

@@ -90,6 +90,10 @@ async def improve_prompt(prompt):
         formatted_prompt = f"{prompt}: {instruction}"
         response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
         improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
+        
+        if len(improved_text) > 200:
+            improved_text = improved_text[:200]
+        
         return improved_text
     except Exception as e:
         return f"Error mejorando el prompt: {e}"
@@ -99,7 +103,7 @@ def get_storage():
         for file in DATA_PATH.glob("*.jpg") 
         if file.is_file()]
     usage = sum([f['size'] for f in files])
-    return [file["name"] for file in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
+    return [file["name"] for f in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
 
 def get_prompts():
     prompt_files = [file for file in DATA_PATH.glob("*.txt") if file.is_file()]
@@ -123,24 +127,25 @@ def delete_image(image_path):
 
 st.set_page_config(layout="wide")
 st.title("Generador de Imágenes FLUX")
-prompt = st.sidebar.text_input("Descripción de la imagen")
-basemodel = st.sidebar.selectbox("Modelo Base", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"])
-lora_model = st.sidebar.selectbox("LORA Realismo", ["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"])
-format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
-process_lora = st.sidebar.checkbox("Procesar LORA")
-process_upscale = st.sidebar.checkbox("Procesar Escalador")
+prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=200)
+
+with st.sidebar.expander("Opciones avanzadas", expanded=False):
+    basemodel = st.selectbox("Modelo Base", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"])
+    lora_model = st.selectbox("LORA Realismo", ["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"])
+    format_option = st.selectbox("Formato", ["9:16", "16:9"])
+    process_lora = st.checkbox("Procesar LORA")
+    process_upscale = st.checkbox("Procesar Escalador")
+    upscale_factor = st.selectbox("Factor de Escala", [2, 4, 8], index=0)
+    scales = st.slider("Escalado", 1, 20, 10)
+    steps = st.slider("Pasos", 1, 100, 20)
+    seed = st.number_input("Semilla", value=-1)
 
 if format_option == "9:16":
-    width = st.sidebar.slider("Ancho", 512, 720, 720, step=8)
-    height = st.sidebar.slider("Alto", 912, 1280, 1280, step=8)
+    width = 720
+    height = 1280
 else:
-    width = st.sidebar.slider("Ancho", 512, 1280, 1280, step=8)
-    height = st.sidebar.slider("Alto", 512, 720, 720, step=8)
-
-upscale_factor = st.sidebar.selectbox("Factor de Escala", [2, 4, 8], index=0)
-scales = st.sidebar.slider("Escalado", 1, 20, 10)
-steps = st.sidebar.slider("Pasos", 1, 100, 20)
-seed = st.sidebar.number_input("Semilla", value=-1)
+    width = 1280
+    height = 720
 
 if st.sidebar.button("Mejorar prompt"):
     improved_prompt = asyncio.run(improve_prompt(prompt))
@@ -149,18 +154,18 @@ if st.sidebar.button("Mejorar prompt"):
 
 if st.sidebar.button("Generar Imagen"):
     with st.spinner("Generando imagen..."):
-        result = run_gen()  
-        image_paths = result[0]  
+        result = run_gen()
+        image_paths = result[0]
         prompt_file = result[1]
-    
-    st.write(f"Image paths: {image_paths}")  
-    
+
+    st.write(f"Image paths: {image_paths}")
+
     if image_paths:
         if Path(image_paths).exists():
             st.image(image_paths, caption="Imagen Generada")
         else:
             st.error("El archivo de imagen no existe.")
-        
+
         if prompt_file and Path(prompt_file).exists():
             prompt_text = Path(prompt_file).read_text()
             st.write(f"Prompt utilizado: {prompt_text}")