File size: 2,046 Bytes
47c195c 8dffd49 47c195c 8dffd49 194b21f 47c195c 194b21f 47c195c 8dffd49 47c195c 8dffd49 194b21f 47c195c 8dffd49 47c195c 8dffd49 47c195c 8dffd49 |
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import google.generativeai as genai
import streamlit as st
import os
from dotenv import load_dotenv
# Load API key from environment variables
load_dotenv()
api_key = os.getenv("GCP_API_KEY")
if not api_key:
st.error("API key is missing. Set GCP_API_KEY as an environment variable.")
st.stop()
# Configure Gemini API
genai.configure(api_key=api_key)
# Streamlit UI
st.title("AI Text Generator using Gemini API")
st.write("Generate text using Google's Gemini AI model.")
# User Inputs
prompt = st.text_area("Enter your prompt:", "Once upon a time...")
output_format = st.selectbox("Select Output Format:", ["Story", "Poem", "Article", "Code"])
tone = st.selectbox("Select Tone:", ["Formal", "Informal", "Humorous", "Technical"])
creativity = st.slider("Creativity Level (Temperature):", 0.0, 1.0, 0.7)
max_length = st.slider("Max Response Length:", 50, 1000, 500)
n_responses = st.number_input("Number of Responses:", 1, 5, 1)
creative_mode = st.checkbox("Enable Creative Mode")
if st.button("Generate Text"):
if not prompt.strip():
st.error("Please enter a prompt.")
else:
full_prompt = f"{prompt}\n\nFormat: {output_format}\nTone: {tone}\nCreative Mode: {creative_mode}"
try:
model = genai.GenerativeModel("gemini-pro") # Use the latest Gemini model
response = model.generate_content(
full_prompt,
generation_config=genai.types.GenerationConfig(
temperature=creativity, # Now correctly used inside generation_config
max_output_tokens=max_length
)
)
st.subheader("Generated Response")
st.write(response.text)
except Exception as e:
st.error(f"An error occurred: {e}")
# Feedback Section
st.subheader("Feedback")
feedback = st.radio("Was this response helpful?", ("Yes", "No"))
if feedback == "No":
st.text_area("What could be improved?")
st.write("Thank you for using the AI Text Generator!")
|