Update app.py

app.py

@@ -36,16 +36,13 @@ class AIShoppingAnalyzer:
         
         return MultimodalWebSurfer(
             name="websurfer_agent",
-            description=description,
             model_client=self.model_client,
+            description=description,
             headless=True,
-            playwright_launch_kwargs={
-                "args": [
-                    "--disable-dev-shm-usage",
-                    "--no-sandbox",
-                    "--disable-setuid-sandbox"
-                ]
-            }
+            to_save_screenshots=True,  # Save screenshots for analysis
+            use_ocr=True,  # Enable OCR for better text extraction
+            to_resize_viewport=True,  # Ensure proper viewport sizing
+            debug_dir="debug_logs"  # Save debug information
         )
 
     def create_assistant(self) -> AssistantAgent:
@@ -122,6 +119,7 @@ class AIShoppingAnalyzer:
         """Run the analysis with proper cleanup"""
         websurfer = None
         try:
+            # Set up the analysis query
             query = (
                 f"Analyze the e-commerce experience for {website_url} focusing on:\n"
                 f"1. Product findability in the {product_category} category\n"
@@ -133,6 +131,7 @@ class AIShoppingAnalyzer:
             if specific_product:
                 query += f"\n5. Detailed analysis of this specific product: {specific_product}"
 
+            # Initialize agents with proper configuration
             websurfer = self.create_websurfer()
             assistant = self.create_assistant()
             team = self.create_team(websurfer, assistant)
@@ -153,9 +152,12 @@ class AIShoppingAnalyzer:
         finally:
             if websurfer:
                 try:
+                    # Properly close the browser
                     await websurfer.close()
+                    print("Browser closed successfully")
                 except Exception as e:
-                    print(f"Cleanup error: {str(e)}")
+                    print(f"Error closing browser: {str(e)}")
+
 
 def create_gradio_interface() -> gr.Blocks:
     """Create the Gradio interface for the AI Shopping Analyzer"""
@@ -342,7 +344,7 @@ def create_gradio_interface() -> gr.Blocks:
                 analyze_button = gr.Button(
                     "Analyze Site",
                     size="lg",
-                    variant="primary"
+                    variant="secondary"
                 )
             
             # Right column for output