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.")