yashasgupta's picture
Update app.py
449267e verified
raw
history blame
806 Bytes
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.header("Translation Application")
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)