import streamlit as st
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI
import os

st.title("Translation Using OpenAI")

languages = ["English", "Hindi", "Telugu", "Malayalam", "Tamil", "Punjabi"]

input_language = st.selectbox("Select your input language", languages)

input_text = st.text_area(f"Write your text in {input_language}")

output_language = st.selectbox("Select your output language", languages)

os.environ['OPENAI_API_KEY'] = os.getenv("openai_key")

llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0, max_tokens=80)

prompt = ChatPromptTemplate.from_messages([
    ("system", "You are a good assistant for translating from {il} to {ol}."),
    ("human", "{i}")
])

chain = prompt | llm

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

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