Update app.py

app.py

@@ -11,7 +11,12 @@ from PIL import Image, ImageFile, UnidentifiedImageError
 import gradio as gr
 import time
 import atexit
-from requests.exceptions import RequestException # HTTPError is a subclass of RequestException, so can catch RequestException
+from requests.exceptions import RequestException
+
+# Direct import of Mistral client and its exception.
+# An ImportError will now occur at script start if the library is not installed.
+from mistralai import Mistral
+from mistralai.exceptions import MistralAPIException
 
 # --- Configuration and Globals ---
 DEFAULT_KEY = os.getenv("MISTRAL_API_KEY", "")
@@ -36,24 +41,6 @@ Image.MAX_IMAGE_PIXELS = 10000 * 10000
 
 DEFAULT_HEADERS = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"}
 
-# Attempt to import Mistral client and its exception.
-# Define a mock exception if the library is not installed, to ensure type hinting and
-# `except MistralAPIException` blocks work correctly.
-try:
-    from mistralai import Mistral
-    from mistralai.exceptions import MistralAPIException
-except ImportError:
-    Mistral = None
-    class MistralAPIException(Exception):
-        def __init__(self, message, status_code=None, *args, **kwargs):
-            super().__init__(message, *args)
-            self.status_code = status_code
-            self._message = message
-
-        @property
-        def message(self):
-            return self._message
-
 # --- Temporary File Cleanup ---
 _temp_preview_files_to_delete = []
 
@@ -72,46 +59,19 @@ atexit.register(_cleanup_all_temp_preview_files)
 # --- Mistral Client and API Helpers ---
 def get_client(key: Optional[str] = None):
     """
-    Returns a Mistral client instance. If the mistralai library is not installed
-    or the API key is missing, a mock client is returned which raises
-    MistralAPIException with specific messages.
+    Returns a Mistral client instance. If the API key is missing,
+    a MistralAPIException is raised.
     """
     api_key = (key or "").strip() or DEFAULT_KEY
 
-    class DummyMistralService:
-        def __init__(self, error_message: str, status_code: Optional[int] = None):
-            self._error_message = error_message
-            self._status_code = status_code
-
-        def complete(self, **kwargs):
-            raise MistralAPIException(self._error_message, status_code=self._status_code)
-
-        def upload(self, **kwargs):
-            raise MistralAPIException(self._error_message, status_code=self._status_code)
-
-    class DummyClient:
-        def __init__(self, k, library_missing_msg: Optional[str] = None, api_key_missing_msg: Optional[str] = None):
-            self.api_key = k
-            self.is_real_client = False # Flag for internal diagnostics, not for logic flow control
-
-            if not k:
-                error_msg = api_key_missing_msg or "Mistral API key is not set."
-                status_code = 401
-            else:
-                error_msg = library_missing_msg or "Mistral client library is not installed."
-                status_code = 500
-
-            self.chat = DummyMistralService(error_msg, status_code)
-            self.files = DummyMistralService(error_msg, status_code)
-
     if not api_key:
-        return DummyClient(api_key, api_key_missing_msg="Mistral API key is not set. Please provide it in the UI or as MISTRAL_API_KEY environment variable.")
-
-    if Mistral is None:
-        return DummyClient(api_key, library_missing_msg="Mistral client library is not installed. Please install it with 'pip install mistralai'.")
+        raise MistralAPIException(
+            "Mistral API key is not set. Please provide it in the UI or as MISTRAL_API_KEY environment variable.",
+            status_code=401 # Unauthorized
+        )
 
+    # Directly instantiate the real Mistral client
     client = Mistral(api_key=api_key)
-    setattr(client, 'is_real_client', True)
     return client
 
 def is_remote(src: str) -> bool:
@@ -539,9 +499,7 @@ def _get_playable_preview_path_from_raw(src_url: str, raw_bytes: bytes, is_image
         jpeg_bytes = convert_to_jpeg_bytes(raw_bytes, base_h=1024)
         if jpeg_bytes:
             return _temp_file(jpeg_bytes, suffix=".jpg")
-        else:
-            print(f"Warning: Could not convert image bytes for {src_url} to JPEG.")
-            return "" # Failed image conversion, do not fall through to video logic
+        return "" # Failed image conversion, do not fall through to video logic
     elif is_video_hint:
         temp_raw_video_path = _temp_file(raw_bytes, suffix=ext_from_src(src_url) or ".mp4")
         if not temp_raw_video_path:
@@ -772,7 +730,7 @@ def create_demo():
                 except UnidentifiedImageError:
                     is_actually_video_for_analysis = True
                 except Exception as e:
-                    print(f"Warning: PIL error during image verification for raw analysis media ({raw_media_path}): {e}. Falling back to video.")
+                    print(f"Warning: PIL error during image verification for raw analysis media ({raw_media_path}): {e}. Assuming video type for fallback.")
                     is_actually_video_for_analysis = True
 
                 client = get_client(key)