First basic sms filter for iOS app.

.gitignore

@@ -0,0 +1,71 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+build/
+dist/
+*.egg-info/
+.eggs/
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Jupyter Notebook checkpoints
+.ipynb_checkpoints
+
+# VSCode settings
+.vscode/
+
+# PyCharm
+.idea/
+
+# macOS and system files
+.DS_Store
+Thumbs.db
+
+# Logs and temporary files
+*.log
+*.tmp
+*.swp
+
+# Hugging Face cache
+~/.cache/huggingface/
+huggingface_cache/
+cached_models/
+
+# Datasets or model outputs
+data/
+output/
+results/
+*.ckpt
+*.pth
+
+# Hugging Face Spaces
+# (if using Gradio, Streamlit, etc.)
+app/__pycache__/
+app/*.pyc
+*.gradio/
+
+# Optional: ignore weights or tokenizer files (use with care)
+pytorch_model.bin
+tf_model.h5
+tokenizer_config.json
+special_tokens_map.json
+merges.txt
+vocab.json
+config.json

Dockerfile

@@ -1,13 +1,17 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.12
+FROM python:3.12-slim
 
 WORKDIR /app
 
 COPY requirements.txt .
 
+RUN pip install --upgrade pip
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY . /app
+
+ENV HF_HOME=/tmp/huggingface_home
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,8 +1,65 @@
 from fastapi import FastAPI
+from fastapi.responses import JSONResponse
+from pydantic import  BaseModel
+from enum import Enum
+from transformers import pipeline
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+class MessageModel(BaseModel):
+    text: str
 
+class QueryModel(BaseModel):
+    sender: str
+    message: MessageModel
+
+class AppModel(BaseModel):
+    version: str
+
+class InputModel(BaseModel):
+    _version: int
+    query: QueryModel
+    app: AppModel
+
+class ActionModel(Enum):
+    # Insufficient information to determine an action to take. In a query response, has the effect of allowing the message to be shown normally.
+    NONE = 0
+    # Allow the message to be shown normally.
+    ALLOW = 1
+    # Prevent the message from being shown normally, filtered as Junk message.
+    JUNK = 2
+    # Prevent the message from being shown normally, filtered as Promotional message.
+    PROMOTION = 3
+    # Prevent the message from being shown normally, filtered as Transactional message.
+    TRANSACTION = 4
+
+class SubActionModel(Enum):
+    NONE = 0
+
+class OutputModel(BaseModel):
+    action: ActionModel
+    sub_action: SubActionModel
+
+pipe = pipeline(task="text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
+
+@app.get("/.well-known/apple-app-site-association", include_in_schema=False)
+def get_well_known_aasa():
+    return JSONResponse(
+        content={
+            "messagefilter": {
+                "apps": [
+                    "X9NN3FSS3T.com.lela.Serenity.SerenityMessageFilterExtension",
+                    "X9NN3FSS3T.com.lela.Serenity"
+                ]
+            }
+        },
+        media_type="application/json"
+    )
+
+@app.post("/predict")
+def predict(input: InputModel) -> OutputModel:
+    label = pipe(input.query.message.text)
+    if label[0]['label'] == 'LABEL_1':
+        return OutputModel(action=ActionModel.JUNK, sub_action=SubActionModel.NONE)
+    else:
+        return OutputModel(action=ActionModel.NONE, sub_action=SubActionModel.NONE)

requirements.txt

@@ -1,2 +1,5 @@
 fastapi
 uvicorn[standard]
+pydantic
+transformers
+torch