SayedShaun's picture
Update app.py
6282d67 verified
raw
history blame contribute delete
856 Bytes
import streamlit as st
import os
import sys
from langchain_google_genai import ChatGoogleGenerativeAI
google_api_key = st.sidebar.text_input("Provide Your Google Gemini API Key")
if not google_api_key:
st.write("Please Provide Your API then Try Again")
sys.exit(0)
def llm_response(question):
llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.5, google_api_key=google_api_key)
response = llm.invoke(question)
return response.content
st.sidebar.title("LangChain QA Chatbot")
query = st.sidebar.text_area("Ask Anything....", height=250)
generate_button = st.sidebar.button("Generate Answer", use_container_width=True)
response = llm_response(query)
if generate_button:
if len(query) == 0:
st.header("Please Ask Your Question.")
else:
st.header("Generated Content")
st.write(response)