Update app.py

app.py

@@ -3,59 +3,54 @@ from PIL import Image
 import textwrap
 import google.generativeai as genai
 
-# Function to display formatted Markdown text
-def to_markdown(text):
-    text = text.replace('•', '  *')
-    return textwrap.indent(text, '> ', predicate=lambda _: True)
-
-# Function to generate content using Gemini API
-def generate_gemini_content(prompt, model_name='gemini-pro-vision', image=None):
-    model = genai.GenerativeModel(model_name)
-    if not image:
-        st.warning("Please add an image to use the gemini-pro-vision model.")
+# Función para mostrar texto formateado en Markdown
+def to_markdown(texto):
+    texto = texto.replace('•', '  *')
+    return textwrap.indent(texto, '> ', predicate=lambda _: True)
+
+# Función para generar contenido usando la API Gemini
+def generar_contenido_gemini(prompt, model_name='gemini-pro-vision', imagen=None):
+    api_key = "AIzaSyCv30EVBXynJjSNZiYkKqc5xx1DovQEaQY"  # Inserta tu API key aquí
+    genai.configure(api_key=api_key)
+    modelo = genai.GenerativeModel(model_name)
+    if not imagen:
+        st.warning("Por favor, agrega una imagen para usar el modelo gemini-pro-vision.")
         return None
 
-    response = model.generate_content([prompt, image])
-    return response
+    respuesta = modelo.generate_content([prompt, imagen])
+    return respuesta
 
-# Streamlit app
+# App de Streamlit
 def main():
-    st.title("Gemini API Demo with Streamlit")
-
-    # Get Gemini API key from user input
-    api_key = st.text_input("Enter your Gemini API key:")
-    genai.configure(api_key=api_key)
-
-    # Set the model to 'gemini-pro-vision'
-    model_name = 'gemini-pro-vision'
+    st.title("Demo de la API Gemini con Streamlit")
 
-    # Get user input prompt
-    prompt = st.text_area("Enter your prompt:")
+    # Obtener entrada de texto del usuario
+    prompt = st.text_area("Ingresa tu indicación:")
 
-    # Get optional image input
-    image_file = st.file_uploader("Upload an image (if applicable):", type=["jpg", "jpeg", "png"])
+    # Obtener entrada opcional de imagen
+    imagen_archivo = st.file_uploader("Subir una imagen (si aplica):", type=["jpg", "jpeg", "png"])
 
-    # Display image if provided
-    if image_file:
-        st.image(image_file, caption="Uploaded Image", use_column_width=True)
+    # Mostrar imagen si se proporciona
+    if imagen_archivo:
+        st.image(imagen_archivo, caption="Imagen Subida", use_column_width=True)
 
-    # Generate content on button click
-    if st.button("Generate Content"):
-        st.markdown("### Generated Content:")
-        if not image_file:
-            st.warning("Please provide an image for the gemini-pro-vision model.")
+    # Generar contenido al hacer clic en el botón
+    if st.button("Generar Contenido"):
+        st.markdown("### Contenido Generado:")
+        if not imagen_archivo:
+            st.warning("Por favor, proporciona una imagen para el modelo gemini-pro-vision.")
         else:
-            image = Image.open(image_file)
-            response = generate_gemini_content(prompt, model_name=model_name, image=image)
-
-            # Display the generated content in Markdown format if response is available
-            if response:
-                if response.candidates:
-                    parts = response.candidates[0].content.parts
-                    generated_text = parts[0].text if parts else "No content generated."
-                    st.markdown(to_markdown(generated_text))
+            imagen = Image.open(imagen_archivo)
+            respuesta = generar_contenido_gemini(prompt, imagen=imagen)
+
+            # Mostrar el contenido generado en formato Markdown si hay respuesta disponible
+            if respuesta:
+                if respuesta.candidates:
+                    partes = respuesta.candidates[0].content.parts
+                    texto_generado = partes[0].text if partes else "No se generó contenido."
+                    st.markdown(to_markdown(texto_generado))
                 else:
-                    st.warning("No candidates found in the response.")
+                    st.warning("No se encontraron candidatos en la respuesta.")
 
 if __name__ == "__main__":
     main()