Add Flask app for bullying classification

Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.9-slim
+
+# Create a non-root user
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user
+WORKDIR $HOME/app
+
+# Copy and install dependencies
+COPY --chown=user requirements.txt requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application files
+COPY --chown=user . .
+
+# Expose the application port
+EXPOSE 7860
+
+# Run the application
+CMD ["python", "app.py"]

README.md

@@ -1,10 +1,38 @@
 ---
-title: Flaskbullybert
-emoji: 🐨
-colorFrom: blue
-colorTo: yellow
+title: Bullying Classifier API
+emoji: 🛡️
+colorFrom: indigo
+colorTo: blue
 sdk: docker
+app_port: 7860
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Bullying Classifier API
+
+This Flask application provides a RESTful API for classifying text inputs as bullying or not using a BERT-based model.
+
+## Endpoints
+
+- `POST /classify/text` – Classifies text as bullying or not.
+- `POST /classify/audio` – Placeholder for audio classification.
+- `POST /classify/image` – Placeholder for image classification.
+
+## Usage
+
+Send a POST request to `/classify/text` with a JSON body:
+
+```json
+{
+  "text": "your text here"
+}
+```
+
+## Docker
+
+Build and run the container:
+
+```sh
+docker build -t bullying-classifier .
+docker run -p 7860:7860 bullying-classifier
+```

app.py

@@ -0,0 +1,33 @@
+from flask import Flask, request, jsonify
+from transformers import pipeline
+
+app = Flask(__name__)
+
+# Load the text classification pipeline
+classifier = pipeline("text-classification", model="Davephoenix/bert-bullying-detector")
+
+
+@app.route("/classify/text", methods=["POST"])
+def classify_text():
+    data = request.get_json()
+    if not data or "text" not in data:
+        return jsonify({"error": "Missing 'text' in request body"}), 400
+    try:
+        result = classifier(data["text"])
+        return jsonify(result)
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+
+@app.route("/classify/audio", methods=["POST"])
+def classify_audio():
+    return jsonify({"error": "Audio classification not implemented yet"}), 501
+
+
+@app.route("/classify/image", methods=["POST"])
+def classify_image():
+    return jsonify({"error": "Image classification not implemented yet"}), 501
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

requirements.txt

@@ -0,0 +1,3 @@
+flask
+transformers
+torch