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

# Load environment variables (containing your Gemini API key)
load_dotenv()
API_KEY = os.getenv('GEMINI_API_KEY')

# Initialize Gemini client
client = genai.Gemini(API_KEY)

# App Title
st.title("AI Poem Generator")

# User input
subject = st.text_input("Enter the subject of your poem:")

# Button to trigger poem generation
if st.button("Generate Poem"):
    if not subject:
        st.warning("Please enter a subject")
    else:
        # Construct prompt for Gemini
        prompt = f"Write a creative and interesting poem about {subject}"

        # Generate poem using Gemini
        response = client.generate_text(prompt)
        poem = response.text

        # Display the poem
        st.markdown(poem)