Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,12 +4,8 @@ os.environ["XDG_CONFIG_HOME"] = os.path.join(os.getcwd(), ".config")
 import streamlit as st
 import pandas as pd
 import openai
-import re
-import asyncio
-import httpx
 import time
 from io import StringIO
-from typing import List
 
 # --- CONFIG ---
 st.set_page_config(page_title="🖼️ Alt Text Generator", layout="wide")
@@ -22,77 +18,68 @@ st.markdown("Generate SEO-friendly alt text for image URLs using GPT-4o Vision.\
 api_key = st.text_input("🔑 Enter your OpenAI API key", type="password")
 if not api_key:
     st.stop()
-openai_client = openai.AsyncOpenAI(api_key=api_key)
+client = openai.OpenAI(api_key=api_key)
 
 # --- INPUT FILE ---
 uploaded_file = st.file_uploader("📁 Upload a CSV with image URLs", type=["csv"])
-if not uploaded_file:
-    st.stop()
+if uploaded_file:
+    df = pd.read_csv(uploaded_file)
+    st.success("✅ File uploaded successfully")
+    st.dataframe(df.head())
 
-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)
 
-# --- 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.")
 
-# --- 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)
 
-# --- FILTER IMAGES ---
-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]
 
-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()
 
-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()
 
-st.info(f"🔍 {len(valid_df)} valid image URLs found. Invalid ones will be skipped.")
-
-# --- GENERATE ALT TEXT ---
-async def generate_alt_text(url: str, prompt: str):
-    try:
-        response = await openai_client.chat.completions.create(
-            model="gpt-4o",
-            messages=[
-                {
-                    "role": "user",
-                    "content": [
-                        {"type": "text", "text": prompt},
-                        {"type": "image_url", "image_url": {"url": url}}
-                    ]
-                }
-            ]
-        )
-        return response.choices[0].message.content.strip()
-    except Exception as e:
-        return f"ERROR: {e}"
-
-st.markdown("### 🚀 Run Alt Text Generation")
-if st.button("Start Processing"):
-    progress = st.progress(0)
-    results = []
-    urls = valid_df[image_col].tolist()
-
-    async def run_all():
         for idx, url in enumerate(urls):
-            alt_text = await generate_alt_text(url, prompt_instruction)
+            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))
-            await asyncio.sleep(1)  # polite delay
-
-    asyncio.run(run_all())
+            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"]
+        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())
+        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")
+        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")