Update app.py

app.py

@@ -46,7 +46,7 @@ col1, col2 = st.columns([4, 6])
 
 # Main input section in left column
 with col1:
-    input_tab1, input_tab2, input_tab3 = st.tabs(["Entrada Manual", "Subir Archivo", "Subir Imagen"])
+    input_tab1, input_tab2 = st.tabs(["Entrada Manual", "Subir Archivo o Imagen"])
     
     with input_tab1:
         skills = st.text_area('💪 Tus Habilidades', height=70, 
@@ -55,53 +55,62 @@ with col1:
                                     help='Describe tu producto o servicio')
     
     with input_tab2:
-        uploaded_file = st.file_uploader("Sube un archivo con tu información", type=['txt', 'pdf', 'docx'])
+        uploaded_file = st.file_uploader("Sube un archivo o imagen", 
+                                        type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
         if uploaded_file is not None:
             file_type = uploaded_file.name.split('.')[-1].lower()
             
-            if file_type == 'txt':
-                file_content = uploaded_file.read().decode('utf-8')
-                st.success(f"Archivo cargado correctamente: {uploaded_file.name}")
-                
-            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 = ""
+            # Handle text files
+            if file_type in ['txt', 'pdf', 'docx']:
+                if file_type == 'txt':
+                    file_content = uploaded_file.read().decode('utf-8')
+                    st.success(f"Archivo de texto cargado correctamente: {uploaded_file.name}")
                     
-            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 = ""
-            
-            with st.expander("Vista previa del contenido"):
-                st.write(file_content[:500] + "..." if len(file_content) > 500 else file_content)
-
-    with input_tab3:
-        uploaded_image = st.file_uploader("Sube una imagen de tu producto o servicio", type=['jpg', 'jpeg', 'png'])
-        if uploaded_image is not None:
-            image = Image.open(uploaded_image)
-            st.image(image, caption="Imagen cargada", use_container_width=True)
+                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 = ""
+                
+                # Display preview of file content
+                with st.expander("Vista previa del contenido"):
+                    st.write(file_content[:500] + "..." if len(file_content) > 500 else file_content)
+                
+                # Set file type flag
+                is_image = False
             
-            image_bytes = uploaded_image.getvalue()
-            image_parts = [
-                {
-                    "mime_type": uploaded_image.type,
-                    "data": image_bytes
-                }
-            ]
-            st.success(f"Imagen cargada correctamente: {uploaded_image.name}")
+            # Handle image files
+            elif file_type in ['jpg', 'jpeg', 'png']:
+                image = Image.open(uploaded_file)
+                st.image(image, caption="Imagen cargada", use_container_width=True)
+                
+                image_bytes = uploaded_file.getvalue()
+                image_parts = [
+                    {
+                        "mime_type": uploaded_file.type,
+                        "data": image_bytes
+                    }
+                ]
+                st.success(f"Imagen cargada correctamente: {uploaded_file.name}")
+                
+                # Set file type flag
+                is_image = True
 
     with st.expander('⚙️ Configuración Avanzada'):
         target_audience = st.text_area('👥 Público Objetivo', height=70,
@@ -118,12 +127,12 @@ with col1:
 
     def generate_offer():
         has_manual_input = bool(skills or product_service)
-        has_file_input = bool(uploaded_file is not None)
-        has_image_input = bool(uploaded_image is not None)
+        has_file_input = bool(uploaded_file is not None and not is_image)
+        has_image_input = bool(uploaded_file is not None and is_image)
 
         # Simple validation - check if we have at least one input type
         if not (has_manual_input or has_file_input or has_image_input):
-            st.error('Por favor ingresa texto, sube un archivo o una imagen')
+            st.error('Por favor ingresa texto o sube un archivo/imagen')
             return
         
         st.session_state.submitted = True
@@ -141,12 +150,8 @@ with col1:
         
         # Set input type based on what's available
         if has_image_input:
-            if has_manual_input and has_file_input:
-                st.session_state.input_type = "all"
-            elif has_manual_input:
+            if has_manual_input:
                 st.session_state.input_type = "manual_image"
-            elif has_file_input:
-                st.session_state.input_type = "file_image"
             else:
                 st.session_state.input_type = "image"
         else: