cursorflow 1.2.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

cursorflow/cli.py

@@ -124,6 +124,218 @@ def test(test_name, base_url, actions, logs, config, verbose):
             console.print(traceback.format_exc())
         raise
 
+@main.command()
+@click.argument('mockup_url', required=True)
+@click.option('--base-url', '-u', default='http://localhost:3000',
+              help='Base URL of work-in-progress implementation')
+@click.option('--mockup-actions', '-ma',
+              help='JSON file with actions to perform on mockup, or inline JSON string')
+@click.option('--implementation-actions', '-ia',
+              help='JSON file with actions to perform on implementation, or inline JSON string')
+@click.option('--viewports', '-v',
+              help='JSON array of viewports to test: [{"width": 1440, "height": 900, "name": "desktop"}]')
+@click.option('--diff-threshold', '-t', type=float, default=0.1,
+              help='Visual difference threshold (0.0-1.0)')
+@click.option('--output', '-o', default='mockup_comparison_results.json',
+              help='Output file for comparison results')
+@click.option('--verbose', is_flag=True,
+              help='Verbose output')
+def compare_mockup(mockup_url, base_url, mockup_actions, implementation_actions, viewports, diff_threshold, output, verbose):
+    """Compare mockup design to work-in-progress implementation"""
+    
+    console.print(f"🎨 Comparing mockup [blue]{mockup_url}[/blue] to implementation [blue]{base_url}[/blue]")
+    
+    # Parse actions
+    def parse_actions(actions_input):
+        if not actions_input:
+            return None
+        
+        if actions_input.startswith('[') or actions_input.startswith('{'):
+            return json.loads(actions_input)
+        else:
+            with open(actions_input, 'r') as f:
+                return json.load(f)
+    
+    try:
+        mockup_actions_parsed = parse_actions(mockup_actions)
+        implementation_actions_parsed = parse_actions(implementation_actions)
+        
+        # Parse viewports
+        viewports_parsed = None
+        if viewports:
+            if viewports.startswith('['):
+                viewports_parsed = json.loads(viewports)
+            else:
+                with open(viewports, 'r') as f:
+                    viewports_parsed = json.load(f)
+        
+        # Build comparison config
+        comparison_config = {
+            "diff_threshold": diff_threshold
+        }
+        if viewports_parsed:
+            comparison_config["viewports"] = viewports_parsed
+        
+    except Exception as e:
+        console.print(f"[red]Error parsing input parameters: {e}[/red]")
+        return
+    
+    # Initialize CursorFlow
+    try:
+        from .core.cursorflow import CursorFlow
+        flow = CursorFlow(
+            base_url=base_url,
+            log_config={'source': 'local', 'paths': ['logs/app.log']},
+            browser_config={'headless': True}
+        )
+    except Exception as e:
+        console.print(f"[red]Error initializing CursorFlow: {e}[/red]")
+        return
+    
+    # Execute mockup comparison
+    try:
+        console.print("🚀 Starting mockup comparison...")
+        results = asyncio.run(flow.compare_mockup_to_implementation(
+            mockup_url=mockup_url,
+            mockup_actions=mockup_actions_parsed,
+            implementation_actions=implementation_actions_parsed,
+            comparison_config=comparison_config
+        ))
+        
+        if "error" in results:
+            console.print(f"[red]❌ Comparison failed: {results['error']}[/red]")
+            return
+        
+        # Display results summary
+        summary = results.get('summary', {})
+        console.print(f"✅ Comparison completed: {results.get('comparison_id', 'unknown')}")
+        console.print(f"📊 Average similarity: [bold]{summary.get('average_similarity', 0)}%[/bold]")
+        console.print(f"📱 Viewports tested: {summary.get('viewports_tested', 0)}")
+        
+        # Show recommendations
+        recommendations = results.get('recommendations', [])
+        if recommendations:
+            console.print(f"💡 Recommendations: {len(recommendations)} improvements suggested")
+            for i, rec in enumerate(recommendations[:3]):  # Show first 3
+                console.print(f"  {i+1}. {rec.get('description', 'No description')}")
+        
+        # Save results
+        with open(output, 'w') as f:
+            json.dump(results, f, indent=2, default=str)
+        
+        console.print(f"💾 Full results saved to: [cyan]{output}[/cyan]")
+        console.print(f"📁 Visual diffs stored in: [cyan].cursorflow/artifacts/[/cyan]")
+        
+    except Exception as e:
+        console.print(f"[red]❌ Comparison failed: {e}[/red]")
+        if verbose:
+            import traceback
+            console.print(traceback.format_exc())
+        raise
+
+@main.command()
+@click.argument('mockup_url', required=True)
+@click.option('--base-url', '-u', default='http://localhost:3000',
+              help='Base URL of work-in-progress implementation')
+@click.option('--css-improvements', '-c', required=True,
+              help='JSON file with CSS improvements to test, or inline JSON string')
+@click.option('--base-actions', '-a',
+              help='JSON file with base actions to perform before each test')
+@click.option('--diff-threshold', '-t', type=float, default=0.1,
+              help='Visual difference threshold (0.0-1.0)')
+@click.option('--output', '-o', default='mockup_iteration_results.json',
+              help='Output file for iteration results')
+@click.option('--verbose', is_flag=True,
+              help='Verbose output')
+def iterate_mockup(mockup_url, base_url, css_improvements, base_actions, diff_threshold, output, verbose):
+    """Iteratively improve implementation to match mockup design"""
+    
+    console.print(f"🔄 Iterating on [blue]{base_url}[/blue] to match [blue]{mockup_url}[/blue]")
+    
+    # Parse CSS improvements
+    def parse_json_input(input_str):
+        if not input_str:
+            return None
+        
+        if input_str.startswith('[') or input_str.startswith('{'):
+            return json.loads(input_str)
+        else:
+            with open(input_str, 'r') as f:
+                return json.load(f)
+    
+    try:
+        css_improvements_parsed = parse_json_input(css_improvements)
+        base_actions_parsed = parse_json_input(base_actions)
+        
+        if not css_improvements_parsed:
+            console.print("[red]Error: CSS improvements are required[/red]")
+            return
+        
+        comparison_config = {"diff_threshold": diff_threshold}
+        
+    except Exception as e:
+        console.print(f"[red]Error parsing input parameters: {e}[/red]")
+        return
+    
+    # Initialize CursorFlow
+    try:
+        from .core.cursorflow import CursorFlow
+        flow = CursorFlow(
+            base_url=base_url,
+            log_config={'source': 'local', 'paths': ['logs/app.log']},
+            browser_config={'headless': True}
+        )
+    except Exception as e:
+        console.print(f"[red]Error initializing CursorFlow: {e}[/red]")
+        return
+    
+    # Execute iterative mockup matching
+    try:
+        console.print(f"🚀 Starting iterative matching with {len(css_improvements_parsed)} CSS improvements...")
+        results = asyncio.run(flow.iterative_mockup_matching(
+            mockup_url=mockup_url,
+            css_improvements=css_improvements_parsed,
+            base_actions=base_actions_parsed,
+            comparison_config=comparison_config
+        ))
+        
+        if "error" in results:
+            console.print(f"[red]❌ Iteration failed: {results['error']}[/red]")
+            return
+        
+        # Display results summary
+        summary = results.get('summary', {})
+        console.print(f"✅ Iteration completed: {results.get('session_id', 'unknown')}")
+        console.print(f"📊 Total improvement: [bold]{summary.get('total_improvement', 0)}%[/bold]")
+        console.print(f"🔄 Successful iterations: {summary.get('successful_iterations', 0)}/{summary.get('total_iterations', 0)}")
+        
+        # Show best iteration
+        best_iteration = results.get('best_iteration')
+        if best_iteration:
+            console.print(f"🏆 Best iteration: {best_iteration.get('css_change', {}).get('name', 'unnamed')}")
+            console.print(f"   Similarity achieved: {best_iteration.get('similarity_achieved', 0)}%")
+        
+        # Show final recommendations
+        recommendations = results.get('final_recommendations', [])
+        if recommendations:
+            console.print(f"💡 Final recommendations: {len(recommendations)} actions suggested")
+            for i, rec in enumerate(recommendations[:3]):
+                console.print(f"  {i+1}. {rec.get('description', 'No description')}")
+        
+        # Save results
+        with open(output, 'w') as f:
+            json.dump(results, f, indent=2, default=str)
+        
+        console.print(f"💾 Full results saved to: [cyan]{output}[/cyan]")
+        console.print(f"📁 Iteration progress stored in: [cyan].cursorflow/artifacts/[/cyan]")
+        
+    except Exception as e:
+        console.print(f"[red]❌ Iteration failed: {e}[/red]")
+        if verbose:
+            import traceback
+            console.print(traceback.format_exc())
+        raise
+
 @main.command()
 @click.option('--project-path', '-p', default='.',
               help='Project directory path')