Update app.py

app.py

@@ -3,7 +3,7 @@ from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
 import torch
 
-# Load model and processor
+# Load model
 processor = BlipProcessor.from_pretrained(
     "Salesforce/blip-image-captioning-base"
 )
@@ -11,7 +11,7 @@ model = BlipForConditionalGeneration.from_pretrained(
     "Salesforce/blip-image-captioning-base"
 )
 
-def generate_alt_text(image: Image.Image):
+def generate_alt_text(image: Image.Image, seo_mode: bool):
     if image is None:
         return ""
 
@@ -20,15 +20,21 @@ def generate_alt_text(image: Image.Image):
     with torch.no_grad():
         output = model.generate(
             **inputs,
-            max_new_tokens=50
+            max_new_tokens=60 if seo_mode else 40
         )
 
     caption = processor.decode(
         output[0],
         skip_special_tokens=True
-    )
+    ).strip()
 
-    alt_text = f"Alt text: {caption.capitalize()}."
+    if seo_mode:
+        alt_text = (
+            f"{caption.capitalize()}, high-quality image suitable for websites, "
+            f"blogs, and SEO-optimized content."
+        )
+    else:
+        alt_text = caption.capitalize() + "."
 
     return alt_text
 
@@ -36,34 +42,37 @@ def generate_alt_text(image: Image.Image):
 with gr.Blocks(title="Alt Text Generator") as demo:
     gr.Markdown("""
     # 🖼️ Alt Text Generator
-    Upload an image and instantly generate SEO-friendly and accessibility-ready alt text.
+    Generate accessibility-friendly and SEO-optimized alt text for images.
     """)
 
-    with gr.Row():
-        image_input = gr.Image(
-            type="pil",
-            label="Upload Image"
-        )
+    image_input = gr.Image(
+        type="pil",
+        label="Upload Image"
+    )
 
-        alt_text_output = gr.Textbox(
-            label="Generated Alt Text",
-            lines=4,
-            placeholder="Your alt text will appear here..."
-        )
+    seo_toggle = gr.Checkbox(
+        label="SEO Mode (longer, keyword-friendly alt text)",
+        value=False
+    )
+
+    alt_text_output = gr.Textbox(
+        label="Generated Alt Text",
+        lines=4,
+        placeholder="Your alt text will appear here..."
+    )
 
     generate_btn = gr.Button("Generate Alt Text 🚀")
 
     generate_btn.click(
         fn=generate_alt_text,
-        inputs=image_input,
+        inputs=[image_input, seo_toggle],
         outputs=alt_text_output
     )
 
     gr.Markdown("""
-    ### ✅ How to use
-    - Upload an image
-    - Click **Generate Alt Text**
-    - Copy & paste into your website's `<img alt="">` attribute
+    ### ℹ️ Tips
+    - **SEO Mode ON** → Best for blog posts, landing pages, e-commerce
+    - **SEO Mode OFF** → Best for pure accessibility & clean HTML
     """)
 
 demo.launch()