QnaChatApp.py

from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai

load_dotenv()

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

model = genai.GenerativeModel("gemini-pro")
chat = model.start_chat(history=[])


def get_gemini_response(question):
    response = chat.send_message(question, stream=True)
    return response


st.header("Gemini Q&A Application")

if "chat_history" not in st.session_state:
    st.session_state["chat_history"] = []

input = st.text_input("input")
submit = st.button("Ask")

if submit and input:
    response = get_gemini_response(input)
    st.session_state["chat_history"].append(("You", input))
    st.subheader("Response")
    for chunk in response:
        st.write(chunk.text)
        st.session_state["chat_history"].append(("Bot", chunk.text))

st.sidebar.subheader('Chat history is')

for role,text in st.session_state['chat_history']:
    st.sidebar.write(f'{role}:{text}')