Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import re
 import gradio as gr
 from PIL import Image
@@ -28,7 +30,6 @@ analysis_generator   = make_pipeline("google/flan-t5-small", 500)
 suggestion_generator = make_pipeline("google/flan-t5-small", 500)
 
 # Hardcoded example ads for gallery
-
 def get_recommendations():
     return [
         "https://i.imgur.com/InC88PP.jpeg",
@@ -44,21 +45,17 @@ def get_recommendations():
     ]
 
 # Step A: Use OCRFlux to generate a detailed caption
-
 def generate_caption(image):
     result = image_to_text(image)
-    text = result[0]["generated_text"].strip()
-    return text
+    return result[0]["generated_text"].strip()
 
 # Step B: Flan interprets caption into concise category
-
 def generate_category(caption):
     prompt = f"Caption: {caption}\nProvide a concise category label for this ad."
     raw = category_generator(prompt)[0]["generated_text"].strip()
     return raw.splitlines()[0]
 
 # Step C: Flan produces exactly five-sentence analysis
-
 def generate_analysis(caption):
     prompt = (
         f"Caption: {caption}\n"
@@ -69,23 +66,22 @@ def generate_analysis(caption):
     return " ".join(sentences[:5])
 
 # Step D: Flan suggests five actionable bullet-point improvements
-
 def generate_suggestions(caption):
     prompt = (
         f"Caption: {caption}\n"
-        "Suggest five distinct improvements as bullet points. "
-        "Each line must start with '- '."
+        "Suggest five distinct improvements as bullet points. Each line must start with '- '."
     )
     raw = suggestion_generator(prompt)[0]["generated_text"].strip()
     lines = [l for l in raw.splitlines() if l.strip().startswith('- ')]
     if len(lines) < 5:
         all_lines = [l.strip() for l in raw.splitlines() if l.strip()]
-        lines = [('- ' + all_lines[i]) if not all_lines[i].startswith('- ') else all_lines[i]
-                 for i in range(min(5, len(all_lines)))]
+        lines = [
+            ('- ' + all_lines[i]) if not all_lines[i].startswith('- ') else all_lines[i]
+            for i in range(min(5, len(all_lines)))
+        ]
     return "\n".join(lines[:5])
 
 # Orchestrator: process image through all steps
-
 def process(image):
     caption     = generate_caption(image)
     category    = generate_category(caption)
@@ -98,8 +94,7 @@ def process(image):
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     gr.Markdown("## 📢 Smart Ad Analyzer")
     gr.Markdown(
-        "Upload an image ad to see: an Ad Category, a five-sentence Analysis, "
-        "five bullet-point Suggestions, and Example Ads."
+        "Upload an image ad to see: an Ad Category, a five-sentence Analysis, five bullet-point Suggestions, and Example Ads."
     )
 
     with gr.Row():
@@ -121,4 +116,4 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     gr.Markdown("Made by Simon Thalmay")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()