Update app.py

app.py

@@ -2,38 +2,74 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
+from PIL import Image
 
 load_dotenv()
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para obtener respuesta del modelo Gemini
-def get_gemini_response(target_audience, product, text_type, length, language, mood, model_choice):
-    model = genai.GenerativeModel(model_choice)
+# Function to get response from Gemini model
+def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
+    model = genai.GenerativeModel('gemini-1.5-flash')
+    # If both image and text are provided
+    if image_data and input_prompt:
+        full_prompt = f"""
+        You are a famous creative writer. Look at the image provided and also consider the following prompt: "{input_prompt}". 
+        Create a {length} {genre} in {language}. The {genre} should be {mood}, based on both the image and the prompt, and contain realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt, image_data[0]])
+
+    # If only image is provided
+    elif image_data:
+        full_prompt = f"""
+        You are a famous creative writer. Look at the image provided and create a {length} {genre} in {language}. 
+        The {genre} should be {mood}. The {genre} should be based on the image and contain realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt, image_data[0]])
+
+    # If only text is provided
+    elif input_prompt:
+        full_prompt = f"""
+        You are a creative writer. Create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the following prompt:
+        
+        "{input_prompt}"
+        
+        Make sure it contains realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt])
     
-    # Crear el prompt completo basado en los campos del frontend
-    full_prompt = f"""
-    Eres un escritor creativo famoso. Crea un {length} {text_type} en {language}. 
-    El {text_type} debe ser {mood}, dirigido a este público objetivo: {target_audience}, y relacionado con el producto: "{product}".
-    Asegúrate de que contenga elementos realistas y emocionales.
-    """
+    # If neither image nor text is provided
+    else:
+        raise ValueError("Please provide either an image or a text prompt.")
     
-    response = model.generate_content([full_prompt])
-
-    # Comprobar si la respuesta es válida y devolver el texto
+    # Check if response is valid and return text
     if response and response.parts:
         return response.parts[0].text
     else:
-        raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
+        raise ValueError("Sorry, please try a different combination of inputs.")
+        
+# Function to setup uploaded image
+def input_image_setup(uploaded_file):
+    if uploaded_file is not None:
+        bytes_data = uploaded_file.getvalue()
+        image_parts = [
+            {
+                "mime_type": uploaded_file.type,
+                "data": bytes_data
+            }
+        ]
+        return image_parts
+    else:
+        return None
 
-# Inicializar la aplicación Streamlit
-st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wide")  # Configurar el diseño en ancho
+# Initialize Streamlit app
+st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wide")  # Set layout to wide
 
-# Componentes principales de la UI
+# Main UI components
 st.markdown("<h1 style='text-align: center;'>VisionTales</h1>", unsafe_allow_html=True)
 st.markdown("<h3 style='text-align: center;'>Convierte tus ideas en historias inolvidables.</h3>", unsafe_allow_html=True)
 
-# Añadir CSS personalizado para el botón
+# Add custom CSS for the button
 st.markdown("""
     <style>
     div.stButton > button {
@@ -56,32 +92,38 @@ st.markdown("""
     </style>
     """, unsafe_allow_html=True)
 
-# Crear dos columnas para el diseño (40% y 60%)
-col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 partes en total
+# Create two columns for layout (40% and 60%)
+col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 parts total
 
 with col1:
-    # Entradas del usuario
-    target_audience = st.text_input("Público objetivo:", placeholder="Especifica tu público aquí...")
-    product = st.text_input("Producto:", placeholder="Especifica el producto aquí...")
-    text_type = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
-    length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
-    language = st.selectbox("Idioma del texto:", ["Español", "Inglés"])
-    mood = st.selectbox("Tono del Texto:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
-    
-    # Opción para seleccionar el modelo
-    model_choice = st.selectbox("Selecciona el modelo:", ["gemini-1.5-flash", "gemini-1.5-pro"], index=0)
+    # Subir imagen primero
+    uploaded_file = st.file_uploader("Elegir una imagen...", type=["jpg", "jpeg", "png"])
 
-    # Botón para generar contenido
+    # Agrupar opciones en un desplegable con un nuevo título
+    with st.expander("Clic para personalizar tu historia", expanded=False):  # El desplegable está cerrado por defecto
+        input_prompt = st.text_input("Escribe de qué quieres que trate tu historia (opcional):", placeholder="Escribe tu mensaje aquí...")
+        genre = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
+        length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
+        language = st.selectbox("Idioma del texto:", ["Español", "Inglés"])
+        mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
+
+    # Generate button con el nuevo estilo, centrado y ajustado
     submit = st.button("Escribir mi historia")
 
-# Manejo del clic en el botón
+# Display uploaded image
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    col1.image(image, caption="Imagen subida", use_column_width=True)
+
+# Handling the button click
 if submit:
-    if target_audience and product:  # Verificar que se haya proporcionado el público objetivo y el producto
+    if uploaded_file or input_prompt:
         try:
-            response = get_gemini_response(target_audience, product, text_type, length, language, mood, model_choice)
+            image_data = input_image_setup(uploaded_file)
+            response = get_gemini_response(input_prompt, image_data, genre, length, language, mood)
             col2.subheader("Contenido generado:")
             col2.write(response)
         except ValueError as e:
             col2.error(f"Error: {str(e)}")
     else:
-        col2.error("Por favor, proporciona el público objetivo y el producto.")
+        col2.error("Por favor, sube una imagen o proporciona un mensaje.")