Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 import io
 import logging
 import traceback
+from pathlib import Path
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -246,12 +247,377 @@ def run_instagram_liker(username, password, max_likes):
                     browser.close()
                     return "\n".join(status_updates), image_path
                 
-                # Rest of the function would go here...
-                status_updates.append("Login field found! Stopping here for now.")
-                # Take final screenshot
-                final_screenshot = f"screenshots/final_{int(time.time())}.png"
-                page.screenshot(path=final_screenshot)
-                image_path = final_screenshot
+                # Try multiple selectors for password input
+                password_selectors = [
+                    "input[name='password']",
+                    "input[aria-label='Password']",
+                    "input[placeholder='Password']",
+                    "input[type='password']"
+                ]
+                
+                password_field = None
+                for selector in password_selectors:
+                    try:
+                        status_updates.append(f"Trying to find password field with selector: {selector}")
+                        field = page.query_selector(selector)
+                        if field:
+                            password_field = field
+                            status_updates.append(f"Found password field with selector: {selector}")
+                            break
+                    except Exception as e:
+                        logger.debug(f"Selector {selector} failed: {str(e)}")
+                
+                if not password_field:
+                    status_updates.append("Could not find password field. Instagram may have changed their interface.")
+                    # Take final screenshot before closing
+                    final_screenshot = f"screenshots/final_error_{int(time.time())}.png"
+                    page.screenshot(path=final_screenshot)
+                    image_path = final_screenshot
+                    browser.close()
+                    return "\n".join(status_updates), image_path
+                
+                # Enter credentials
+                status_updates.append(f"Entering username: {username}")
+                username_field.fill(username)
+                password_field.fill(password)
+                status_updates.append("Credentials entered")
+                
+                # Take a screenshot of filled form
+                try:
+                    creds_screenshot = f"screenshots/credentials_pw_{int(time.time())}.png"
+                    page.screenshot(path=creds_screenshot)
+                    image_path = creds_screenshot
+                    status_updates.append("Credentials screenshot saved")
+                except Exception as e:
+                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                
+                # Find login button
+                login_button_selectors = [
+                    "button[type='submit']",
+                    "button:has-text('Log in')",
+                    "button:has-text('Sign in')",
+                    "button:has-text('Log In')",
+                    "form button"
+                ]
+                
+                login_button = None
+                for selector in login_button_selectors:
+                    try:
+                        status_updates.append(f"Trying to find login button with selector: {selector}")
+                        button = page.query_selector(selector)
+                        if button:
+                            login_button = button
+                            status_updates.append(f"Found login button with selector: {selector}")
+                            break
+                    except Exception as e:
+                        logger.debug(f"Selector {selector} failed: {str(e)}")
+                
+                if not login_button:
+                    status_updates.append("Could not find login button. Instagram may have changed their interface.")
+                    # Take final screenshot before closing
+                    final_screenshot = f"screenshots/final_error_{int(time.time())}.png"
+                    page.screenshot(path=final_screenshot)
+                    image_path = final_screenshot
+                    browser.close()
+                    return "\n".join(status_updates), image_path
+                
+                # Click login button
+                status_updates.append("Clicking login button...")
+                login_button.click()
+                
+                # Wait for navigation to complete
+                status_updates.append("Waiting for login process...")
+                page.wait_for_timeout(5000)
+                
+                # Take post-login screenshot
+                try:
+                    post_login_screenshot = f"screenshots/post_login_pw_{int(time.time())}.png"
+                    page.screenshot(path=post_login_screenshot)
+                    image_path = post_login_screenshot
+                    status_updates.append("Post-login screenshot saved")
+                except Exception as e:
+                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                
+                # Check if login was successful
+                current_url = page.url
+                if "/accounts/login" in current_url or "/login" in current_url:
+                    # Still on login page - check for error messages
+                    error_selectors = [
+                        "#slfErrorAlert",
+                        "p[data-testid='login-error-message']",
+                        "div[role='alert']",
+                        "p.sIKKJ",
+                        ".coreSpriteAccessUpsell + div"
+                    ]
+                    
+                    error_message = ""
+                    for selector in error_selectors:
+                        try:
+                            el = page.query_selector(selector)
+                            if el:
+                                error_message = el.text_content()
+                                break
+                        except:
+                            pass
+                    
+                    if error_message:
+                        status_updates.append(f"Login failed: {error_message}")
+                    else:
+                        status_updates.append("Login failed: Reason unknown. Check your credentials.")
+                    
+                    browser.close()
+                    return "\n".join(status_updates), image_path
+                
+                status_updates.append("Login successful! Now handling post-login dialogs...")
+                
+                # Handle "Save Login Info" popup if it appears
+                try:
+                    page.wait_for_timeout(2000)  # Wait a bit for dialog to appear
+                    
+                    save_info_selectors = [
+                        "text=Save Login Info",
+                        "text=Save Your Login Info",
+                        "text=Save Info"
+                    ]
+                    not_now_selectors = [
+                        "text=Not Now",
+                        "button:has-text('Not Now')",
+                        "button.sqdOP.yWX7d",
+                        "button:not(:has-text('Save'))"
+                    ]
+                    
+                    # Check if any save info dialog appears
+                    dialog_found = False
+                    for selector in save_info_selectors:
+                        if page.query_selector(selector):
+                            dialog_found = True
+                            status_updates.append(f"Save Login Info dialog found with: {selector}")
+                            break
+                    
+                    if dialog_found:
+                        # Try to click "Not Now"
+                        dismissed = False
+                        for not_now in not_now_selectors:
+                            try:
+                                button = page.query_selector(not_now)
+                                if button:
+                                    button.click()
+                                    status_updates.append(f"Dismissed 'Save Login Info' popup using: {not_now}")
+                                    page.wait_for_timeout(2000)
+                                    dismissed = True
+                                    break
+                            except Exception as e:
+                                logger.debug(f"Not Now button {not_now} failed: {str(e)}")
+                                continue
+                        
+                        if not dismissed:
+                            status_updates.append("Found Save Login dialog but couldn't dismiss it")
+                    else:
+                        status_updates.append("No 'Save Login Info' popup detected")
+                except Exception as e:
+                    status_updates.append(f"Error handling Save Login dialog: {str(e)}")
+                
+                # Take a screenshot after handling first dialog
+                try:
+                    after_save_screenshot = f"screenshots/after_save_dialog_{int(time.time())}.png"
+                    page.screenshot(path=after_save_screenshot)
+                    image_path = after_save_screenshot
+                    status_updates.append("Screenshot after Save Info dialog")
+                except Exception as e:
+                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                
+                # Handle notifications popup if it appears
+                try:
+                    page.wait_for_timeout(2000)  # Wait a bit for dialog to appear
+                    
+                    notifications_selectors = [
+                        "text=Turn on Notifications",
+                        "text=Enable Notifications",
+                        "h2:has-text('Notifications')"
+                    ]
+                    not_now_selectors = [
+                        "text=Not Now",
+                        "button:has-text('Not Now')",
+                        "button.sqdOP.yWX7d",
+                        "button:not(:has-text('Allow'))"
+                    ]
+                    
+                    # Check if any notifications dialog appears
+                    dialog_found = False
+                    for selector in notifications_selectors:
+                        if page.query_selector(selector):
+                            dialog_found = True
+                            status_updates.append(f"Notifications dialog found with: {selector}")
+                            break
+                    
+                    if dialog_found:
+                        # Try to click "Not Now"
+                        dismissed = False
+                        for not_now in not_now_selectors:
+                            try:
+                                button = page.query_selector(not_now)
+                                if button:
+                                    button.click()
+                                    status_updates.append(f"Dismissed notifications popup using: {not_now}")
+                                    page.wait_for_timeout(2000)
+                                    dismissed = True
+                                    break
+                            except Exception as e:
+                                logger.debug(f"Not Now button {not_now} failed: {str(e)}")
+                                continue
+                        
+                        if not dismissed:
+                            status_updates.append("Found Notifications dialog but couldn't dismiss it")
+                    else:
+                        status_updates.append("No notifications popup detected")
+                except Exception as e:
+                    status_updates.append(f"Error handling Notifications dialog: {str(e)}")
+                
+                # Take feed screenshot
+                try:
+                    feed_screenshot = f"screenshots/feed_pw_{int(time.time())}.png"
+                    page.screenshot(path=feed_screenshot)
+                    image_path = feed_screenshot
+                    status_updates.append("Feed screenshot saved")
+                except Exception as e:
+                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                
+                status_updates.append("Successfully navigated to Instagram feed!")
+                
+                # Start liking posts
+                status_updates.append(f"Starting to like posts (target: {max_likes})...")
+                
+                # Like posts
+                likes_count = 0
+                scroll_count = 0
+                max_scrolls = 30
+                
+                # Try to find posts with multiple selectors
+                article_selectors = [
+                    "article", 
+                    "div[role='presentation']", 
+                    "div[data-testid='post-content']",
+                    "div._aac4._aac5._aac6", 
+                    "div._ab6k"
+                ]
+                
+                article_found = False
+                for selector in article_selectors:
+                    try:
+                        if page.query_selector(selector):
+                            article_found = True
+                            status_updates.append(f"Found posts using selector: {selector}")
+                            break
+                    except Exception as e:
+                        logger.debug(f"Article selector {selector} failed: {str(e)}")
+                
+                if not article_found:
+                    status_updates.append("Could not find posts on feed. Instagram may have changed their interface.")
+                    browser.close()
+                    return "\n".join(status_updates), image_path
+                
+                while likes_count < max_likes and scroll_count < max_scrolls:
+                    # Try different selectors for like buttons
+                    like_button_selectors = [
+                        "article section svg[aria-label='Like']",
+                        "article svg[aria-label='Like']",
+                        "svg[aria-label='Like']",
+                        "span[class*='_aamw'] svg[aria-label='Like']",
+                        "article button[type='button'] svg:not([aria-label='Unlike'])"
+                    ]
+                    
+                    like_buttons = []
+                    used_selector = ""
+                    for selector in like_button_selectors:
+                        try:
+                            buttons = page.query_selector_all(selector)
+                            if buttons and len(buttons) > 0:
+                                like_buttons = buttons
+                                used_selector = selector
+                                status_updates.append(f"Found {len(buttons)} like buttons with selector: {selector}")
+                                break
+                        except Exception as e:
+                            logger.debug(f"Like button selector {selector} failed: {str(e)}")
+                    
+                    status_updates.append(f"Found {len(like_buttons)} like buttons on scroll {scroll_count}")
+                    
+                    if len(like_buttons) == 0 and scroll_count > 5:
+                        status_updates.append("No more like buttons found. Stopping.")
+                        break
+                    
+                    # Click like buttons
+                    for i, button in enumerate(like_buttons):
+                        if likes_count >= max_likes:
+                            break
+                        
+                        try:
+                            # Scroll to button
+                            button.scroll_into_view_if_needed()
+                            page.wait_for_timeout(1000)
+                            
+                            # Take pre-click screenshot occasionally
+                            if likes_count % 5 == 0:
+                                try:
+                                    pre_like_screenshot = f"screenshots/pre_like_pw_{likes_count}_{int(time.time())}.png"
+                                    page.screenshot(path=pre_like_screenshot)
+                                    image_path = pre_like_screenshot
+                                    status_updates.append(f"Pre-like screenshot {likes_count} saved")
+                                except Exception as e:
+                                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                            
+                            # Get parent button if needed
+                            if "svg" in used_selector:
+                                try:
+                                    # Find the parent button of the SVG
+                                    parent_button = button.evaluate("node => node.closest('button')")
+                                    if parent_button:
+                                        parent_button.click()
+                                    else:
+                                        button.click()
+                                except Exception as e:
+                                    # If error finding parent, click the button directly
+                                    button.click()
+                            else:
+                                # Click directly if it's already a button
+                                button.click()
+                                
+                            likes_count += 1
+                            status_updates.append(f"Liked post {likes_count}/{max_likes}")
+                            
+                            # Take screenshot occasionally
+                            if likes_count % 5 == 0:
+                                try:
+                                    like_screenshot = f"screenshots/after_like_pw_{likes_count}_{int(time.time())}.png"
+                                    page.screenshot(path=like_screenshot)
+                                    image_path = like_screenshot
+                                    status_updates.append(f"After-like screenshot {likes_count} saved")
+                                except Exception as e:
+                                    status_updates.append(f"Error taking screenshot: {str(e)}")
+                            
+                            # Wait between likes to avoid rate limiting
+                            page.wait_for_timeout(2000 + (likes_count % 3) * 1000)  # Varying delay
+                        except Exception as e:
+                            status_updates.append(f"Error liking post {i+1}: {str(e)}")
+                            continue
+                    
+                    # Scroll down to load more
+                    page.evaluate("window.scrollBy(0, 1000)")
+                    status_updates.append("Scrolled down to load more posts")
+                    page.wait_for_timeout(3000)
+                    scroll_count += 1
+                
+                # Final status
+                final_message = f"Finished! Liked {likes_count} posts."
+                status_updates.append(final_message)
+                
+                # Final screenshot
+                try:
+                    final_screenshot = f"screenshots/final_{int(time.time())}.png"
+                    page.screenshot(path=final_screenshot)
+                    image_path = final_screenshot
+                    status_updates.append("Final screenshot saved")
+                except Exception as e:
+                    status_updates.append(f"Error taking final screenshot: {str(e)}")
                 
                 # Close the browser
                 browser.close()
@@ -262,6 +628,13 @@ def run_instagram_liker(username, password, max_likes):
                 traceback_msg = traceback.format_exc()
                 logger.error(traceback_msg)
                 status_updates.append("See logs for detailed error information")
+                
+                # Try to close browser in case of error
+                try:
+                    browser.close()
+                    status_updates.append("Browser closed after error")
+                except:
+                    pass
         
     except Exception as e:
         error_message = f"Error with Playwright: {str(e)}"