import streamlit as st
from transformers import AutoProcessor, SeamlessM4TModel

st.title("Ed's new Hot Dog? Or Not!!!!!?")

processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-large")

model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")

if "texttotranslate" not in st.session_state:
    st.session_state.texttotranslate = ""

def submit():
	st.write('method')
	mytitle  =  st.session_state.texttotranslate
	text_inputs = processor(text = mytitle, src_lang="eng", return_tensors="pt")
	output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
	translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
	st.write(translated_text_from_text)


title = "my dog is fat"
title = st.text_input('hello', "fat cats", on_change=submit)



text_inputs = processor(text = title, src_lang="eng", return_tensors="pt")

# from text
output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)

st.write(translated_text_from_text)
st.write("fool me")