Update app.py

app.py

@@ -1,28 +1,37 @@
 import gradio as gr
+import cv2
 import numpy as np
-import matplotlib.pyplot as plt
 
-def draw_rectangle(color, x, y, width, height):
-    fig, ax = plt.subplots()
-    rect = plt.Rectangle((x, y), width, height, color=color)
-    ax.add_patch(rect)
-    ax.set_xlim(0, 10)
-    ax.set_ylim(0, 10)
-    plt.axis('off')
-    plt.close(fig)
-    return fig
+# Función para dibujar rectángulos en la imagen
+def draw_rectangle(img, x1, y1, x2, y2, color):
+    # Convertir los valores a enteros
+    x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
+    # Dibujar el rectángulo en la imagen
+    cv2.rectangle(img, (x1, y1), (x2, y2), color, -1)  # -1 para rellenar el rectángulo
 
+# Función principal para la interfaz
+def paint_app(color, x1, y1, x2, y2):
+    # Crear una imagen en blanco
+    img = np.zeros((400, 400, 3), dtype=np.uint8)
+    # Dibujar el rectángulo
+    draw_rectangle(img, x1, y1, x2, y2, color)
+    return img
+
+# Definir los colores disponibles
+colors = ["Red", "Green", "Blue", "Yellow", "White"]
+
+# Crear la interfaz de Gradio
 iface = gr.Interface(
-    fn=draw_rectangle,
+    fn=paint_app,
     inputs=[
-        gr.ColorPicker(label="Color del Rectángulo"),
-        gr.Slider(0, 10, 1, label="Posición X"),
-        gr.Slider(0, 10, 1, label="Posición Y"),
-        gr.Slider(0, 10, 1, label="Ancho"),
-        gr.Slider(0, 10, 1, label="Alto")
+        gr.Slider(minimum=0, maximum=399, step=1, label="X1"),  # Coordenada X1 del rectángulo
+        gr.Slider(minimum=0, maximum=399, step=1, label="Y1"),  # Coordenada Y1 del rectángulo
+        gr.Slider(minimum=0, maximum=399, step=1, label="X2"),  # Coordenada X2 del rectángulo
+        gr.Slider(minimum=0, maximum=399, step=1, label="Y2"),  # Coordenada Y2 del rectángulo
+        gr.Dropdown(colors, label="Color")  # Selección de color
     ],
-    outputs="plot",
-    live=True
+    outputs=gr.Image(width=400, height=400)
 )
 
+# Ejecutar la interfaz
 iface.launch()