Update app.py

app.py

@@ -1,41 +1,53 @@
 import streamlit as st
 from langchain_openai import ChatOpenAI
 import os
-from langchain_core.prompts import ChatPromptTemplate
+from langchain.prompts import ChatPromptTemplate
 
+# Set the OpenAI API key
 os.environ['OPENAI_API_KEY'] = os.getenv('api_key')
 
+# Initialize the OpenAI model
 llm = ChatOpenAI(model='gpt-3.5-turbo', temperature=0, max_tokens=150)
 
+# Streamlit app header
 st.header("Translation Application")
 
+# Input language selection
 il = st.selectbox(
-    "Please select a Language?",
-    ("Hindi", "English", "Bengali"))
-
+    "Please select the input language",
+    ("Hindi", "English", "Bengali")
+)
 st.write("You selected:", il)
 
+# Output language selection
 ol = st.selectbox(
-    "Please select a Language?",
-    ("Hindi", "English", "Bengali"))
-
-st.write("You selected:", ol)
-
-prompt = ChatPromptTemplate.from_messages(
-    [('system', 'you are a good assistant for translation to {il} to {ol}'),
-     'user', '{i}']
+    "Please select the output language",
+    ("Hindi", "English", "Bengali")
 )
+st.write("You selected:", ol)
 
+# Input text area
 input_text = st.text_area('Input Text', height=300)
 
-chain = prompt|llm
+# Define the translation prompt
+prompt = ChatPromptTemplate.from_messages(
+    [('system', 'You are a good assistant for translation from {il} to {ol}'),
+     ('user', '{i}')]
+)
 
-re = chain.invoke({
-    'il':il, 'ol': ol, 'i': input_text
-})
+# Create the chain
+chain = prompt | llm
 
+# Handle the submit button
 if st.button('Submit'):
     if input_text:
-        st.write(re.content)
+        # Invoke the chain with input parameters
+        response = chain.invoke({
+            'il': il,
+            'ol': ol,
+            'i': input_text
+        })
+        # Display the translated text
+        st.write(response.content)
     else:
-        st.write('Please enter correct text to translate.')
+        st.write('Please enter text to translate.')