Update app.py

app.py

@@ -14,38 +14,19 @@ from dataclasses import dataclass, field
 from typing import List
 
 st.set_page_config(layout="wide")
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-logger = logging.getLogger(__name__)
-
-@dataclass
-class AppConfig:
-    MAX_SEED: int = np.iinfo(np.int32).max
-    DEFAULT_WIDTH: int = 1280
-    DEFAULT_HEIGHT: int = 720
-    SUPPORTED_FORMATS: List[str] = field(default_factory=lambda: ["9:16", "16:9", "1:1"])
-    CLEANUP_DAYS: int = 7
-    MAX_GALLERY_SIZE: int = 1000
-
-credentials = {"username": "", "password": ""}
 
 try:
-    with open("credentials.yaml", "r") as file:
-        file_credentials = yaml.safe_load(file)
-        credentials["username"] = file_credentials.get("username", credentials["username"])
-        credentials["password"] = file_credentials.get("password", credentials["password"])
+    with open("config.yaml", "r") as file:
+        credentials = yaml.safe_load(file)
 except Exception as e:
-    logger.warning(f"No se pudo cargar el archivo de credenciales: {e}")
+    st.error(f"Error al cargar el archivo de configuración: {e}")
+    credentials = {"username": "", "password": ""}
 
-    config = {
-        "development": {
-            "max_image_size": 1280,
-            "supported_formats": ["9:16", "16:9", "1:1"],
-            "models": ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-dev"],
-            "storage": {"cleanup_after_days": 7, "max_gallery_size": 1000}
-        }
-    }
+@dataclass
+class AppConfig:
+    MAX_SEED: int = 1000000  
+    CLEANUP_DAYS: int = 7    
 
-MAX_SEED = AppConfig.MAX_SEED
 client = InferenceClient()
 DATA_PATH = Path("./data")
 DATA_PATH.mkdir(exist_ok=True)
@@ -100,7 +81,7 @@ def swap_faces(source_image, source_face_index, destination_image, destination_f
 def generate_image(prompt, width, height, seed, model_name):
     try:
         with st.spinner("Generando imagen..."):
-            seed = int(seed) if seed != -1 else random.randint(0, MAX_SEED)
+            seed = int(seed) if seed != -1 else random.randint(0, AppConfig.MAX_SEED)
             image = client.text_to_image(
                 prompt=prompt,
                 height=height,