Update app.py

app.py

@@ -4,6 +4,12 @@ import base64
 import requests
 from PIL import Image
 from io import BytesIO
+import logging
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+
+# Configure logging
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 app = Flask(__name__)
 
@@ -18,13 +24,21 @@ HF_API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-5
 
 def query_hugging_face(image_data):
     if not HF_API_TOKEN:
+        logging.error("Hugging Face API token not set.")
         return {"error": "Hugging Face API token not set. Please set the HF_API_TOKEN environment variable."}
     try:
+        session = requests.Session()
+        # Configure retries: 3 attempts, backoff factor of 1s, retry on 502/503/504
+        retries = Retry(total=3, backoff_factor=1, status_forcelist=[502, 503, 504])
+        session.mount('https://', HTTPAdapter(max_retries=retries))
         headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
-        response = requests.post(HF_API_URL, headers=headers, data=image_data, timeout=10)
-        response.raise_for_status()  # Raise an error for bad status codes
+        logging.debug(f"Sending request to {HF_API_URL}")
+        response = session.post(HF_API_URL, headers=headers, data=image_data, timeout=10)
+        response.raise_for_status()
+        logging.debug("Hugging Face API request successful.")
         return response.json()
     except requests.exceptions.RequestException as e:
+        logging.error(f"Failed to connect to Hugging Face API: {str(e)}")
         return {"error": f"Failed to connect to Hugging Face API: {str(e)}"}
 
 @app.route('/')
@@ -57,6 +71,17 @@ def capture():
             'hf_result': hf_result
         })
     except Exception as e:
+        logging.error(f"Error in capture: {str(e)}")
+        return jsonify({'status': 'error', 'message': str(e)})
+
+@app.route('/test_connectivity')
+def test_connectivity():
+    try:
+        response = requests.get('https://www.google.com', timeout=5)
+        logging.debug(f"Connectivity test to Google: {response.status_code}")
+        return jsonify({'status': 'success', 'code': response.status_code})
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Connectivity test failed: {str(e)}")
         return jsonify({'status': 'error', 'message': str(e)})
 
 if __name__ == '__main__':