Enhance image analysis and generation with improved error handling, format normalization, and user guidance

app_pages/image_analyser.py

@@ -48,6 +48,7 @@ def render_image_analyser_page(uid: str) -> None:
         "Upload image",
         type=["jpg", "jpeg", "png", "webp"],
         key="ia_upload",
+        help="Supported formats: JPG, JPEG, PNG, WEBP"
     )
 
     img_bytes, img_ext, file_name, thumb_b64 = None, "jpg", None, ""
@@ -57,6 +58,9 @@ def render_image_analyser_page(uid: str) -> None:
         file_name = uploaded.name
         if "." in file_name:
             img_ext = file_name.rsplit(".", 1)[-1].lower()
+            # Normalize jpg to jpeg for consistency
+            if img_ext == "jpg":
+                img_ext = "jpeg"
 
         thumb_b64 = base64.b64encode(img_bytes).decode("utf-8")
         st.caption("Preview")
@@ -78,12 +82,28 @@ def render_image_analyser_page(uid: str) -> None:
             try:
                 with st.spinner("Analyzing image..."):
                     result = get_analysis(img_bytes, img_ext)
+                
+                # Validate result to ensure no N/A values
+                if result and isinstance(result, dict):
+                    # Check for common N/A patterns and replace them
+                    def clean_na_values(obj):
+                        if isinstance(obj, dict):
+                            return {k: clean_na_values(v) for k, v in obj.items()}
+                        elif isinstance(obj, list):
+                            return [clean_na_values(item) for item in obj]
+                        elif isinstance(obj, str) and obj.upper() in ['N/A', 'NOT AVAILABLE', 'NA']:
+                            return "Analysis not available - please try with a clearer image"
+                        return obj
+                    
+                    result = clean_na_values(result)
+                
                 st.session_state["analysis_result"] = result
                 st.session_state["file_name"] = file_name
                 st.session_state["thumb_b64"] = thumb_b64
                 st.session_state["category"] = category
             except Exception as e:
                 st.error(f"Analysis failed: {e}")
+                st.error("Please try uploading a different image or check if the image format is supported.")
 
 
 

app_pages/image_generation.py

@@ -138,14 +138,18 @@ def render_bulk_image_generator(uid: str, prefix: str = "ig_img"):
 
     if demo_btn or gen_all_btn:
         if zip_file and category and user_prompt:
+            # Clear previous images when starting new generation
+            st.session_state[gallery_key] = []
+            
             with st.spinner(" Generating image variations..."):
                 temp_dir = tempfile.mkdtemp()
                 try:
-                    temp_path = os.path.join(
-                        temp_dir,
-                        zip_file.name if zip_file.name.lower().endswith(".zip")
-                        else f"input.{zip_file.name.split('.')[-1]}"
-                    )
+                    # Use original filename for single files, or generate unique name for zip
+                    if zip_file.name.lower().endswith(".zip"):
+                        temp_path = os.path.join(temp_dir, zip_file.name)
+                    else:
+                        # Use original filename to preserve uniqueness
+                        temp_path = os.path.join(temp_dir, zip_file.name)
                     with open(temp_path, "wb") as f:
                         f.write(zip_file.read())
 
@@ -161,7 +165,7 @@ def render_bulk_image_generator(uid: str, prefix: str = "ig_img"):
                             platform,
                             num_images,
                             demo_btn,
-                            gallery,
+                            None,  # Pass None instead of gallery to avoid appending
                             blur,
                             uid,
                         )
@@ -176,7 +180,7 @@ def render_bulk_image_generator(uid: str, prefix: str = "ig_img"):
                             platform,
                             num_images,
                             demo_btn,
-                            gallery,
+                            None,  # Pass None instead of gallery to avoid appending
                             blur,
                             uid,
                             selected_model_key,

generator_function/image_processor.py

@@ -46,7 +46,8 @@ def process_zip_and_generate_images(
         for u in all_urls:
             if u not in seen:
                 seen.add(u); deduped.append(u)
-        return (existing_images or []) + deduped
+        # Return only new images, not appended to existing ones
+        return deduped
     except Exception:
         logger.exception(f"Global error during processing file: {zip_path}")
         return existing_images or []

generator_function/multimodel_image_processor.py

@@ -55,7 +55,8 @@ def process_zip_and_generate_images_multimodel(
         for u in all_urls:
             if u not in seen:
                 seen.add(u); deduped.append(u)
-        return (existing_images or []) + deduped
+        # Return only new images, not appended to existing ones
+        return deduped
     except Exception:
         logger.exception(f"Global error during processing file: {zip_path}")
         return existing_images or []

prompt/image_analyzer_prompt.py

@@ -1,11 +1,15 @@
 SYSTEM_PROMPT = """
-You are a senior performance-creative analyst who analyze a SINGLE image ad. Do not include explanations, markdown, or extra keys.
+You are a senior performance-creative analyst who analyzes a SINGLE image ad. Do not include explanations, markdown, or extra keys.
 Rules:
 - Be precise and concise. Avoid generic fluff.
 - When unsure, infer cautiously and note uncertainty.
 - Explain "why it matters" in business terms (CTR, CPC, CVR, RPM, bounce, fatigue, policy).
 - Do not invent medical claims or promises; flag policy risks instead.
 - Keep color names human-readable (e.g., "pure orange", "purple", "yellow").
+- NEVER use "N/A" or "Not Available" - always provide specific analysis based on what you can observe.
+- If you cannot determine a specific value, provide your best professional assessment based on visual cues.
+- For text analysis, if text is not clearly visible, state "Text not clearly visible" rather than "N/A".
+- For color analysis, describe what you can see rather than using "N/A".
 """
 
 USER_PROMPT_TEMPLATE = """
@@ -22,4 +26,8 @@ Task: Analyze the attached image ad and produce a structured breakdown:
 Constraints:
 - Use crisp, decision-ready language.
 - If text is in Spanish or another language, keep your summaries in English but preserve original CTA text.
+- Provide specific, actionable insights for each section.
+- Avoid using "N/A", "Not Available", or similar placeholder text.
+- If information is not clearly visible, describe what you can observe and make reasonable inferences.
+- Focus on actionable insights that can improve ad performance.
 """