import streamlit as st
from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
import os
st.title("Translation Using OpenAI")
input_lang=st.selectbox("Select your input laguage",["Hindi","Telugu","Malayalam","Tamil","Punjabi","English"])
input_text=st.text_area("Write your text")
output_lang=st.selectbox("Select your output laguage",["Hindi","Telugu","Malayalam","Tamil","Punjabi","English"])


os.environ['OPENAI_API_KEY']=os.getenv("open_ai")
llm=ChatOpenAI(model="gpt-3.5-turbo",temperature=0)
prompt=ChatPromptTemplate.from_messages([("system","You are a good assistant as a translator from {il} to {ol}. Please translate the following text accurately and maintain the original meaning and context."),
("human","{i}")])

chain=prompt|llm

def translate(input_text, input_lang, output_lang):
    response = chain.invoke({"il": input_lang, "ol": output_lang, "i": input_text})
    return response.content

if st.button("Submit"):
    if input_text:
        translation = translate(input_text, input_lang, output_lang)
        st.write("Translated Text:", translation)
    else:
        st.write("Please enter text to translate.")