Add more logging

app.py

@@ -281,7 +281,10 @@ def analyze_and_generate_french_style(image):
         return None, "Model not loaded", ""
     
     try:
+        progress_log = "🔄 **Step 1/4:** Starting flower analysis...\n\n"
+        
         # Identify flower
+        progress_log += "🔍 Identifying flower type using AI model...\n"
         results = zs_classifier(
             image,
             candidate_labels=FLOWER_LABELS,
@@ -290,19 +293,28 @@ def analyze_and_generate_french_style(image):
         
         top_flower = results[0]["label"] if results else "flower"
         confidence = results[0]["score"] if results else 0
+        progress_log += f"✅ Identified: **{top_flower}** (confidence: {confidence:.2%})\n\n"
         
         # Extract dominant colors
+        progress_log += "🔄 **Step 2/4:** Analyzing color palette...\n\n"
+        progress_log += "🎨 Extracting dominant colors from image...\n"
         color_names, color_rgb = extract_dominant_colors(image, num_colors=3)
         
         # Create color description
         main_colors = color_names[:3]  # Top 3 colors
         color_desc = ", ".join(main_colors)
+        progress_log += f"✅ Color palette: **{color_desc}**\n\n"
         
         # Generate French-style prompt
+        progress_log += "🔄 **Step 3/4:** Creating French-style arrangement prompt...\n\n"
         prompt = f"elegant French-style floral arrangement featuring {top_flower}s in {color_desc} colors, displayed in a clear crystal vase on a marble kitchen countertop, soft natural lighting, minimalist French country kitchen background, professional photography, sophisticated composition"
+        progress_log += f"✅ Prompt created: *{prompt[:100]}...*\n\n"
         
         # Generate the image
+        progress_log += "🔄 **Step 4/4:** Generating French-style arrangement image...\n\n"
+        progress_log += "🖼️ Using AI image generation (SDXL-Turbo)...\n"
         generated_image = generate(prompt, steps=4, width=1024, height=1024, seed=-1)
+        progress_log += "✅ French-style arrangement generated successfully!\n\n"
         
         # Create analysis summary
         analysis = f"""
@@ -312,12 +324,20 @@ def analyze_and_generate_french_style(image):
 
 **🇫🇷 Generated Prompt:**
 "{prompt}"
+
+---
+
+**📋 Process Log:**
+{progress_log}
 """
         
         return generated_image, "✅ Analysis complete! French-style arrangement generated.", analysis
         
     except Exception as e:
-        return None, f"❌ Error: {str(e)}", ""
+        error_log = f"❌ **Error occurred during processing:**\n\n{str(e)}\n\n"
+        if 'progress_log' in locals():
+            error_log += f"**Progress before error:**\n{progress_log}"
+        return None, f"❌ Error: {str(e)}", error_log
 
 # ---------- UI ----------
 with gr.Blocks() as demo:
@@ -403,8 +423,14 @@ with gr.Blocks() as demo:
     load_model_btn.click(load_trained_model, inputs=[model_dropdown], outputs=[model_status])
     train_btn.click(start_training, inputs=[epochs, batch_size, learning_rate], outputs=[training_output])
     
-    # French Style tab events
+    # French Style tab events - update status during processing
+    def update_french_status():
+        return "🔄 Processing... Please wait while we analyze your flower image...", ""
+    
     analyze_btn.click(
+        update_french_status,
+        outputs=[french_status, analysis_details]
+    ).then(
         analyze_and_generate_french_style, 
         inputs=[upload_img], 
         outputs=[french_result, french_status, analysis_details]