Update app.py

app.py

@@ -13,10 +13,12 @@ client = AsyncInferenceClient()
 DATA_PATH = Path("./data")
 DATA_PATH.mkdir(exist_ok=True)
 
+PREDEFINED_SEED = random.randint(0, MAX_SEED)
+
 async def generate_image(prompt, width, height, seed):
     try:
         if seed == -1:
-            seed = random.randint(0, MAX_SEED)
+            seed = PREDEFINED_SEED
         seed = int(seed)
         image = await client.text_to_image(
             prompt=prompt, height=height, width=width, model="enhanceaiteam/Flux-uncensored"
@@ -35,12 +37,10 @@ def save_prompt(prompt_text, seed):
         st.error(f"Error al guardar el prompt: {e}")
         return None
 
-async def gen(prompt, width, height, seed):
+async def gen(prompt, width, height):
     combined_prompt = prompt  
 
-    if seed == -1:
-        seed = random.randint(0, MAX_SEED)
-    seed = int(seed)
+    seed = PREDEFINED_SEED
     progress_bar = st.progress(0)
     image, seed = await generate_image(combined_prompt, width, height, seed)
     progress_bar.progress(100)
@@ -88,7 +88,7 @@ def main():
 
     if st.sidebar.button("Generar Imagen"):
         with st.spinner("Generando imagen..."):
-            result = asyncio.run(gen(prompt, width, height, seed))
+            result = asyncio.run(gen(prompt, width, height))
             image_paths = result[0]
             prompt_file = result[1]