Chat_Bot_using_LangChain / pages /2_Non-Interactive Chat Bot.py
AjayKr09's picture
Update pages/2_Non-Interactive Chat Bot.py
ab4aae3 verified
raw
history blame contribute delete
948 Bytes
import streamlit as st
import os
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_core.prompts import ChatPromptTemplate
# Set your Google API key from environment variables
gemini_api_key = os.getenv("GOOGLE_API_KEY")
# Initialize the Gemini API-powered chatbot model
model_gemini = ChatGoogleGenerativeAI(
model='gemini-pro',
temperature=0,
max_output_tokens=500,
convert_system_message_to_human=True
)
# Create a chat prompt template
prompt = ChatPromptTemplate.from_template("{content}")
chain = prompt | model_gemini
# Streamlit app title
st.title("Non-Interactive Chat Bot")
# Text area for user input
input_text = st.text_area('Ask your Question :', height=100)
# Button to submit user query
if st.button('Submit'):
# Invoke the chain with user input text
response = chain.invoke({'content': input_text})
# Display the chatbot response
st.write(response.content)