Update app.py

app.py

@@ -260,7 +260,7 @@ MODELS = {
         "supports_images": False
     },
 
-    # FLUX (generación de imágenes, lo dejamos como texto aquí)
+    # FLUX
     "flux_pro": {
         "provider": "openrouter",
         "name": "black-forest-labs/flux-1.1-pro",
@@ -283,17 +283,11 @@ def encode_image_to_base64(image):
 
 
 def build_messages(system_prompt, user_input, history, image_b64, supports_images):
-    """
-    Construye una lista de mensajes válida para ambas APIs:
-    - Siempre una lista de diccionarios {role, content}
-    - content es string, excepto cuando se envía imagen (lista de bloques)
-    """
     messages = []
 
     if system_prompt:
         messages.append({"role": "system", "content": system_prompt})
 
-    # Historial en formato (usuario, asistente)
     for item in history or []:
         if not isinstance(item, (list, tuple)) or len(item) != 2:
             continue
@@ -305,9 +299,7 @@ def build_messages(system_prompt, user_input, history, image_b64, supports_image
         if isinstance(assistant_msg, str) and assistant_msg.strip():
             messages.append({"role": "assistant", "content": assistant_msg})
 
-    # Mensaje actual
     if image_b64 and supports_images:
-        # Formato compatible con APIs tipo OpenAI/OpenRouter/SambaNova para visión
         content = [
             {"type": "text", "text": user_input or ""},
             {
@@ -336,7 +328,6 @@ def call_sambanova(model_name, messages):
     }
 
     if DEBUG:
-        # Para depurar formato de mensajes
         print(json.dumps(payload, indent=2))
 
     headers = {
@@ -364,7 +355,6 @@ def call_openrouter(model_name, messages):
         messages=messages,
         temperature=0.7
     )
-    # El contenido siempre se devuelve como string en esta configuración
     return response.choices[0].message.content
 
 
@@ -382,7 +372,6 @@ def chat_logic(user_text, user_image, model_key, history):
         return history, history
 
     model_cfg = MODELS[model_key]
-
     image_b64 = encode_image_to_base64(user_image) if user_image else None
 
     messages = build_messages(
@@ -465,7 +454,5 @@ def create_ui():
 demo = create_ui()
 
 if __name__ == "__main__":
-    # Asegúrate de tener instalado el paquete openai:
-    #   pip install openai
     demo.launch()