Update app.py

app.py

@@ -3,24 +3,19 @@ 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"))
 
-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):
+# 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 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.
         """
+        response = model.generate_content([full_prompt, image_data[0]])
     else:
         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:
@@ -29,37 +24,11 @@ def get_gemini_response(input_prompt, image_data, genre, length, language, mood,
         
         Make sure it contains realistic and emotional elements.
         """
-
-    # 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")
-
+        response = model.generate_content([full_prompt])
+    
     # Check if response is valid and return text
-    if response and response.text:
-        return response.text
+    if response and response.parts:
+        return response.parts[0].text
     else:
         raise ValueError("Sorry, But please try a different combination of inputs.")
 
@@ -92,6 +61,12 @@ 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
@@ -99,12 +74,7 @@ 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,
-                                            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)
+            response = get_gemini_response(input_prompt, image_data, genre, length, language, mood)
             st.subheader("Generated Content:")
             st.write(response)
         except ValueError as e:
@@ -115,3 +85,4 @@ if submit:
 st.sidebar.title("About")
 st.sidebar.info("This app uses AI to create stories, shayari, sher, ghazal, poems, or quotes based on an uploaded image or a provided prompt. "
                 "Upload an image or enter a prompt, and select the genre, length, language, and mood to generate your content.")
+