File size: 4,305 Bytes
e22befe | 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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | import streamlit as st
import google.generativeai as genai
import openai
from apikey import google_gemini_api_key, openai_api_key
from PIL import Image
import requests
from io import BytesIO
import time
# Configure the Gemini API with your API key
genai.configure(api_key=google_gemini_api_key)
# Configure the OpenAI API with your API key
openai.api_key = openai_api_key
# Define generation configuration settings for Gemini
generation_config = {
"temperature": 0.9,
"top_p": 1,
"top_k": 1,
"max_output_tokens": 2048,
}
# Define safety settings to filter out harmful content
safety_settings = [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
]
# Initialize the GenerativeModel with the specified settings
model = genai.GenerativeModel(
model_name="gemini-1.5-pro",
generation_config=generation_config,
safety_settings=safety_settings,
)
# Set the Streamlit app layout to wide mode
st.set_page_config(layout="wide")
# Adding emojis to the title
st.title("✍️🤖 BlogCraft: Your AI Writing Companion")
# Adding emojis to the subheader
st.subheader("Now you can craft perfect blogs with the help of AI—BlogCraft is your new AI Blog Companion. 📝✨")
# Sidebar for user inputs
with st.sidebar:
st.title("Input Your Blog Details")
st.subheader("Enter Details of the Blog you want to generate")
# Blog title input
blog_title = st.text_input("Blog Title")
# Keywords input
keywords = st.text_area("Keywords (comma-separated)")
# Number of words slider
num_words = st.slider("Number of words", min_value=250, max_value=2500, step=250)
# Number of images input
num_images = st.number_input("Number of Images", min_value=0, max_value=10, step=1)
# Generate Blog button
submit_button = st.button("Generate Blog")
# Function to generate images with retry mechanism
def generate_image_with_retry(prompt, max_retries=3, delay=5):
for attempt in range(max_retries):
try:
response = openai.Image.create(
prompt=prompt,
n=1,
size="1024x1024",
)
return response['data'][0]['url']
except Exception as e:
if attempt < max_retries - 1:
time.sleep(delay)
delay *= 2 # Exponential backoff
else:
st.error(f"An error occurred while generating the image: {e}")
return None
# When the user clicks the "Generate Blog" button
if submit_button:
# Ensure that the blog title and keywords are provided
if not blog_title:
st.error("Please enter a blog title.")
elif not keywords:
st.error("Please enter at least one keyword.")
else:
# Create the prompt for the Gemini model
prompt = (
f"Generate a comprehensive, engaging blog post relevant to the given title \"{blog_title}\" "
f"and keywords \"{keywords}\". Make sure to incorporate these keywords in the blog post. "
f"The blog should be approximately {num_words} words in length, suitable for an online audience. "
f"Ensure the content is original, informative, and maintains a consistent tone throughout."
)
# Generate the content using the Gemini model
response = model.generate_content(prompt)
# Display the generated blog content
st.title("Your Blog Post:")
st.write(response.text)
# Generate images using OpenAI's DALL·E 3 model
for i in range(num_images):
image_prompt = f"{blog_title} - {keywords} - Image {i+1}"
image_url = generate_image_with_retry(image_prompt)
if image_url:
response = requests.get(image_url)
image = Image.open(BytesIO(response.content))
st.image(image, caption=f"Generated Image {i+1}")
|