added error handling

.gitignore

@@ -0,0 +1,53 @@
+# Ignore compiled Python files
+*.pyc
+*.pyo
+__pycache__/
+
+# Ignore local development settings
+.env
+
+# Ignore the virtual environment directory
+venv/
+
+# Ignore the pipenv virtual environment
+Pipfile
+Pipfile.lock
+
+# Ignore generated files
+*.pyc
+*.pyo
+*.pyd
+*.so
+*.egg-info/
+dist/
+build/
+*.egg
+
+# Ignore IDE and editor files
+.idea/
+.vscode/
+*.sublime-project
+*.sublime-workspace
+
+# Ignore system-specific files
+.DS_Store
+Thumbs.db
+
+# Ignore logs and other generated data
+logs/
+*.log
+
+# Ignore any sensitive or private information
+config.ini
+secrets.json
+
+# Ignore cache and temporary files
+*.swp
+*.swo
+*.bak
+*.cache
+*.tmp
+
+# Ignore dependency directories
+lib/
+libs/

app.py

@@ -1,7 +1,5 @@
 import PIL.Image
 import gradio as gr
-import numpy as np
-import cv2
 from PIL import ImageOps
 import requests
 from pathlib import Path
@@ -11,17 +9,16 @@ import numpy as np
 import io
 import os
 from random import choice
+from dotenv import load_dotenv
 
+load_dotenv()
 
-def get_mask(img_in):
 
-    print(f"{img_in=}")
+def get_mask(img_in):
 
-    # covert to 8-bit RGB
+    # print(f"{img_in=}")
     img_in = img_in.convert("RGB")
 
-    # img_in_smol = img_in.resize((256, 256))
-
     file_path = "/tmp/img_in.jpg"
     img_in.save(file_path)
 
@@ -32,6 +29,12 @@ def get_mask(img_in):
     response = requests.post(upload_url, files=files)
 
     if response.status_code == 200:
+
+        if 'error' in response.json():
+            print(f"ERROR: {response.json()['error']}")
+            gr.Error(response.json()['error'])
+            return (None, None)
+
         result = response.json()
         print('Result:', result)
 
@@ -56,7 +59,8 @@ def get_mask(img_in):
 
         return (img_in, result['emotion'])
     else:
-        print('error:', response.text)
+        gr.Error(response.text)
+        # print('error:', response.text)
 
 
 footer = r"""

requirements.txt

@@ -1,6 +1,5 @@
-onnx
-onnxruntime-gpu
 opencv-python
 numpy
 pillow~=9.5.0
-streamlit
+gradio
+python-dotenv