Spaces:
Sleeping
Sleeping
| import os | |
| os.environ["XDG_CONFIG_HOME"] = os.path.join(os.getcwd(), ".config") | |
| import streamlit as st | |
| import pandas as pd | |
| import openai | |
| import time | |
| # --- CONFIG --- | |
| st.set_page_config(page_title="🖼️ Alt Text Generator", layout="wide") | |
| VALID_IMAGE_EXTENSIONS = [".jpg", ".jpeg", ".png", ".webp", ".gif", ".bmp"] | |
| st.title("🖼️ Alt Text Generator") | |
| st.markdown("Generate SEO-friendly alt text for image URLs using GPT-4o Vision.\nMade by [Florian Potier](https://twitter.com/FloPots)") | |
| # --- INPUT API KEY --- | |
| api_key = st.text_input("🔑 Enter your OpenAI API key", type="password") | |
| if not api_key: | |
| st.stop() | |
| client = openai.OpenAI(api_key=api_key) | |
| # --- INPUT FILE --- | |
| uploaded_file = st.file_uploader("📁 Upload a CSV with image URLs", type=["csv"]) | |
| if uploaded_file: | |
| df = pd.read_csv(uploaded_file) | |
| st.success("✅ File uploaded successfully") | |
| st.dataframe(df.head()) | |
| # --- COLUMN SELECT --- | |
| image_col = st.selectbox("🧭 Select the column with image URLs", df.columns) | |
| # --- PROMPT INPUT --- | |
| prompt_instruction = st.text_area("📝 Enter the prompt to generate alt text", | |
| value="Write a concise, SEO-friendly alt text for the image below. Do not use brand names unless visible.") | |
| def is_valid_image_url(url: str): | |
| return any(url.lower().endswith(ext) for ext in VALID_IMAGE_EXTENSIONS) | |
| df["Valid Image"] = df[image_col].astype(str).apply(is_valid_image_url) | |
| valid_df = df[df["Valid Image"] == True] | |
| if valid_df.empty: | |
| st.error("❌ No valid image URLs found. Make sure URLs end with .jpg, .png, .webp, etc.") | |
| st.stop() | |
| st.info(f"🔍 {len(valid_df)} valid image URLs found. Invalid ones will be skipped.") | |
| # --- PROCESS --- | |
| if st.button("🚀 Start Processing"): | |
| progress = st.progress(0) | |
| results = [] | |
| urls = valid_df[image_col].tolist() | |
| for idx, url in enumerate(urls): | |
| try: | |
| response = client.chat.completions.create( | |
| model="gpt-4o", | |
| messages=[ | |
| { | |
| "role": "user", | |
| "content": [ | |
| {"type": "text", "text": prompt_instruction}, | |
| {"type": "image_url", "image_url": {"url": url}} | |
| ] | |
| } | |
| ] | |
| ) | |
| alt_text = response.choices[0].message.content.strip() | |
| except Exception as e: | |
| alt_text = f"ERROR: {e}" | |
| results.append(alt_text) | |
| progress.progress((idx + 1) / len(urls)) | |
| time.sleep(1) | |
| valid_df["Generated Alt Text"] = results | |
| output_df = df.copy() | |
| output_df.loc[valid_df.index, "Generated Alt Text"] = valid_df["Generated Alt Text"] | |
| st.success("✅ Done! Preview of the results:") | |
| st.dataframe(output_df[[image_col, "Generated Alt Text"]].head()) | |
| csv = output_df.to_csv(index=False).encode("utf-8") | |
| st.download_button("📥 Download CSV with Alt Text", csv, "alt_text_output.csv", "text/csv") | |