Merge pull request #27 from iurwpoietknckvjndfsm-gndvkd/authentication

added create_superuser method, Added ML dir and dependacies

agriwise/crop_recomendation/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class CropRecomendationConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "agriwise.crop_recomendation"

agriwise/ml/crop_recommendation/crop_recommendation.py

@@ -0,0 +1,55 @@
+import joblib
+import numpy
+import pandas as pd
+
+
+class RandomForestClassifier:
+    def __init__(self):
+        self.model = joblib.load("./random_forest.joblib")
+
+    def preprocessing(self, input_data):
+        # JSON to pandas DataFrame
+        input_data = pd.DataFrame(input_data, index=[0])
+
+        return input_data
+
+    def predict(self, input_data):
+        return self.model.predict_proba(input_data)
+
+    def postprocessing(self, prediction):
+        categories = [
+            "apple",
+            "banana",
+            "blackgram",
+            "chickpea",
+            "coconut",
+            "coffee",
+            "cotton",
+            "grapes",
+            "jute",
+            "kidneybeans",
+            "lentil",
+            "maize",
+            "mango",
+            "mothbeans",
+            "mungbean",
+            "muskmelon",
+            "orange",
+            "papaya",
+            "pigeonpeas",
+            "pomegranate",
+            "rice",
+            "watermelon",
+        ]
+        index_max_predict = numpy.argmax(prediction)
+        return categories[index_max_predict]
+
+    def compute_prediction(self, input_data):
+        try:
+            input_data = self.preprocessing(input_data)
+            prediction = self.predict(input_data)[0]  # only one sample
+            prediction = self.postprocessing(prediction)
+        except Exception as e:
+            return {"status": "Error", "message": str(e)}
+
+        return prediction

agriwise/users/models.py

@@ -17,6 +17,22 @@ def create_user(self, username, email, password=None):
         user.save()
         return user
 
+    def create_superuser(self, username, email, password=None):
+        if not username:
+            raise TypeError("superuser must have username")
+        if not email:
+            raise TypeError("superuser must have email")
+        if not password:
+            raise TypeError("superuser must have password")
+        email = self.normalize_email(email)
+        user = self.model(email=email, username=username)
+        user.set_password(password)
+        user.is_superuser = True
+        user.is_staff = True
+        user.is_active = True
+        user.save()
+        return user
+
 
 class User(AbstractUser):
     name = CharField(_("Name of User"), blank=True, max_length=255)

config/settings/base.py

@@ -81,6 +81,7 @@
 
 LOCAL_APPS = [
     "agriwise.users",
+    "agriwise.crop_recomendation",
     # Your stuff: custom apps go here
 ]
 # https://docs.djangoproject.com/en/dev/ref/settings/#installed-apps
@@ -110,6 +111,7 @@
     "SERIALIZERS": {
         "user_create": "agriwise.users.api.serializers.UserCreateSerializer"
     },
+    "PASSWORD_RESET_CONFIRM_URL": "users/auth/api/users/password/reset/confirm/{uid}/{token}",
     "ACTIVATION_URL": "users/auth/api/users/activation/{uid}/{token}",
     "SEND_ACTIVATION_EMAIL": True,
     "LOGIN_FIELD": "email",

requirements/base.txt

@@ -25,3 +25,7 @@ django-cors-headers==3.13.0  # https://github.com/adamchainz/django-cors-headers
 drf-spectacular==0.25.1  # https://github.com/tfranzel/drf-spectacular
 #New
 djoser==2.1.0
+#ML dependancies
+pandas==1.5.3
+joblib==1.2.0
+numpy==1.24.2