import streamlit as st

from langchain_openai import ChatOpenAI

from langchain_core.prompts import ChatPromptTemplate

import os 


os.environ["OPENAI_API_KEY"] = os.getenv("k1")

st.title("Translation Application :robot_face:")
rainbow_text = """
<div style="font-size:24px; font-weight:bold;">
    <span style="color:red;">R</span>
    <span style="color:orange;">A</span>
    <span style="color:yellow;">I</span>
    <span style="color:green;">N</span>
    <span style="color:blue;">B</span>
    <span style="color:indigo;">O</span>
    <span style="color:violet;">W</span>
</div>
"""

st.markdown(rainbow_text, unsafe_allow_html=True)

options1 = ["English","Telugu","Hindi"]

st.selectbox("Input Langauge: ",options1)

options2 = ["Hindi","Telugu","English"]

st.selectbox("Output langauge: ",options2)

text = st.text_input("Text Input: ")

if st.button("Submit"):

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

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

    chain = prompt|llm

    re = chain.invoke({"il":options1,"ol":options2, "i":text})

    st.write("Response: ")

    st.write(re.content)