File size: 870 Bytes
fc99cc4 530e0fd fc99cc4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import streamlit as st
import google.generativeai as genai
import os
# Load API key from environment variable
genai.configure(api_key=os.getenv("GOOGLE_GENAI_API_KEY"))
# Title and description
st.title("Gemini AI Chatbot")
st.write("Chatbot powered by Google's Gemini API")
# User input
user_prompt = st.text_area("Enter your question:", height=150)
if st.button("Generate Response"):
if user_prompt.strip():
try:
with st.spinner("Fetching response..."):
response = genai.GenerativeModel("gemini-1.5-flash").generate_content(
"Assume I am Kokila. " + user_prompt
)
st.success("Response:")
st.write(response.text)
except Exception as e:
st.error(f"An error occurred: {str(e)}")
else:
st.warning("Please enter a valid prompt.")
|