Agregar el campo estilo

add_estilo.py

@@ -0,0 +1,53 @@
+"""
+Script para agregar el campo 'estilo' a la tabla 'registro' en la base de datos remota
+"""
+
+import mysql.connector
+from dotenv import load_dotenv
+import os
+
+# Cargar variables de entorno
+load_dotenv()
+
+try:
+    # Conexión a la base de datos
+    conexion = mysql.connector.connect(
+        host=os.getenv('DB_HOST'),
+        user=os.getenv('DB_USER'),
+        password=os.getenv('mariadb_c'),
+        database=os.getenv('DB_NAME'),
+        port=int(os.getenv('DB_PORT', 3306)),
+        autocommit=True
+    )
+    
+    cursor = conexion.cursor()
+    
+    # Verificar si la columna ya existe
+    cursor.execute(f"""
+        SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS 
+        WHERE TABLE_NAME = 'registro' AND COLUMN_NAME = 'estilo'
+    """)
+    
+    if cursor.fetchone():
+        print("ℹ️  El campo 'estilo' ya existe en la tabla")
+    else:
+        # Agregar la columna después de 'prompt_eval'
+        cursor.execute("""
+            ALTER TABLE registro 
+            ADD COLUMN estilo varchar(255) AFTER prompt_eval
+        """)
+        print("✅ Campo 'estilo' agregado correctamente")
+    
+    # Mostrar estructura actualizada
+    cursor.execute("DESCRIBE registro")
+    columnas = cursor.fetchall()
+    
+    print("\n📋 Estructura actualizada:")
+    for col in columnas:
+        print(f"  - {col[0]}: {col[1]}")
+    
+    cursor.close()
+    conexion.close()
+    
+except Exception as e:
+    print(f"❌ Error: {str(e)}")

funciones.py

@@ -40,23 +40,24 @@ def crear_registro(datos: RegistroRequest) -> dict:
         prompt = datos.prompt
         prompt_type = datos.prompt_type
         prompt_eval = datos.prompt_eval
+        estilo = datos.estilo
         proveedor = datos.proveedor
         seed = datos.seed
         
-        logger.info(f"📊 Datos a insertar - UID: {uid}, Usuario: {display_name}, País: {pais}, Email: {correo}, Prompt: {prompt}, Tipo: {prompt_type}, Prompt Eval: {prompt_eval}, Proveedor: {proveedor}")
+        logger.info(f"📊 Datos a insertar - UID: {uid}, Usuario: {display_name}, País: {pais}, Email: {correo}, Prompt: {prompt}, Tipo: {prompt_type}, Eval: {prompt_eval}, Estilo: {estilo}, Proveedor: {proveedor}")
         
         # Insertar en la BD
         cursor = conn.cursor()
         
         sql = """
         INSERT INTO `registro` 
-        (uid, display_name, pais, correo, fecha_registro, usos, prompt, prompt_type, prompt_eval, proveedor, seed)
-        VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
+        (uid, display_name, pais, correo, fecha_registro, usos, prompt, prompt_type, prompt_eval, estilo, proveedor, seed)
+        VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
         """
         
         valores = (
             uid, display_name, pais, correo, fecha_registro,
-            usos, prompt, prompt_type, prompt_eval, proveedor, seed
+            usos, prompt, prompt_type, prompt_eval, estilo, proveedor, seed
         )
         
         cursor.execute(sql, valores)
@@ -72,7 +73,7 @@ def crear_registro(datos: RegistroRequest) -> dict:
         cursor.execute(
             """
             SELECT id, uid, display_name, pais, correo, fecha_registro, 
-                   usos, prompt, prompt_type, prompt_eval, proveedor, seed, created_at 
+                   usos, prompt, prompt_type, prompt_eval, estilo, proveedor, seed, created_at 
             FROM `registro` 
             WHERE id = %s
             """,
@@ -98,9 +99,10 @@ def crear_registro(datos: RegistroRequest) -> dict:
                     "prompt": registro[7],
                     "prompt_type": registro[8],
                     "prompt_eval": registro[9],
-                    "proveedor": registro[10],
-                    "seed": registro[11],
-                    "created_at": registro[12]
+                    "estilo": registro[10],
+                    "proveedor": registro[11],
+                    "seed": registro[12],
+                    "created_at": registro[13]
                 }
             }
         else:
@@ -152,7 +154,7 @@ def obtener_registros(limit: int = 10, offset: int = 0) -> dict:
         cursor.execute(
             """
             SELECT id, uid, display_name, pais, correo, fecha_registro, 
-                   usos, prompt, prompt_type, prompt_eval, proveedor, seed, created_at 
+                   usos, prompt, prompt_type, prompt_eval, estilo, proveedor, seed, created_at 
             FROM `registro` 
             ORDER BY created_at DESC 
             LIMIT %s OFFSET %s
@@ -179,9 +181,10 @@ def obtener_registros(limit: int = 10, offset: int = 0) -> dict:
                 "prompt": reg[7],
                 "prompt_type": reg[8],
                 "prompt_eval": reg[9],
-                "proveedor": reg[10],
-                "seed": reg[11],
-                "created_at": reg[12]
+                "estilo": reg[10],
+                "proveedor": reg[11],
+                "seed": reg[12],
+                "created_at": reg[13]
             })
         
         logger.info(f"📈 Retornando {len(datos)} registros")

imagen_log.sql

@@ -9,6 +9,7 @@ CREATE TABLE `registro` (
   `prompt` varchar(255),
   `prompt_type` varchar(255),
   `prompt_eval` varchar(255),
+  `estilo` varchar(255),
   `proveedor` varchar(255),
   `seed` int,
   `created_at` timestamp

main.py

@@ -61,13 +61,14 @@ async def registrar(datos: RegistroRequest):
     - prompt: Prompt utilizado
     - prompt_type: Tipo de prompt
     - prompt_eval: Evaluación del prompt procesado
+    - estilo: Estilo para la generación
     - proveedor: Proveedor del servicio
     - seed: Seed para generación
     
     **Respuesta:** Retorna los datos del registro creado con su ID único
     """
     
-    logger.info(f"📝 Nuevo registro - UID: {datos.uid}, Email: {datos.correo}, Prompt: {datos.prompt}, Tipo: {datos.prompt_type}, Eval: {datos.prompt_eval}")
+    logger.info(f"📝 Nuevo registro - UID: {datos.uid}, Email: {datos.correo}, Prompt: {datos.prompt}, Tipo: {datos.prompt_type}, Eval: {datos.prompt_eval}, Estilo: {datos.estilo}")
     
     resultado = crear_registro(datos)
     

models.py

@@ -16,6 +16,7 @@ class RegistroRequest(BaseModel):
     prompt: Optional[str] = Field(None, max_length=255, description="Prompt utilizado")
     prompt_type: Optional[str] = Field(None, max_length=255, description="Tipo de prompt")
     prompt_eval: Optional[str] = Field(None, max_length=255, description="Evaluación del prompt")
+    estilo: Optional[str] = Field(None, max_length=255, description="Estilo para la generación")
     proveedor: Optional[str] = Field(None, max_length=255, description="Proveedor del servicio")
     seed: Optional[int] = Field(None, description="Seed para generación")
 
@@ -42,6 +43,7 @@ class RegistroResponse(BaseModel):
     prompt: Optional[str]
     prompt_type: Optional[str]
     prompt_eval: Optional[str]
+    estilo: Optional[str]
     proveedor: Optional[str]
     seed: Optional[int]
     created_at: datetime