78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
import streamlit as st
|
|
import base64
|
|
import requests
|
|
from PIL import Image
|
|
import os
|
|
|
|
|
|
# Function to encode the image
|
|
def encode_image(image_path):
|
|
with open(image_path, "rb") as image_file:
|
|
return base64.b64encode(image_file.read()).decode('utf-8')
|
|
|
|
def get_image_description(image_path):
|
|
base64_image = encode_image(image_path)
|
|
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"Authorization": f"Bearer {os.getenv('OPENAI_API_KEY')}"
|
|
}
|
|
|
|
payload = {
|
|
"model": "gpt-4o-mini",
|
|
"messages": [
|
|
{
|
|
"role": "user",
|
|
"content": [
|
|
{
|
|
"type": "text",
|
|
"text": """You are an SEO expert specializing in writing optimized Alt text for images.
|
|
Your goal is to create clear, descriptive, and concise Alt text that accurately represents
|
|
the content and context of the given image. Make sure your response is optimized for search engines and accessibility."""
|
|
},
|
|
{
|
|
"type": "image_url",
|
|
"image_url": {
|
|
"url": f"data:image/jpeg;base64,{base64_image}"
|
|
}
|
|
}
|
|
]
|
|
}
|
|
],
|
|
"max_tokens": 300
|
|
}
|
|
|
|
response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
|
|
response_data = response.json()
|
|
|
|
# Extract the content field from the response
|
|
content = response_data['choices'][0]['message']['content']
|
|
return content
|
|
|
|
|
|
def alt_text_gen():
|
|
st.title("Image Description Generator")
|
|
|
|
image_path = st.text_input("Enter the full path of the image file")
|
|
|
|
if image_path:
|
|
if os.path.exists(image_path) and image_path.lower().endswith(('jpg', 'jpeg', 'png')):
|
|
try:
|
|
image = Image.open(image_path)
|
|
st.image(image, caption='Uploaded Image', use_column_width=True)
|
|
|
|
if st.button("Get Image Alt Text"):
|
|
with st.spinner("Generating Alt Text..."):
|
|
try:
|
|
description = get_image_description(image_path)
|
|
st.success("Alt Text generated successfully!")
|
|
st.write("ALt Text:", description)
|
|
except Exception as e:
|
|
st.error(f"Error generating description: {e}")
|
|
except Exception as e:
|
|
st.error(f"Error processing image: {e}")
|
|
else:
|
|
st.error("Please enter a valid image file path ending with .jpg, .jpeg, or .png")
|
|
else:
|
|
st.info("Please enter the full path of an image file.")
|