Fix OptimizedResult field access - support both property-based (real backend) and attribute-based (mock backend) result formats

app.py

@@ -351,7 +351,23 @@ def safe_optimize(seed_prompt, dataset, model, custom_model="", max_iterations=5
             if not result:
                 return False, "Optimization returned no result.", None
             
-            if not hasattr(result, 'optimized_prompt'):
+            # Check for both property-based (real backend) and attribute-based (mock backend)
+            # Try to access the prompt to see if it exists (works for both attributes and properties)
+            has_optimized_prompt = False
+            try:
+                if hasattr(result, 'optimized_prompt'):
+                    # Mock backend - direct attribute
+                    has_optimized_prompt = True
+                elif hasattr(result, 'prompt'):
+                    # Real backend - property-based, try to access it
+                    _ = result.prompt
+                    has_optimized_prompt = True
+                elif hasattr(result, '_result') and hasattr(result._result, 'optimized_prompt'):
+                    has_optimized_prompt = True
+            except Exception:
+                pass
+            
+            if not has_optimized_prompt:
                 return False, "Optimization result is missing required fields.", None
             
             return True, "Success", result
@@ -657,14 +673,47 @@ def run_optimization_flow(seed, dataset, model, custom_model, iter_count, call_c
             if not result:
                 raise gr.Error("Optimization completed but returned no result.")
             
-            if not hasattr(result, 'optimized_prompt'):
+            # Check for both property-based (real backend) and attribute-based (mock backend)
+            # Try to access the prompt to see if it exists (works for both attributes and properties)
+            has_optimized_prompt = False
+            try:
+                if hasattr(result, 'optimized_prompt'):
+                    # Mock backend - direct attribute
+                    has_optimized_prompt = True
+                elif hasattr(result, 'prompt'):
+                    # Real backend - property-based, try to access it
+                    _ = result.prompt
+                    has_optimized_prompt = True
+                elif hasattr(result, '_result') and hasattr(result._result, 'optimized_prompt'):
+                    has_optimized_prompt = True
+            except Exception:
+                pass
+            
+            if not has_optimized_prompt:
                 raise gr.Error("Optimization result is missing required fields.")
             
             # Show results
             try:
-                optimized_prompt = result.optimized_prompt if result.optimized_prompt else ""
-                improvement_metrics = result.improvement_metrics if hasattr(result, 'improvement_metrics') else {}
-                iteration_history = result.iteration_history if hasattr(result, 'iteration_history') else []
+                # Handle both property-based (real backend) and attribute-based (mock backend)
+                if hasattr(result, 'optimized_prompt'):
+                    # Mock backend - direct attribute
+                    optimized_prompt = result.optimized_prompt or ""
+                    improvement_metrics = getattr(result, 'improvement_metrics', {})
+                    iteration_history = getattr(result, 'iteration_history', [])
+                elif hasattr(result, 'prompt'):
+                    # Real backend - property-based
+                    optimized_prompt = result.prompt or ""
+                    improvement_data = result.improvement_data if hasattr(result, 'improvement_data') else {}
+                    # improvement_metrics might be in improvement_data or as separate field
+                    improvement_metrics = improvement_data if isinstance(improvement_data, dict) else {}
+                    # For iteration_history, check if it exists in improvement_data or as separate attribute
+                    iteration_history = getattr(result, 'iteration_history', [])
+                    if not iteration_history and isinstance(improvement_data, dict):
+                        iteration_history = improvement_data.get('iteration_history', [])
+                else:
+                    optimized_prompt = ""
+                    improvement_metrics = {}
+                    iteration_history = []
                 
                 history_text = "\n".join(iteration_history) if isinstance(iteration_history, list) else str(iteration_history)