Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 import logging
 import time
 from gradio_client import Client
+from json.decoder import JSONDecodeError  # Import JSONDecodeError
 
 app = Flask(__name__)
 
@@ -12,7 +13,37 @@ logging.basicConfig(level=logging.INFO,
 logger = logging.getLogger(__name__)
 
 # Initialize JARVIS client
-jarvis = Client("hadadrjt/ai")
+jarvis = None # Initialize as None
+
+def initialize_jarvis_client(max_retries=3, retry_delay=5):
+    global jarvis # Use the global jarvis variable
+    for attempt in range(max_retries):
+        try:
+            jarvis = Client("hadadrjt/ai")
+            logger.info("JARVIS client initialized successfully.")
+            return True # Indicate success
+        except JSONDecodeError as e: # Catch JSONDecodeError specifically
+            logger.warning(f"Attempt {attempt + 1}/{max_retries}: JSONDecodeError during JARVIS client initialization: {e}")
+            if attempt < max_retries - 1:
+                time.sleep(retry_delay)  # Wait before retrying
+            else:
+                logger.error("Max retries reached. JARVIS client initialization failed due to JSONDecodeError.")
+                return False # Indicate failure
+        except Exception as e: # Catch other exceptions as well
+            logger.error(f"Attempt {attempt + 1}/{max_retries}: Error during JARVIS client initialization: {e}")
+            if attempt < max_retries - 1:
+                time.sleep(retry_delay)
+            else:
+                logger.error("Max retries reached. JARVIS client initialization failed due to general exception.")
+                return False # Indicate failure
+    return False # Return False if loop completes without success
+
+
+if not initialize_jarvis_client(): # Initialize client with retry mechanism
+    print("Failed to initialize JARVIS client after multiple retries. API will likely not function correctly.")
+    # It's up to you if you want to exit here or let the Flask app start but potentially fail on requests.
+    # For now, let's continue and handle potential None jarvis later.
+
 
 # Define available models
 models = [
@@ -53,6 +84,10 @@ def chat_completions():
     if not authenticate_request(request):
         return jsonify({"error": {"message": "Invalid API key", "code": "invalid_api_key"}}), 401
 
+    if jarvis is None: # Check if jarvis client is initialized
+        return jsonify({"error": {"message": "JARVIS client failed to initialize. API not available.", "code": "jarvis_not_initialized"}}), 500
+
+
     data = request.json
     messages = data.get("messages", [])
     model = data.get("model", "JARVIS: 2.1.2")
@@ -104,6 +139,9 @@ def chat_completions():
 def list_models():
     if not authenticate_request(request):
         return jsonify({"error": {"message": "Invalid API key", "code": "invalid_api_key"}}), 401
+    if jarvis is None: # Check if jarvis client is initialized
+        return jsonify({"error": {"message": "JARVIS client failed to initialize. API not available.", "code": "jarvis_not_initialized"}}), 500
+
 
     return jsonify({"data": [{"id": model} for model in models], "object": "list"})