Update app.py

app.py

@@ -1,19 +1,11 @@
-# =====================================
-# INSTALL DEPENDENCIES
-# =====================================
-
-# =====================================
-# IMPORTS
-# =====================================
 import gradio as gr
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
 from groq import Groq
-import traceback
 
-# =====================================
-# LOAD IMAGE CAPTION MODEL
-# =====================================
+# ==============================
+# LOAD MODEL (ONCE)
+# ==============================
 processor = BlipProcessor.from_pretrained(
     "Salesforce/blip-image-captioning-base"
 )
@@ -21,34 +13,33 @@ model = BlipForConditionalGeneration.from_pretrained(
     "Salesforce/blip-image-captioning-base"
 )
 
-# =====================================
-# CORE FUNCTION (FIXED)
-# =====================================
+# ==============================
+# CORE FUNCTION
+# ==============================
 def generate_caption(api_key, image, style):
     try:
-        # API KEY CHECK
-        if api_key is None or api_key.strip() == "":
+        if not api_key or api_key.strip() == "":
             return "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"
 
-        # IMAGE CHECK
         if image is None:
             return "❌ Please upload an image."
 
-        # Ensure image is RGB
         if image.mode != "RGB":
             image = image.convert("RGB")
 
-        # Step 1: Generate base caption
+        # Image → basic caption
         inputs = processor(image, return_tensors="pt")
         output = model.generate(**inputs, max_new_tokens=30)
-        basic_caption = processor.decode(output[0], skip_special_tokens=True)
+        basic_caption = processor.decode(
+            output[0], skip_special_tokens=True
+        )
 
-        # Step 2: Call Groq API
+        # Groq refinement
         client = Groq(api_key=api_key)
 
         prompt = f"""
-Rewrite this image caption in a {style.lower()} style.
-Keep it short (1–2 lines) and friendly.
+Rewrite the following image caption in a {style.lower()} style.
+Keep it short (1–2 lines).
 
 Caption:
 "{basic_caption}"
@@ -56,64 +47,51 @@ Caption:
 
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
-            messages=[
-                {"role": "user", "content": prompt}
-            ],
+            messages=[{"role": "user", "content": prompt}],
             temperature=0.7,
         )
 
-        refined_caption = response.choices[0].message.content
+        refined = response.choices[0].message.content
 
         return (
             f"🖼️ **Basic Caption:** {basic_caption}\n\n"
-            f"✨ **AI Refined Caption ({style}):**\n{refined_caption}"
+            f"✨ **AI Refined Caption ({style}):**\n{refined}"
         )
 
     except Exception as e:
-        # Show real error instead of silent fail
-        return f"❌ Error occurred:\n{str(e)}"
+        return f"❌ Error:\n{str(e)}"
 
-# =====================================
-# GRADIO UI
-# =====================================
+# ==============================
+# UI
+# ==============================
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("""
-    # 🖼️ Image Caption Generator App  
-    **Upload image → Generate caption → Refine with Groq AI**  
-    Beginner-friendly • API-based • Fixed & Working ✅
+    # 🖼️ Image Caption Generator  
+    Image → Caption → Groq AI ✨  
+    Hugging Face Deployment
     """)
 
-    api_key_input = gr.Textbox(
+    api_key = gr.Textbox(
         label="🔑 Groq API Key",
         type="password",
-        placeholder="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"
+        placeholder="Paste your Groq API key here"
     )
 
-    image_input = gr.Image(
-        type="pil",
-        label="📷 Upload Image"
-    )
+    image = gr.Image(type="pil", label="📷 Upload Image")
 
-    style_input = gr.Dropdown(
-        choices=["Normal", "Creative", "Fun / Gen-Z"],
+    style = gr.Dropdown(
+        ["Normal", "Creative", "Fun / Gen-Z"],
         value="Normal",
         label="🎨 Caption Style"
     )
 
-    generate_btn = gr.Button("🚀 Generate Caption")
-
-    output_box = gr.Markdown()
+    btn = gr.Button("🚀 Generate Caption")
+    output = gr.Markdown()
 
-    generate_btn.click(
-        fn=generate_caption,
-        inputs=[api_key_input, image_input, style_input],
-        outputs=output_box
+    btn.click(
+        generate_caption,
+        inputs=[api_key, image, style],
+        outputs=output
     )
 
-    gr.Markdown("""
-    ---
-    ⚠️ **Educational AI Project**  
-    Uses Image Captioning + Groq LLM API
-    """)
-
 app.launch()