Chat-bot / app.py
Bhagyajoshi's picture
Update app.py
911a99a verified
raw
history blame contribute delete
833 Bytes
import keyfile
import streamlit as st
import google.generativeai as genai
genai.configure(api_key=keyfile.GEMINI_API_KEY)
if "history" not in st.session_state:
st.session_state["history"] = []
def get_response(question):
model = genai.GenerativeModel("gemini-1.5-flash")
response = model.generate_content(question)
return response.text
st.set_page_config(page_title="Chat Bot", page_icon=":robot:")
st.header("Chat Application")
user_input = st.text_input("Enter your question:")
if st.button("Ask!"):
if user_input:
response = get_response(user_input)
st.session_state["history"].append({"user": user_input, "bot": response})
st.session_state["user_input"] = ""
for chat in st.session_state["history"]:
st.write(f"You: {chat['user']}")
st.write(f"Answer: {chat['bot']}")