BATUTO-ART commited on
Commit
f666d93
verified
1 Parent(s): acb0ed9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +54 -0
app.py ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import google.generativeai as genai
3
+ import os
4
+ from PIL import Image
5
+
6
+ # Configuraci贸n del mero mero
7
+ # Recuerda meter tu llave en "Settings > Secrets" de Hugging Face
8
+ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
9
+ genai.configure(api_key=GOOGLE_API_KEY)
10
+
11
+ def describir_imagen(input_img):
12
+ if input_img is None:
13
+ return "隆No te me pases de lanza, BATUTO! Sube una imagen primero."
14
+
15
+ try:
16
+ # Usamos gemini-1.5-flash que es r谩pido y efectivo
17
+ model = genai.GenerativeModel('gemini-1.5-flash')
18
+
19
+ # Generamos el contenido
20
+ prompt = "Describe esta imagen de forma detallada y profesional."
21
+ response = model.generate_content([prompt, input_img])
22
+
23
+ return response.text
24
+ except Exception as e:
25
+ return f"Chales, algo sali贸 mal: {str(e)}"
26
+
27
+ # Dise帽o de la interfaz con Gradio
28
+ with gr.Blocks(theme=gr.themes.Soft(primary_hue="amber")) as demo:
29
+ gr.Markdown("# 馃帹 BATUTO-ART Vision")
30
+ gr.Markdown("### El rinc贸n donde las im谩genes hablan. 隆S煤bele algo, mi rey!")
31
+
32
+ with gr.Row():
33
+ with gr.Column(scale=1):
34
+ input_image = gr.Image(type="pil", label="Carga tu imagen aqu铆")
35
+ enviar_btn = gr.Button("Analizar Imagen", variant="primary")
36
+
37
+ with gr.Column(scale=1):
38
+ output_text = gr.Textbox(
39
+ label="Descripci贸n de BATUTO-ART",
40
+ placeholder="Aqu铆 aparecer谩 el chisme de la foto...",
41
+ lines=10
42
+ )
43
+
44
+ enviar_btn.click(
45
+ fn=describir_imagen,
46
+ inputs=input_image,
47
+ outputs=output_text
48
+ )
49
+
50
+ gr.Footer("Desarrollado con flow para BATUTO-ART")
51
+
52
+ if __name__ == "__main__":
53
+ demo.launch()
54
+