Update app.py

app.py

@@ -6,6 +6,8 @@ from styles import get_custom_css
 from formulas import offer_formulas
 import PyPDF2
 import docx
+from PIL import Image
+import io
 
 # Set page to wide mode to use full width
 st.set_page_config(layout="wide")
@@ -15,7 +17,7 @@ load_dotenv()
 
 # Configure Google Gemini API
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
-model = genai.GenerativeModel('gemini-2.0-flash')
+model = genai.GenerativeModel('gemini-1.5-flash')  # Updated to model that supports images
 
 # Initialize session state variables if they don't exist
 if 'submitted' not in st.session_state:
@@ -45,7 +47,7 @@ col1, col2 = st.columns([4, 6])
 # Main input section in left column
 with col1:
     # Add tabs for different input methods
-    input_tab1, input_tab2 = st.tabs(["Entrada Manual", "Subir Archivo"])
+    input_tab1, input_tab2, input_tab3 = st.tabs(["Entrada Manual", "Subir Archivo", "Subir Imagen"])
     
     with input_tab1:
         skills = st.text_area('💪 Tus Habilidades', height=70, 
@@ -90,6 +92,23 @@ with col1:
             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:
+            # Display the uploaded image
+            image = Image.open(uploaded_image)
+            st.image(image, caption="Imagen cargada", use_column_width=True)
+            
+            # Convert to format for Gemini
+            image_bytes = uploaded_image.getvalue()
+            image_parts = [
+                {
+                    "mime_type": uploaded_image.type,
+                    "data": image_bytes
+                }
+            ]
+            st.success(f"Imagen cargada correctamente: {uploaded_image.name}")
+
     # Accordion for additional settings
     with st.expander('⚙️ Configuración Avanzada'):
         target_audience = st.text_area('👥 Público Objetivo', height=70,
@@ -110,11 +129,12 @@ with col1:
     # Generate button with callback
     def generate_offer():
         has_manual_input = bool(skills and product_service)
-        has_file_input = bool(uploaded_file is not None)
+        has_file_input = bool('uploaded_file' in locals() and uploaded_file is not None)
+        has_image_input = bool('uploaded_image' in locals() and uploaded_image is not None)
 
-        # Handle all three scenarios
-        if not has_manual_input and not has_file_input:
-            st.error('Por favor ingresa texto o sube un archivo')
+        # Handle all scenarios
+        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')
             return
         
         st.session_state.submitted = True
@@ -126,14 +146,27 @@ with col1:
         
         if has_file_input:
             st.session_state.file_content = file_content
+            
+        if has_image_input:
+            st.session_state.image_parts = image_parts
         
         # Set input type
-        if has_manual_input and has_file_input:
-            st.session_state.input_type = "both"
-        elif has_manual_input:
-            st.session_state.input_type = "manual"
+        if has_image_input:
+            if has_manual_input and has_file_input:
+                st.session_state.input_type = "all"
+            elif 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:
-            st.session_state.input_type = "file"
+            if has_manual_input and has_file_input:
+                st.session_state.input_type = "both"
+            elif has_manual_input:
+                st.session_state.input_type = "manual"
+            else:
+                st.session_state.input_type = "file"
         
         # Store common settings
         st.session_state.target_audience = target_audience
@@ -147,33 +180,67 @@ with col2:
     # Check if form has been submitted
     if st.session_state.submitted:
         with st.spinner('Creando tu oferta perfecta...'):
-            # Determine which input source to use
+            # Determine which input source to use and create appropriate prompt
+            base_prompt = f"""You are a professional copywriter specializing in creating irresistible offers. 
+            Create a compelling and persuasive offer using the {st.session_state.formula_type} formula.
+            
+            Target Audience: {st.session_state.target_audience if hasattr(st.session_state, 'target_audience') and st.session_state.target_audience else 'General audience'}
+            """
+            
             if st.session_state.input_type == "manual":
-                prompt = f"""Based on the following information, create a compelling offer using the {st.session_state.formula_type} formula:
+                prompt = base_prompt + f"""
+                Based on the following information:
                 Skills: {st.session_state.skills}
                 Product/Service: {st.session_state.product_service}
-                Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
                 """
             elif st.session_state.input_type == "file":
-                prompt = f"""Based on the following information from the uploaded file, create a compelling offer using the {st.session_state.formula_type} formula:
+                prompt = base_prompt + f"""
+                Based on the following information from the uploaded file:
                 File Content: {st.session_state.file_content}
-                Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
                 """
-            else:  # both inputs
-                prompt = f"""Based on the following combined information, create a compelling offer using the {st.session_state.formula_type} formula:
+            elif st.session_state.input_type == "image":
+                prompt = base_prompt + f"""
+                Based on the image provided, create an offer that highlights the visual elements and appeals to the target audience.
+                """
+            elif st.session_state.input_type == "both":
+                prompt = base_prompt + f"""
+                Based on the following combined information:
                 Skills: {st.session_state.skills}
                 Product/Service: {st.session_state.product_service}
                 Additional Information from File: {st.session_state.file_content}
-                Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
                 
                 Please consider both the manual input and file content to create a comprehensive offer.
                 """
+            elif st.session_state.input_type == "manual_image":
+                prompt = base_prompt + f"""
+                Based on the following information and the image provided:
+                Skills: {st.session_state.skills}
+                Product/Service: {st.session_state.product_service}
+                
+                Please analyze both the text information and the visual elements in the image to create a comprehensive offer.
+                """
+            elif st.session_state.input_type == "file_image":
+                prompt = base_prompt + f"""
+                Based on the following information from the uploaded file and the image provided:
+                File Content: {st.session_state.file_content}
+                
+                Please analyze both the file content and the visual elements in the image to create a comprehensive offer.
+                """
+            else:  # all inputs
+                prompt = base_prompt + f"""
+                Based on all the following information:
+                Skills: {st.session_state.skills}
+                Product/Service: {st.session_state.product_service}
+                Additional Information from File: {st.session_state.file_content}
+                
+                Please analyze the text information, file content, and the visual elements in the image to create the most comprehensive offer.
+                """
                 
             prompt += f"""
             Formula Description:
             {offer_formulas[st.session_state.formula_type]["description"]}
             
-            Please create a professional and engaging offer that highlights the value proposition.
+            Please create a professional, engaging, and irresistible offer that highlights the value proposition and creates urgency.
             
             IMPORTANT: Provide ONLY the final offer text. Do not include any explanations, labels, formatting instructions, brackets, or call to action at the end."""
             
@@ -182,7 +249,11 @@ with col2:
                 generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
                 
                 # Pass the generation config to generate_content
-                response = model.generate_content(prompt, generation_config=generation_config)
+                if "image" in st.session_state.input_type:
+                    response = model.generate_content([prompt, st.session_state.image_parts[0]], generation_config=generation_config)
+                else:
+                    response = model.generate_content(prompt, generation_config=generation_config)
+                    
                 st.session_state.offer_result = response.text
                 
                 # Display result - without emoji