Upload 9 files

app.py

@@ -16,7 +16,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Fórmulas con ejemplos y explicaciones
 # headline_formulas dictionary has been moved to formulas/headline_formulas.py
 
-def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -72,6 +72,15 @@ IMPORTANT:
     # Iniciar el prompt con las instrucciones del sistema
     headlines_instruction = f"{system_prompt}\n\n"
 
+    # Añadir contenido del archivo si existe
+    if file_content:
+        headlines_instruction += f"""
+CONTENIDO DE REFERENCIA:
+Utiliza el siguiente contenido como referencia para generar los titulares:
+{file_content[:3000]}  # Limitamos a 3000 caracteres para evitar tokens excesivos
+
+"""
+    
     # Añadir instrucciones de ángulo solo si no es "NINGUNO"
     if selected_angle != "NINGUNO":
         headlines_instruction += f"""
@@ -151,16 +160,31 @@ GENERA AHORA:
 Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
 """
 
-    chat_session = model.start_chat(
-        history=[
-            {
-                "role": "user",
-                "parts": [headlines_instruction],
-            },
-        ]
-    )
-
-    response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados.")
+    # Modificar la forma de enviar el mensaje según si hay imagen o no
+    if is_image and image_parts:
+        chat_session = model.start_chat(
+            history=[
+                {
+                    "role": "user",
+                    "parts": [
+                        headlines_instruction,
+                        image_parts
+                    ],
+                },
+            ]
+        )
+        response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados, inspirándote en la imagen proporcionada.")
+    else:
+        chat_session = model.start_chat(
+            history=[
+                {
+                    "role": "user",
+                    "parts": [headlines_instruction],
+                },
+            ]
+        )
+        response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados.")
+    
     return response.text
 
 # Configurar la interfaz de usuario con Streamlit
@@ -209,6 +233,65 @@ with col1:
             "Selecciona el ángulo para tus titulares",
             options=angle_keys
         )
+        
+        # Añadir cargador de archivos dentro del acordeón
+        uploaded_file = st.file_uploader("📄 Archivo o imagen de referencia", 
+                                        type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
+        
+        file_content = ""
+        is_image = False
+        image_parts = None
+        
+        if uploaded_file is not None:
+            file_type = uploaded_file.name.split('.')[-1].lower()
+            
+            # Manejar archivos de texto
+            if file_type in ['txt', 'pdf', 'docx']:
+                if file_type == 'txt':
+                    try:
+                        file_content = uploaded_file.read().decode('utf-8')
+                        st.success(f"Archivo TXT cargado correctamente: {uploaded_file.name}")
+                    except Exception as e:
+                        st.error(f"Error al leer el archivo TXT: {str(e)}")
+                        file_content = ""
+                
+                elif file_type == 'pdf':
+                    try:
+                        import PyPDF2
+                        pdf_reader = PyPDF2.PdfReader(uploaded_file)
+                        file_content = ""
+                        for page in pdf_reader.pages:
+                            file_content += page.extract_text() + "\n"
+                        st.success(f"Archivo PDF cargado correctamente: {uploaded_file.name}")
+                    except Exception as e:
+                        st.error(f"Error al leer el archivo PDF: {str(e)}")
+                        file_content = ""
+                        
+                elif file_type == 'docx':
+                    try:
+                        import docx
+                        doc = docx.Document(uploaded_file)
+                        file_content = "\n".join([para.text for para in doc.paragraphs])
+                        st.success(f"Archivo DOCX cargado correctamente: {uploaded_file.name}")
+                    except Exception as e:
+                        st.error(f"Error al leer el archivo DOCX: {str(e)}")
+                        file_content = ""
+            
+            # Manejar archivos de imagen
+            elif file_type in ['jpg', 'jpeg', 'png']:
+                try:
+                    from PIL import Image
+                    image = Image.open(uploaded_file)
+                    image_bytes = uploaded_file.getvalue()
+                    image_parts = {
+                        "mime_type": uploaded_file.type,
+                        "data": image_bytes
+                    }
+                    is_image = True
+                    st.image(image, caption="Imagen cargada", use_column_width=True)
+                except Exception as e:
+                    st.error(f"Error al procesar la imagen: {str(e)}")
+                    is_image = False
 
     selected_formula = headline_formulas[selected_formula_key]
 
@@ -225,7 +308,10 @@ if submit:
                 product,
                 temperature,
                 selected_formula,
-                selected_angle  # Agregar el ángulo seleccionado
+                selected_angle,
+                file_content if 'file_content' in locals() else "",
+                image_parts if 'image_parts' in locals() else None,
+                is_image if 'is_image' in locals() else False
             )
             col2.markdown(f"""
                 <div class="results-container">