riabayonaor commited on
Commit
cb6994c
1 Parent(s): 0cd019f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +54 -0
app.py CHANGED
@@ -1,3 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
  import google.generativeai as genai
3
  from PIL import Image
 
1
+ import os
2
+ import streamlit as st
3
+ import google.generativeai as genai
4
+ from PIL import Image
5
+ import io
6
+
7
+ # Configurar la API de Gemini
8
+ api_key = os.getenv('GEMINI_API_KEY')
9
+ genai.configure(api_key=api_key)
10
+
11
+ # Función para generar contenido a partir de texto
12
+ def generate_text(prompt):
13
+ model = genai.GenerativeModel('gemini-1.5-flash')
14
+ response = model.generate_content(prompt)
15
+ return response.text
16
+
17
+ # Función para generar contenido a partir de texto e imagen
18
+ def generate_text_from_image(image, prompt):
19
+ model = genai.GenerativeModel('gemini-1.5-flash')
20
+ response = model.generate_content([prompt, image], stream=True)
21
+ return response.text
22
+
23
+ # Configuración de la aplicación Streamlit
24
+ st.title('Aplicación con Gemini Vision')
25
+ st.write('Esta es una aplicación de demostración utilizando la API de Gemini Vision.')
26
+
27
+ # Cargar imagen
28
+ uploaded_file = st.file_uploader("Elige una imagen", type=["jpg", "jpeg", "png"])
29
+
30
+ if uploaded_file is not None:
31
+ # Mostrar la imagen cargada
32
+ image = Image.open(uploaded_file)
33
+ st.image(image, caption='Imagen Cargada', use_column_width=True)
34
+
35
+ # Texto de entrada del usuario
36
+ prompt = st.text_input("Introduce el texto para generar la respuesta:")
37
+
38
+ if st.button("Generar Respuesta"):
39
+ # Convertir la imagen a bytes
40
+ img_bytes = io.BytesIO()
41
+ image.save(img_bytes, format='JPEG')
42
+ img_bytes = img_bytes.getvalue()
43
+
44
+ # Generar respuesta a partir de la imagen y el texto
45
+ response = generate_text_from_image(img_bytes, prompt)
46
+ st.markdown(response)
47
+ else:
48
+ # Texto de entrada del usuario
49
+ prompt = st.text_input("Introduce el texto para generar la respuesta:")
50
+
51
+ if st.button("Generar Respuesta"):
52
+ # Generar respuesta a partir del texto
53
+ response = generate_text(prompt)
54
+ st.markdown(response)import os
55
  import streamlit as st
56
  import google.generativeai as genai
57
  from PIL import Image