Update app.py

app.py

@@ -2,17 +2,45 @@ import streamlit as st
 from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
 import os
+
 st.title("Translation Using OpenAI")
 
-i=st.text_area("Write your text in english")
-o=st.radio("Select your output laguage",["hindi","telugu","malayalam","tamil","punjabi"])
+# Language options for the dropdown menus
+languages = ["English", "Hindi", "Telugu", "Malayalam", "Tamil", "Punjabi"]
+
+# Dropdown menu for selecting input language
+input_language = st.selectbox("Select your input language", languages)
+
+# Text area for input text
+input_text = st.text_area(f"Write your text in {input_language}")
+
+# Dropdown menu for selecting output language
+output_language = st.selectbox("Select your output language", languages)
+
+# Set the environment variable for the API key
+os.environ['GOOGLE_API_KEY'] = os.getenv("geminiapi_key")
+
+# Initialize the OpenAI language model
+llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0, max_tokens=80)
+
+# Create the prompt template
+prompt = ChatPromptTemplate.from_messages([
+    ("system", "You are a good assistant for translating from {il} to {ol}."),
+    ("human", "{i}")
+])
+
+# Define the translation chain
+chain = prompt | llm
 
+# Function to perform the translation
+def translate(input_text, input_language, output_language):
+    response = chain.invoke({"il": input_language, "ol": output_language, "i": input_text})
+    return response.content
 
-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 translator from {il} to {ol}"),
-("human","{i}")])
-chain=prompt|llm
-re=chain.invoke({"il":"English","ol":o,"i":i})
+# Submit button
 if st.button("Submit"):
-    st.write(re.content)
+    if input_text:
+        translation = translate(input_text, input_language, output_language)
+        st.write("Translated Text:", translation)
+    else:
+        st.write("Please enter text to translate.")