First

api.py

@@ -20,6 +20,7 @@ from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException, NoSuchElementException, StaleElementReferenceException
 import random
+import pyautogui
 from streaming import StreamProcessor, create_streaming_response, StreamConfig, StreamingResponseGenerator
 import base64
 
@@ -170,12 +171,14 @@ class DriverManager:
         # Randomly choose between F5 and FN+F5
         if random.random() > 0.5:
             logger.info("Using F5 key")
-            body = driver.find_element(By.TAG_NAME, 'body')
-            body.send_keys(Keys.F5)
+            pyautogui.press('f5')
         else:
             logger.info("Using FN+F5 key combination")
-            body = driver.find_element(By.TAG_NAME, 'body')
-            body.send_keys(Keys.F5)  # Simulate FN+F5 as F5 since FN is hardware specific
+            pyautogui.keyDown('ctrl')
+            time.sleep(0.3)
+            pyautogui.press('f5')
+            time.sleep(0.3)
+            pyautogui.keyUp('ctrl')
         
         # Add random delay to simulate human behavior
         sleep_time = random.uniform(0.5, 2.0)
@@ -428,6 +431,35 @@ If you cannot find the checkbox, respond with "No checkbox found".
             logger.error(f"Error executing _select_model in executor: {e_exec}", exc_info=True)
             raise ModelSelectionError(f"Failed to select model {model_id} due to executor error: {e_exec}") from e_exec
 
+    async def _retry_with_reload(self, driver: Driver, model_id: str):
+        """Handle ModelSelectionError by reloading and retrying once"""
+        try:
+            # Manual reload simulation
+            pyautogui.press('f5')
+            time.sleep(0.5)
+            # Hard refresh (Ctrl+F5) - simulates FN+F5 behavior
+            pyautogui.keyDown('ctrl')
+            time.sleep(0.3)
+            pyautogui.press('f5')
+            time.sleep(0.3)
+            pyautogui.keyUp('ctrl')
+            
+            # Wait for page reload
+            WebDriverWait(driver, config.page_load_wait_after_refresh_seconds).until(
+                EC.presence_of_element_located((By.XPATH, "//input[@placeholder='Search models' and @cmdk-input]"))
+            )
+            
+            # Retry model selection
+            await self._select_model(model_id)
+            
+        except Exception as reload_err:
+            logger.error(f"Reload and retry failed: {reload_err}", exc_info=True)
+            # Capture screenshot after failed reload
+            png = driver.get_screenshot_as_png()
+            b64 = base64.b64encode(png).decode('utf-8')
+            logger.error(f"Screenshot base64 after failed reload: {b64}")
+            raise ModelSelectionError(f"Failed after reload attempt: {reload_err}") from reload_err
+
     def generate_reload_button_location(self, driver: Driver) -> str:
         # This function is not used by the refined captcha logic, but keeping it as it might be used elsewhere.
         logger.info("Generating reload button location with Gemini")
@@ -472,12 +504,13 @@ If you cannot find the reload button, respond with "No reload button found".
             if "No reload button found" in full_response_text:
                 logger.info("AI did not find reload button, performing manual F5 and FN+F5 reloads.")
                 try:
-                    body = driver.find_element(By.TAG_NAME, 'body')
-                    # F5 reload simulation
-                    body.send_keys(Keys.F5)
+                    pyautogui.press('f5')
                     time.sleep(0.5)
-                    # FN+F5 simulation as F5
-                    body.send_keys(Keys.F5)
+                    pyautogui.keyDown('ctrl')
+                    time.sleep(0.3)
+                    pyautogui.press('f5')
+                    time.sleep(0.3)
+                    pyautogui.keyUp('ctrl')
                     time.sleep(0.5)
                     # Capture screenshot after manual reload
                     png2 = driver.get_screenshot_as_png()