File size: 1,166 Bytes
27e6630
 
179361d
27e6630
9910bb3
27e6630
 
9910bb3
 
 
 
 
 
 
 
179361d
9910bb3
179361d
9910bb3
 
 
 
 
 
 
27e6630
9910bb3
 
 
27e6630
 
9910bb3
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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.")