File size: 814 Bytes
97d0f66
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
29
30
31
32
33
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)