Aaram / app.py
Nikhil0987's picture
Update app.py
de23dfa verified
raw
history blame
654 Bytes
import streamlit as st
import json
import google.generativeai as genai
# Load JSON configuration
with open('config.json', 'r') as f:
config = json.load(f)
GOOGLE_API_KEY="AIzaSyCUBaL7TdISL7lRuBy19_X0-OsZfgbIgEc"
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-pro')
if prompt := st.chat_input("Hi"):
enprom = f"""
Act as a personal assistant, Understand user intent from the point of view for asking a few questions. ask to know more details about {prompt}
:- {prompt}"""
completetion = model.generate_content(enprom)
with st.chat_message("Assistant"):
st.write(completetion.text)