modified: main.py

main.py

@@ -554,15 +554,29 @@ Rules:
             if not self.local:
                 image = Image.open(marked_image_path)
                 original_image = Image.open(self.path)
-                response = self.client.models.generate_content(
-                    model=self.model_name,
-                    contents=[image, original_image, prompt],
-                    config=types.GenerateContentConfig(
-                        response_mime_type="application/json",
-                        response_schema=get_solution,
-                        thinking_config=types.ThinkingConfig(thinking_budget=self.thinking_budget if self.think else 0),
-                    ),
-                )
+                try:
+                    response = self.client.models.generate_content(
+                        model=self.model_name,
+                        contents=[image, original_image, prompt],
+                        config=types.GenerateContentConfig(
+                            response_mime_type="application/json",
+                            response_schema=get_solution,
+                            thinking_config=types.ThinkingConfig(thinking_budget=self.thinking_budget if self.think else 0),
+                        ),
+                    )
+                except genai.errors.ServerError:
+                    if self.model_name == "gemini-3-flash-preview":
+                        print("The thinking model is currently not available - falling back to gemini-2.5-flash")
+                        self.model_name = "gemini-2.5-flash"
+                        response = self.client.models.generate_content(
+                            model=self.model_name,
+                            contents=[image, original_image, prompt],
+                            config=types.GenerateContentConfig(
+                                response_mime_type="application/json",
+                                response_schema=get_solution,
+                                thinking_config=types.ThinkingConfig(thinking_budget=self.thinking_budget if self.think else 0),
+                            ),
+                        )
                 output = response.parsed
             else:
                 if self.model_name == "qwen3-vl:8b-thinking" and self.think: