Update app.py

app.py

@@ -3,15 +3,21 @@ import streamlit as st
 import os
 import google.generativeai as genai
 from PIL import Image
+import textwrap
 
 load_dotenv()
 
+# Configurar la API de Google Gemini
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Function to get response from Gemini model
-def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
-    # Prepare the prompts
+def upload_to_gemini(path, mime_type=None):
+    """Sube el archivo a Google Gemini."""
+    file = genai.upload_file(path, mime_type=mime_type)
+    return file
+
+def get_gemini_response(input_prompt, image_data, genre, length, language, mood, model_name, temperature, top_p, top_k, max_output_tokens):
     if image_data:
+        uploaded_file = upload_to_gemini(image_data[0], mime_type=image_data[0]["mime_type"])
         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.
         """
@@ -24,12 +30,36 @@ def get_gemini_response(input_prompt, image_data, genre, length, language, mood)
         Make sure it contains realistic and emotional elements.
         """
 
-    # Call the API (the method might differ; check the documentation)
-    response = genai.generate([full_prompt])  # Cambiar aquí si es necesario según la documentación
+    # Configurar el modelo seleccionado
+    generation_config = {
+        "temperature": temperature,
+        "top_p": top_p,
+        "top_k": top_k,
+        "max_output_tokens": max_output_tokens,
+        "response_mime_type": "text/plain",
+    }
+
+    model = genai.GenerativeModel(
+        model_name=model_name,
+        generation_config=generation_config,
+    )
+
+    # Crear una sesión de chat para interactuar con Gemini
+    chat_session = model.start_chat(
+        history=[
+            {
+                "role": "user",
+                "parts": [uploaded_file, full_prompt],
+            },
+        ]
+    )
+
+    # Enviar el mensaje y obtener la respuesta
+    response = chat_session.send_message("Generate content based on the prompt")
 
     # Check if response is valid and return text
-    if response and response[0]:  # Ajusta el índice según la respuesta de la API
-        return response[0].text
+    if response and response.text:
+        return response.text
     else:
         raise ValueError("Sorry, But please try a different combination of inputs.")
 
@@ -62,12 +92,6 @@ length = st.selectbox("Select Length:", ["short", "long"])
 language = st.selectbox("Select Language:", ["English", "Hindi"])
 mood = st.selectbox("Select Mood:", ["emotional", "sad", "happy", "horror", "comedy", "romantic"])
 
-image = None
-
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
 submit = st.button("Generate")
 
 # Handle submit action
@@ -75,7 +99,12 @@ if submit:
     if uploaded_file or input_prompt:
         try:
             image_data = input_image_setup(uploaded_file)
-            response = get_gemini_response(input_prompt, image_data, genre, length, language, mood)
+            response = get_gemini_response(input_prompt, image_data, genre, length, language, mood,
+                                            model_name="gemini-1.5-pro",  # O el modelo que selecciones
+                                            temperature=0.9,  # Ajusta estos valores según tu interfaz
+                                            top_p=0.95,
+                                            top_k=64,
+                                            max_output_tokens=1024)
             st.subheader("Generated Content:")
             st.write(response)
         except ValueError as e: