Update app.py

app.py

@@ -37,26 +37,6 @@ IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 CHAT_HISTORY = List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]
 
-# Estado para almacenar índices votados
-INDEX_STATE = []
-
-# Función para votar
-def vote(tmp, index_state, data: gr.LikeData):
-    """Procesa y guarda las interacciones de retroalimentación."""
-    value_new = data.value
-    index_new = data.index
-    if len(index_state) == 0:
-        index_state.append(index_new)
-    else:
-        if index_new in index_state:
-            return "Your feedback is already saved", index_state
-        else:
-            index_state.append(index_new)
-    return (
-        f"Feedback saved: Value={data.value}, Index={data.index}, Liked={data.liked}, State={index_state}",
-        index_state,
-    )
-
 # Función para preprocesar una imagen
 def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
     """Redimensiona una imagen manteniendo la relación de aspecto."""
@@ -125,30 +105,23 @@ def response(message, history):
         time.sleep(0.01)
         yield response.text[: i + 1]
 
-# Crear la interfaz de usuario con retroalimentación
-demo = gr.Blocks()
+# Función para manejar votos
+def vote(data: gr.LikeData):
+    """Procesa la retroalimentación del usuario (like/dislike)."""
+    if data.liked:
+        print(f"Respuesta aprobada: {data.value['value']}")
+    else:
+        print(f"Respuesta desaprobada: {data.value['value']}")
 
-with demo:
-    with gr.Row():
-        gr.Markdown("## Chat con Gemini + Votación")
+# Crear la interfaz de usuario
+with gr.Blocks() as demo:
     chatbot = gr.Chatbot(label="Chat con Gemini")
-    with gr.Row():
-        message = gr.Textbox(label="Mensaje")
-        send_button = gr.Button("Enviar")
-    with gr.Row():
-        like_button = gr.LikeButton()
-        feedback_output = gr.Textbox(label="Retroalimentación", interactive=False)
+    chatbot.like(vote, None, None)  # Asociar la función `vote` al botón de like/dislike
     
-    # Acciones
-    send_button.click(
+    gr.ChatInterface(
         fn=response,
-        inputs=[message, chatbot],
-        outputs=chatbot,
-    )
-    like_button.click(
-        fn=vote,
-        inputs=[None, INDEX_STATE, like_button],
-        outputs=[feedback_output, INDEX_STATE],
+        type="message",
+        chatbot=chatbot,
     )
 
 # Lanzar la aplicación