yashasgupta's picture
Update app.py
d311da2 verified
raw
history blame contribute delete
942 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.title("Translation Application :robot_face:")
options1 = ["English","Telugu","Hindi","French","German","Russian","Spanish"]
input_langauge = st.selectbox("Input Langauge: ",options1)
options2 = ["Hindi","Telugu","Spanish","English","German","Russian","French"]
output_langauge = 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":input_langauge,"ol":output_langauge, "i":text})
st.write("Response: ")
st.write(re.content)