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