import streamlit as st
from PIL import Image
from transformers import pipeline

from google.colab import drive
drive.mount('/content/drive')

ruta_google_drive = "/content/drive/MyDrive/Lenguaje/"

translator = pipeline("translation", model="google/mbt-youtube-100k")

letras = ['A', 'B', 'C']
categorias = ['Frutas', 'Colores', 'Familiares', 'Animales']

st.title("Proyecto de Lenguaje de Señas")

opcion = st.sidebar.selectbox("Seleccionar una opción", ["Letras", "Categorías"])


if opcion == "Letras":
    letra = st.selectbox("Seleccionar una letra", letras)
    st.write(f"Lenguaje de señas para la letra {letra}")
    st.image(f"ruta_a_tu_google_drive/A.jpg")

elif opcion == "Categorías":
    categoria = st.selectbox("Seleccionar una categoría", categorias)
    st.write(f"Lenguaje de señas para la categoría {categoria}")
    st.image(f"ruta_a_tu_google_drive/{categoria}.jpg")